fixes

Author: edwinokonkwo

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

@@ -12,14 +12,15 @@
 from ldai.chat import ChatResponse, TrackedChat
 # Export main client
 from ldai.client import LDAIClient
+from ldai.config import LDMessage, ModelConfig, ProviderConfig
 # Export judge
 from ldai.judge import AIJudge, EvalScore, JudgeResponse, StructuredResponse
 # Export models for convenience
 from ldai.models import (  # Deprecated aliases for backward compatibility
     AIAgentConfig, AIAgentConfigDefault, AIAgentConfigRequest, AIAgents,
     AICompletionConfig, AICompletionConfigDefault, AIConfig, AIJudgeConfig,
     AIJudgeConfigDefault, JudgeConfiguration, LDAIAgent, LDAIAgentConfig,
-    LDAIAgentDefaults, LDMessage, ModelConfig, ProviderConfig)
+    LDAIAgentDefaults)
 
 __all__ = [
     'LDAIClient',

packages/sdk/server-ai/src/ldai/chat/tracked_chat.py

@@ -5,9 +5,10 @@
 from typing import Dict, List, Optional
 
 from ldai.chat.types import ChatResponse
+from ldai.config.types import LDMessage
 from ldai.judge import AIJudge
 from ldai.judge.types import JudgeResponse
-from ldai.models import AICompletionConfig, LDMessage
+from ldai.models import AICompletionConfig
 from ldai.providers.ai_provider import AIProvider
 from ldai.tracker import LDAIConfigTracker
 

packages/sdk/server-ai/src/ldai/chat/types.py

@@ -3,8 +3,8 @@
 from dataclasses import dataclass
 from typing import Any, List, Optional
 
+from ldai.config.types import LDMessage
 from ldai.metrics import LDAIMetrics
-from ldai.models import LDMessage
 
 
 @dataclass

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

@@ -7,12 +7,12 @@
 from ldclient.client import LDClient
 
 from ldai.chat import TrackedChat
+from ldai.config import LDMessage, ModelConfig, ProviderConfig
 from ldai.judge import AIJudge
 from ldai.models import (AIAgentConfig, AIAgentConfigDefault,
                          AIAgentConfigRequest, AIAgents, AICompletionConfig,
                          AICompletionConfigDefault, AIJudgeConfig,
-                         AIJudgeConfigDefault, JudgeConfiguration, LDMessage,
-                         ModelConfig, ProviderConfig)
+                         AIJudgeConfigDefault, JudgeConfiguration)
 from ldai.providers.ai_provider_factory import (AIProviderFactory,
                                                 SupportedAIProvider)
 from ldai.tracker import LDAIConfigTracker

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

@@ -0,0 +1,5 @@
+"""Config module for LaunchDarkly AI SDK."""
+
+from ldai.config.types import LDMessage, ModelConfig, ProviderConfig
+
+__all__ = ['LDMessage', 'ModelConfig', 'ProviderConfig']

packages/sdk/server-ai/src/ldai/config/types.py

