2929from elastic_evals .export .git_metadata import get_git_metadata
3030from elastic_evals .inference import KibanaInferenceClient
3131from elastic_evals .tracing import (
32- ElasticsearchTraceClient ,
3332 get_current_trace_id ,
3433 with_evaluator_span ,
3534 with_task_span ,
@@ -76,7 +75,6 @@ def __init__(self, config: ElasticEvalsConfig, logger: logging.Logger | None = N
7675 kibana_url = self .config .kibana_url ,
7776 api_key = self .config .kibana_api_key ,
7877 )
79- self ._trace_client : ElasticsearchTraceClient | None = None
8078
8179 def get_inference_client (self ) -> KibanaInferenceClient :
8280 if self ._inference_client is None :
@@ -91,16 +89,6 @@ def get_inference_client(self) -> KibanaInferenceClient:
9189 def get_evaluators_client (self ) -> KibanaEvaluatorsClient :
9290 return self ._evaluators_client
9391
94- def get_trace_client (self ) -> ElasticsearchTraceClient :
95- if self ._trace_client is None :
96- if not self .config .elasticsearch_url :
97- raise ValueError ("ELASTICSEARCH_URL is required for trace-based evaluators" )
98- self ._trace_client = ElasticsearchTraceClient (
99- elasticsearch_url = self .config .elasticsearch_url ,
100- api_key = self .config .elasticsearch_api_key ,
101- )
102- return self ._trace_client
103-
10492 async def run_experiment (
10593 self ,
10694 * ,
0 commit comments