Files
lcel-teacher/eval/test_chain.py
2024-01-26 12:09:10 -08:00

25 lines
830 B
Python

import langsmith
import langsmith.env
from app.context_stuffing_chain import chain as context_stuffing
from langchain.smith import RunEvalConfig
import uuid
if __name__ == "__main__":
client = langsmith.Client()
git_info = langsmith.env.get_git_info()
branch, commit = git_info["branch"], git_info["commit"]
project_name = f"lcel-teacher-{branch}-{commit[:4]}-{uuid.uuid4().hex[:4]}"
eval_config = RunEvalConfig(
evaluators=["qa"],
)
test_results = client.run_on_dataset(
dataset_name="lcel-teacher-eval",
llm_or_chain_factory=lambda: (lambda x: x["question"]) | context_stuffing,
project_name=project_name,
evaluation=eval_config,
verbose=True,
project_metadata={"context": "regression-tests"},
)
test_results.get_aggregate_feedback()