move type to judge and chat instead

Author: edwinokonkwo

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

@@ -9,11 +9,11 @@
 __path__ = extend_path(__path__, __name__)
 
 # Export chat
-from ldai.chat import TrackedChat
+from ldai.chat import ChatResponse, TrackedChat
 # Export main client
 from ldai.client import LDAIClient
 # Export judge
-from ldai.judge import AIJudge, EvalScore, JudgeResponse
+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,
@@ -33,9 +33,11 @@
     'AIJudgeConfigDefault',
     'AIJudge',
     'TrackedChat',
+    'ChatResponse',
     'EvalScore',
     'JudgeConfiguration',
     'JudgeResponse',
+    'StructuredResponse',
     'LDMessage',
     'ModelConfig',
     'ProviderConfig',

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

@@ -1,5 +1,6 @@
 """Chat module for LaunchDarkly AI SDK."""
 
 from ldai.chat.tracked_chat import TrackedChat
+from ldai.chat.types import ChatResponse
 
-__all__ = ['TrackedChat']
+__all__ = ['TrackedChat', 'ChatResponse']

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

@@ -4,11 +4,11 @@
 import logging
 from typing import Dict, List, Optional
 
+from ldai.chat.types import ChatResponse
 from ldai.judge import AIJudge
 from ldai.judge.types import JudgeResponse
 from ldai.models import AICompletionConfig, LDMessage
 from ldai.providers.ai_provider import AIProvider
-from ldai.providers.types import ChatResponse
 from ldai.tracker import LDAIConfigTracker
 
 

…dk/server-ai/src/ldai/providers/types.py …ges/sdk/server-ai/src/ldai/chat/types.pypackages/sdk/server-ai/src/ldai/providers/types.py renamed to packages/sdk/server-ai/src/ldai/chat/types.py packages/sdk/server-ai/src/ldai/providers/types.py renamed to packages/sdk/server-ai/src/ldai/chat/types.py

@@ -1,4 +1,4 @@
-"""Types for AI provider responses."""
+"""Types for chat responses."""
 
 from dataclasses import dataclass
 from typing import Any, List, Optional
@@ -15,13 +15,3 @@ class ChatResponse:
     message: LDMessage
     metrics: LDAIMetrics
     evaluations: Optional[List[Any]] = None  # List of JudgeResponse, will be populated later
-
-
-@dataclass
-class StructuredResponse:
-    """
-    Structured response from AI models.
-    """
-    data: dict[str, Any]
-    raw_response: str
-    metrics: LDAIMetrics

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

@@ -1,6 +1,6 @@
 """Judge module for LaunchDarkly AI SDK."""
 
 from ldai.judge.ai_judge import AIJudge
-from ldai.judge.types import EvalScore, JudgeResponse
+from ldai.judge.types import EvalScore, JudgeResponse, StructuredResponse
 
-__all__ = ['AIJudge', 'EvalScore', 'JudgeResponse']
+__all__ = ['AIJudge', 'EvalScore', 'JudgeResponse', 'StructuredResponse']

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

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

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

@@ -3,6 +3,8 @@
 from dataclasses import dataclass
 from typing import Any, Dict, Optional
 
+from ldai.metrics import LDAIMetrics
+
 
 @dataclass
 class EvalScore:
@@ -42,3 +44,13 @@ def to_dict(self) -> Dict[str, Any]:
         if self.error is not None:
             result['error'] = self.error
         return result
+
+
+@dataclass
+class StructuredResponse:
+    """
+    Structured response from AI models.
+    """
+    data: dict[str, Any]
+    raw_response: str
+    metrics: LDAIMetrics

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

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