fixes

Author: edwinokonkwo

packages/ai-providers/server-ai-langchain/pyproject.toml

@@ -24,6 +24,9 @@ packages = [{ include = "ldai_langchain", from = "src" }]
 
 [tool.poetry.dependencies]
 python = ">=3.9,<4"
+# TODO: Before publishing, change to PyPI version: launchdarkly-server-sdk-ai = ">=X.Y.Z"
+# Path dependency is used for development because the published version doesn't yet export
+# the required types (LDMessage, AIConfigKind, etc.). Release server-ai first, then update.
 launchdarkly-server-sdk-ai = { path = "../../sdk/server-ai", develop = true }
 langchain-core = ">=0.2.0"
 langchain = ">=0.2.0"

packages/ai-providers/server-ai-langchain/src/ldai_langchain/langchain_provider.py

@@ -4,19 +4,12 @@
 
 from langchain_core.language_models.chat_models import BaseChatModel
 from langchain_core.messages import AIMessage, BaseMessage, HumanMessage, SystemMessage
-from ldai import (
-    AIAgentConfig,
-    AICompletionConfig,
-    AIJudgeConfig,
-    LDMessage,
-)
+from ldai import LDMessage
+from ldai.models import AIConfigKind
 from ldai.providers import AIProvider
 from ldai.providers.types import ChatResponse, LDAIMetrics, StructuredResponse
 from ldai.tracker import TokenUsage
 
-# Type alias matching the one in ldai.models
-AIConfigKind = Union[AIAgentConfig, AICompletionConfig, AIJudgeConfig]
-
 
 class LangChainProvider(AIProvider):
     """
@@ -44,7 +37,7 @@ async def create(ai_config: AIConfigKind, logger: Optional[Any] = None) -> 'Lang
         :param logger: Optional logger for the provider
         :return: Configured LangChainProvider instance
         """
-        llm = await LangChainProvider.create_langchain_model(ai_config)
+        llm = LangChainProvider.create_langchain_model(ai_config)
         return LangChainProvider(llm, logger)
 
     async def invoke_model(self, messages: List[LDMessage]) -> ChatResponse:
@@ -253,7 +246,7 @@ def convert_messages_to_langchain(
         return result
 
     @staticmethod
-    async def create_langchain_model(ai_config: AIConfigKind) -> BaseChatModel:
+    def create_langchain_model(ai_config: AIConfigKind) -> BaseChatModel:
         """
         Create a LangChain model from an AI configuration.