add key to model

Author: edwinokonkwo

ldai/client.py

@@ -47,6 +47,7 @@ def completion_config(
         )
 
         config = AICompletionConfig(
+            key=key,
             enabled=bool(enabled),
             model=model,
             messages=messages,
@@ -104,6 +105,7 @@ def judge_config(
         evaluation_metric_keys = variation.get('evaluationMetricKeys', default_value.evaluation_metric_keys or [])
 
         config = AIJudgeConfig(
+            key=key,
             enabled=bool(enabled),
             evaluation_metric_keys=evaluation_metric_keys,
             model=model,
@@ -551,6 +553,7 @@ def __evaluate_agent(
         final_instructions = instructions if instructions is not None else default_value.instructions
 
         return AIAgentConfig(
+            key=key,
             enabled=bool(enabled) if enabled is not None else (default_value.enabled or False),
             model=model or default_value.model,
             provider=provider or default_value.provider,

ldai/models.py

@@ -170,6 +170,7 @@ class AIConfig:
     """
     Base AI Config interface without mode-specific fields.
     """
+    key: str
     enabled: bool
     model: Optional[ModelConfig] = None
     provider: Optional[ProviderConfig] = None