klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-02 21:19:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Optionally pass references used by agent for response to eval scorers

Browse Source 
This will allow the eval framework to evaluate retrieval quality too
This commit is contained in:
Debanjum
2025-01-03 12:19:07 +07:00
parent 8231f4bb6e
commit daeba66c0d
1 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/evals/eval.py
View File

@@ -238,14 +238,18 @@ def get_agent_response(prompt: str) -> Dict[str, Any]:
)
response.raise_for_status()
response_json = response.json()
return {"response": response_json.get("response", ""), "usage": response_json.get("usage", {})}
return {
"response": response_json.get("response", ""),
"usage": response_json.get("usage", {}),
"references": response_json.get("references", {}),
}
except Exception as e:
logger.error(f"Error getting agent response: {e}")
return {"response": "", "usage": {}}
return {"response": "", "usage": {}, "references": {}}
def evaluate_response_with_mcq_match(
query: str, agent_response: str, ground_truth: str
query: str, agent_response: str, ground_truth: str, agent_references: dict = {}
) -> tuple[bool | None, str, float]:
"""Evaluate Khoj response against benchmark ground truth using string matching"""
try:
@@ -266,7 +270,7 @@ def evaluate_response_with_mcq_match(
def evaluate_response_with_gemini(
query: str, agent_response: str, ground_truth: str, eval_model=GEMINI_EVAL_MODEL
query: str, agent_response: str, ground_truth: str, agent_references: dict = {}, eval_model=GEMINI_EVAL_MODEL
) -> tuple[bool | None, str, float]:
"""Evaluate Khoj response against benchmark ground truth using Gemini"""
evaluation_prompt = f"""
@@ -331,13 +335,14 @@ def process_batch(batch, batch_start, results, dataset_length, response_evaluato
response = get_agent_response(prompt)
agent_response = response["response"]
agent_usage = response["usage"]
agent_references = response["references"]
# Evaluate response
if is_none_or_empty(agent_response):
decision = None
explanation = "Agent response is empty. This maybe due to a service error."
else:
decision, explanation, eval_cost = response_evaluator(prompt, agent_response, answer)
decision, explanation, eval_cost = response_evaluator(prompt, agent_response, answer, agent_references)
# Store results
results.append(
@@ -350,6 +355,7 @@ def process_batch(batch, batch_start, results, dataset_length, response_evaluato
"evaluation_explanation": explanation,
"reasoning_type": reasoning_type,
"usage": agent_usage,
"references": agent_references,
}
)