Skip to content

Commit ba4b5dd

Browse files
jsonbaileyclaude
andcommitted
chore: fix import sort order in LangChain provider tests
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
1 parent c65f6a8 commit ba4b5dd

2 files changed

Lines changed: 9 additions & 5 deletions

File tree

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

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,9 @@
11
"""Tests for LangChain Provider."""
22

3-
import pytest
43
from unittest.mock import AsyncMock, MagicMock
54

5+
import pytest
66
from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
7-
87
from ldai import LDMessage
98
from ldai.evaluator import Evaluator
109

@@ -404,6 +403,7 @@ class TestCreateAgent:
404403
def test_creates_agent_runner_with_instructions_and_tool_definitions(self):
405404
"""Should create LangChainAgentRunner wrapping a compiled graph."""
406405
from unittest.mock import patch
406+
407407
from ldai_langchain import LangChainAgentRunner
408408

409409
mock_ai_config = MagicMock()
@@ -436,6 +436,7 @@ def test_creates_agent_runner_with_instructions_and_tool_definitions(self):
436436
def test_creates_agent_runner_with_no_tools(self):
437437
"""Should create LangChainAgentRunner with no tool definitions."""
438438
from unittest.mock import patch
439+
439440
from ldai_langchain import LangChainAgentRunner
440441

441442
mock_ai_config = MagicMock()
@@ -522,6 +523,7 @@ class TestBuildTools:
522523

523524
def test_registers_sync_callable_as_structured_tool_func(self):
524525
from ldai.models import AIAgentConfig, ModelConfig, ProviderConfig
526+
525527
from ldai_langchain.langchain_helper import build_structured_tools
526528

527529
def sync_tool(x: str = '') -> str:
@@ -546,6 +548,7 @@ def sync_tool(x: str = '') -> str:
546548

547549
def test_registers_async_callable_as_structured_tool_coroutine(self):
548550
from ldai.models import AIAgentConfig, ModelConfig, ProviderConfig
551+
549552
from ldai_langchain.langchain_helper import build_structured_tools
550553

551554
async def async_tool(x: str = '') -> str:

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

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,16 @@
11
"""Tests for LangGraphAgentGraphRunner and LangChainRunnerFactory.create_agent_graph()."""
22

3-
import pytest
43
from unittest.mock import AsyncMock, MagicMock, patch
54

5+
import pytest
66
from ldai.agent_graph import AgentGraphDefinition
77
from ldai.evaluator import Evaluator
8-
from ldai.models import AIAgentGraphConfig, AIAgentConfig, ModelConfig, ProviderConfig
8+
from ldai.models import AIAgentConfig, AIAgentGraphConfig, ModelConfig, ProviderConfig
99
from ldai.providers import ToolRegistry
1010
from ldai.providers.types import AgentGraphRunnerResult
11-
from ldai_langchain.langgraph_agent_graph_runner import LangGraphAgentGraphRunner
11+
1212
from ldai_langchain.langchain_runner_factory import LangChainRunnerFactory
13+
from ldai_langchain.langgraph_agent_graph_runner import LangGraphAgentGraphRunner
1314

1415

1516
def _make_graph(enabled: bool = True) -> AgentGraphDefinition:

0 commit comments

Comments
 (0)