Skip to content

Commit 350f884

Browse files
committed
fixed linter issues
1 parent 8d01693 commit 350f884

2 files changed

Lines changed: 7 additions & 4 deletions

File tree

packages/sdk/server-ai/src/ldai/judge/__init__.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,7 @@
99
from ldai.judge.evaluation_schema_builder import EvaluationSchemaBuilder
1010
from ldai.models import AIJudgeConfig, LDMessage
1111
from ldai.providers.ai_provider import AIProvider
12-
from ldai.providers.types import (ChatResponse, EvalScore, JudgeResponse,
13-
StructuredResponse)
12+
from ldai.providers.types import ChatResponse, EvalScore, JudgeResponse
1413
from ldai.tracker import LDAIConfigTracker
1514

1615

@@ -70,8 +69,8 @@ async def evaluate(
7069
return None
7170

7271
messages = self._construct_evaluation_messages(input_text, output_text)
72+
assert self._evaluation_response_structure is not None
7373

74-
# Track metrics of the structured model invocation
7574
response = await self._ai_config_tracker.track_metrics_of(
7675
lambda: self._ai_provider.invoke_structured_model(messages, self._evaluation_response_structure),
7776
lambda result: result.metrics,
@@ -190,6 +189,10 @@ def _parse_evaluation_response(self, data: Dict[str, Any]) -> Dict[str, EvalScor
190189
evaluations = data['evaluations']
191190

192191
metric_key = self._ai_config.evaluation_metric_key
192+
if not metric_key:
193+
log.warn('Evaluation metric key is missing')
194+
return results
195+
193196
evaluation = evaluations.get(metric_key)
194197

195198
if not evaluation or not isinstance(evaluation, dict):

packages/sdk/server-ai/src/ldai/judge/evaluation_schema_builder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ def build(evaluation_metric_key: Optional[str]) -> Optional[Dict[str, Any]]:
1717
:param evaluation_metric_key: Evaluation metric key, or None if not available
1818
:return: Schema dictionary for structured output, or None if evaluation_metric_key is None
1919
"""
20-
if evaluation_metric_key is None:
20+
if not evaluation_metric_key:
2121
return None
2222

2323
return {

0 commit comments

Comments
 (0)