Skip to content

Commit 00a265e

Browse files
committed
Linting
1 parent 350f884 commit 00a265e

2 files changed

Lines changed: 11 additions & 8 deletions

File tree

packages/sdk/server-ai/src/ldai/client.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -99,28 +99,31 @@ def judge_config(
9999
)
100100

101101
variation = self._client.variation(key, context, default_value.to_dict())
102-
def _extract_evaluation_metric_key(variation: Dict[str, Any], default_value: AIJudgeConfigDefault) -> Optional[str]:
102+
103+
def _extract_evaluation_metric_key(
104+
variation: Dict[str, Any], default_value: AIJudgeConfigDefault
105+
) -> Optional[str]:
103106
"""
104107
Extract evaluation_metric_key with backward compatibility.
105-
108+
106109
Priority: 1) evaluationMetricKey from variation, 2) evaluation_metric_key from default,
107110
3) first from evaluationMetricKeys in variation, 4) first from evaluation_metric_keys in default
108111
"""
109112
if evaluation_metric_key := variation.get('evaluationMetricKey'):
110113
return evaluation_metric_key
111-
114+
112115
if default_value.evaluation_metric_key:
113116
return default_value.evaluation_metric_key
114-
117+
115118
variation_keys = variation.get('evaluationMetricKeys')
116119
if isinstance(variation_keys, list) and variation_keys:
117120
return variation_keys[0]
118-
121+
119122
if default_value.evaluation_metric_keys:
120123
return default_value.evaluation_metric_keys[0]
121-
124+
122125
return None
123-
126+
124127
evaluation_metric_key = _extract_evaluation_metric_key(variation, default_value)
125128

126129
config = AIJudgeConfig(

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
@@ -19,7 +19,7 @@ def build(evaluation_metric_key: Optional[str]) -> Optional[Dict[str, Any]]:
1919
"""
2020
if not evaluation_metric_key:
2121
return None
22-
22+
2323
return {
2424
'title': 'EvaluationResponse',
2525
'description': f"Response containing evaluation results for {evaluation_metric_key} metric",

0 commit comments

Comments
 (0)