@@ -0,0 +1,102 @@
+"""Types for configuration."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Any, Dict, Literal, Optional
+
+
+@dataclass
+class LDMessage:
+    role: Literal['system', 'user', 'assistant']
+    content: str
+
+    def to_dict(self) -> dict:
+        """
+        Render the given message as a dictionary object.
+        """
+        return {
+            'role': self.role,
+            'content': self.content,
+        }
+
+
+class ModelConfig:
+    """
+    Configuration related to the model.
+    """
+
+    def __init__(self, name: str, parameters: Optional[Dict[str, Any]] = None, custom: Optional[Dict[str, Any]] = None):
+        """
+        :param name: The name of the model.
+        :param parameters: Additional model-specific parameters.
+        :param custom: Additional customer provided data.
+        """
+        self._name = name
+        self._parameters = parameters
+        self._custom = custom
+
+    @property
+    def name(self) -> str:
+        """
+        The name of the model.
+        """
+        return self._name
+
+    def get_parameter(self, key: str) -> Any:
+        """
+        Retrieve model-specific parameters.
+
+        Accessing a named, typed attribute (e.g. name) will result in the call
+        being delegated to the appropriate property.
+        """
+        if key == 'name':
+            return self.name
+
+        if self._parameters is None:
+            return None
+
+        return self._parameters.get(key)
+
+    def get_custom(self, key: str) -> Any:
+        """
+        Retrieve customer provided data.
+        """
+        if self._custom is None:
+            return None
+
+        return self._custom.get(key)
+
+    def to_dict(self) -> dict:
+        """
+        Render the given model config as a dictionary object.
+        """
+        return {
+            'name': self._name,
+            'parameters': self._parameters,
+            'custom': self._custom,
+        }
+
+
+class ProviderConfig:
+    """
+    Configuration related to the provider.
+    """
+
+    def __init__(self, name: str):
+        self._name = name
+
+    @property
+    def name(self) -> str:
+        """
+        The name of the provider.
+        """
+        return self._name
+
+    def to_dict(self) -> dict:
+        """
+        Render the given provider config as a dictionary object.
+        """
+        return {
+            'name': self._name,
+        }

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

@@ -7,9 +7,10 @@
 import chevron
 
 from ldai.chat.types import ChatResponse
+from ldai.config.types import LDMessage
 from ldai.judge.evaluation_schema_builder import EvaluationSchemaBuilder
 from ldai.judge.types import EvalScore, JudgeResponse, StructuredResponse
-from ldai.models import AIJudgeConfig, LDMessage
+from ldai.models import AIJudgeConfig
 from ldai.providers.ai_provider import AIProvider
 from ldai.tracker import LDAIConfigTracker
 

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

@@ -1,110 +1,17 @@
+from __future__ import annotations
+
 import warnings
 from dataclasses import dataclass, field
-from typing import Any, Dict, List, Literal, Optional, Union
+from typing import Any, Dict, List, Optional, Union
 
+from ldai.config.types import LDMessage, ModelConfig, ProviderConfig
 from ldai.tracker import LDAIConfigTracker
 
-
-@dataclass
-class LDMessage:
-    role: Literal['system', 'user', 'assistant']
-    content: str
-
-    def to_dict(self) -> dict:
-        """
-        Render the given message as a dictionary object.
-        """
-        return {
-            'role': self.role,
-            'content': self.content,
-        }
-
-
-class ModelConfig:
-    """
-    Configuration related to the model.
-    """
-
-    def __init__(self, name: str, parameters: Optional[Dict[str, Any]] = None, custom: Optional[Dict[str, Any]] = None):
-        """
-        :param name: The name of the model.
-        :param parameters: Additional model-specific parameters.
-        :param custom: Additional customer provided data.
-        """
-        self._name = name
-        self._parameters = parameters
-        self._custom = custom
-
-    @property
-    def name(self) -> str:
-        """
-        The name of the model.
-        """
-        return self._name
-
-    def get_parameter(self, key: str) -> Any:
-        """
-        Retrieve model-specific parameters.
-
-        Accessing a named, typed attribute (e.g. name) will result in the call
-        being delegated to the appropriate property.
-        """
-        if key == 'name':
-            return self.name
-
-        if self._parameters is None:
-            return None
-
-        return self._parameters.get(key)
-
-    def get_custom(self, key: str) -> Any:
-        """
-        Retrieve customer provided data.
-        """
-        if self._custom is None:
-            return None
-
-        return self._custom.get(key)
-
-    def to_dict(self) -> dict:
-        """
-        Render the given model config as a dictionary object.
-        """
-        return {
-            'name': self._name,
-            'parameters': self._parameters,
-            'custom': self._custom,
-        }
-
-
-class ProviderConfig:
-    """
-    Configuration related to the provider.
-    """
-
-    def __init__(self, name: str):
-        self._name = name
-
-    @property
-    def name(self) -> str:
-        """
-        The name of the provider.
-        """
-        return self._name
-
-    def to_dict(self) -> dict:
-        """
-        Render the given provider config as a dictionary object.
-        """
-        return {
-            'name': self._name,
-        }
-
-
 # ============================================================================
 # Judge Types
 # ============================================================================
 
+
 @dataclass(frozen=True)
 class JudgeConfiguration:
     """
@@ -128,7 +35,7 @@ def to_dict(self) -> dict:
                 'samplingRate': self.sampling_rate,
             }
 
-    judges: List['JudgeConfiguration.Judge']
+    judges: List[JudgeConfiguration.Judge]
 
     def to_dict(self) -> dict:
         """

packages/sdk/server-ai/src/ldai/providers/ai_provider.py

@@ -5,9 +5,10 @@
 from typing import Any, Dict, List, Optional
 
 from ldai.chat.types import ChatResponse
+from ldai.config.types import LDMessage
 from ldai.judge.types import StructuredResponse
 from ldai.metrics import LDAIMetrics
-from ldai.models import AIConfigKind, LDMessage
+from ldai.models import AIConfigKind
 
 
 class AIProvider(ABC):