[REL-11697] move response data class to avoid circular import

Author: andrewklatzke

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

@@ -2,13 +2,13 @@
 
 from ldclient import log
 
-from ldai.agent_graph import AgentGraphDefinition
+from ldai.agent_graph import AgentGraphDefinition, AIAgentGraphResponse
 from ldai.chat import Chat
 from ldai.client import LDAIClient
 from ldai.judge import Judge
 from ldai.models import (  # Deprecated aliases for backward compatibility
     AIAgentConfig, AIAgentConfigDefault, AIAgentConfigRequest,
-    AIAgentGraphConfig, AIAgentGraphResponse, AIAgents, AICompletionConfig,
+    AIAgentGraphConfig, AIAgents, AICompletionConfig,
     AICompletionConfigDefault, AIConfig, AIJudgeConfig, AIJudgeConfigDefault,
     Edge, JudgeConfiguration, LDAIAgent, LDAIAgentConfig, LDAIAgentDefaults,
     LDMessage, ModelConfig, ProviderConfig)

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

@@ -1,5 +1,6 @@
 """Graph implementation for managing AI agent graphs."""
 
+from dataclasses import dataclass
 from typing import Any, Callable, Dict, List, Optional, Set
 
 from ldclient import Context
@@ -240,3 +241,16 @@ def reverse_traverse(
                 )
 
         return execution_context[self._agent_graph.root_config_key]
+
+
+# ============================================================================
+# AI Config Agent Graph Response
+# ============================================================================
+@dataclass
+class AIAgentGraphResponse:
+    """
+    Agent graph response.
+    """
+
+    enabled: bool
+    graph: Optional[AgentGraphDefinition] = None

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

@@ -5,15 +5,15 @@
 from ldclient.client import LDClient
 
 from ldai import log
-from ldai.agent_graph import AgentGraphDefinition
+from ldai.agent_graph import AgentGraphDefinition, AIAgentGraphResponse
 from ldai.chat import Chat
 from ldai.judge import Judge
 from ldai.models import (AIAgentConfig, AIAgentConfigDefault,
-                         AIAgentConfigRequest, AIAgentGraphConfig,
-                         AIAgentGraphResponse, AIAgents, AICompletionConfig,
-                         AICompletionConfigDefault, AIJudgeConfig,
-                         AIJudgeConfigDefault, Edge, JudgeConfiguration,
-                         LDMessage, ModelConfig, ProviderConfig)
+                         AIAgentConfigRequest, AIAgentGraphConfig, AIAgents,
+                         AICompletionConfig, AICompletionConfigDefault,
+                         AIJudgeConfig, AIJudgeConfigDefault, Edge,
+                         JudgeConfiguration, LDMessage, ModelConfig,
+                         ProviderConfig)
 from ldai.providers.ai_provider_factory import AIProviderFactory
 from ldai.tracker import LDAIConfigTracker
 

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

@@ -372,19 +372,6 @@ class AIAgentGraphConfig:
     description: Optional[str] = ""
 
 
-# ============================================================================
-# AI Config Agent Graph Response
-# ============================================================================
-@dataclass
-class AIAgentGraphResponse:
-    """
-    Agent graph response.
-    """
-
-    enabled: bool
-    graph: Optional[AgentGraphDefinition] = None
-
-
 # ============================================================================
 # Deprecated Type Aliases for Backward Compatibility
 # ============================================================================