Update imports, add isort configuration, and improve type aliasing for better clarity and organization.

Author: edwinokonkwo

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

@@ -42,6 +42,11 @@ ignore_missing_imports = true
 install_types = true
 non_interactive = true
 
+[tool.isort]
+profile = "black"
+known_third_party = ["langchain", "langchain_core", "ldai"]
+sections = ["FUTURE", "STDLIB", "THIRDPARTY", "FIRSTPARTY", "LOCALFOLDER"]
+
 
 [tool.pytest.ini_options]
 addopts = ["-ra"]

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

@@ -5,11 +5,19 @@
 from langchain_core.language_models.chat_models import BaseChatModel
 from langchain_core.messages import AIMessage, BaseMessage, HumanMessage, SystemMessage
 
-from ldai.models import AIConfigKind, LDMessage
-from ldai.providers.ai_provider import AIProvider
+from ldai import (
+    AIAgentConfig,
+    AICompletionConfig,
+    AIJudgeConfig,
+    LDMessage,
+)
+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):
     """

packages/ai-providers/server-ai-langchain/tests/test_langchain_provider.py

@@ -1,13 +1,13 @@
 """Tests for LangChain Provider."""
 
 import pytest
-from unittest.mock import AsyncMock, MagicMock, patch
+from unittest.mock import AsyncMock, MagicMock
 
 from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
 
+from ldai import LDMessage
+
 from ldai_langchain import LangChainProvider
-from ldai.models import LDMessage
-from ldai.tracker import TokenUsage
 
 
 class TestConvertMessagesToLangchain: