File tree Expand file tree Collapse file tree
integrations/agent_builder
integrations/agent_builder Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -94,7 +94,7 @@ async def evaluate(params: EvaluatorParams) -> EvaluationResult:
9494 metadata = result .metadata ,
9595 )
9696
97- return SimpleEvaluator (name = "correctness" , kind = "LLM" , evaluate = evaluate )
97+ return SimpleEvaluator (name = QUALITATIVE_EVALUATOR_NAME , kind = "LLM" , evaluate = evaluate )
9898
9999
100100def _analysis_explanation (summary : Any , fallback : str | None ) -> str | None :
Original file line number Diff line number Diff line change 1010
1111import httpx
1212
13+ from elastic_evals .api .response import raise_kibana_error
14+ from elastic_evals .api .retry import retry_kibana_api_call
1315from elastic_evals .integrations .agent_builder .constants import (
1416 AGENT_URL ,
1517 AGENTS_URL ,
2931 UpdateAgentRequest ,
3032 UpdateToolRequest ,
3133)
32- from elastic_evals .api .response import raise_kibana_error
33- from elastic_evals .api .retry import retry_kibana_api_call
3434from elastic_evals .utils .logging import log
3535
3636logger = log .getChild (__name__ )
Original file line number Diff line number Diff line change 66
77from __future__ import annotations
88
9- from elastic_evals .integrations .agent_builder .constants import AGENT_BUILDER_API_VERSION
109from elastic_evals .api .headers import build_kibana_headers
10+ from elastic_evals .integrations .agent_builder .constants import AGENT_BUILDER_API_VERSION
1111
1212
1313def build_agent_builder_headers (api_key : str | None ) -> dict [str , str ]:
Original file line number Diff line number Diff line change @@ -202,6 +202,7 @@ async def test_analysis_creators_preserve_legacy_results_through_kibana(
202202 correctness_result = await correctness .evaluate (params )
203203 groundedness_result = await groundedness .evaluate (params )
204204
205+ assert correctness .name == "Correctness Analysis"
205206 assert correctness_result .score is None
206207 assert correctness_result .label == "correctness-analysis"
207208 assert correctness_result .explanation == ("Factuality: FULLY_SUPPORTED, Relevance: RELEVANT, Sequence: MATCH" )
Original file line number Diff line number Diff line change 1313from pydantic import ValidationError
1414from tenacity import wait_none
1515
16+ from elastic_evals .api .errors import KibanaAPIError
1617from elastic_evals .integrations .agent_builder import (
1718 AgentBuilderClient ,
1819 AgentBuilderError ,
2526 UpdateAgentRequest ,
2627 UpdateToolRequest ,
2728)
28- from elastic_evals .api .errors import KibanaAPIError
2929
3030
3131class _RecordingAsyncClient :
You can’t perform that action at this time.
0 commit comments