langchain-dev-utils 1.4.1__tar.gz → 1.4.2__tar.gz

langchain_dev_utils-1.4.2/AGENTS.md

@@ -0,0 +1,173 @@
+# Agentic Coding Guidelines for langchain-dev-utils
+
+This file provides guidelines for AI agents working in this repository.
+
+## Project Overview
+
+A Python utility library for LangChain and LangGraph development. Uses hatchling build system, uv for package management, ruff for linting, and pytest for testing.
+
+---
+
+## Build/Lint/Test Commands
+
+### Package Management (uv)
+```bash
+# Install dependencies
+uv sync
+
+# Install with optional dependencies
+uv sync --extra standard
+
+# Install dev dependencies
+uv sync --group dev
+
+# Install test dependencies
+uv sync --group tests
+```
+
+### Linting & Formatting (ruff)
+```bash
+# Check linting
+uv run ruff check .
+
+# Check specific file
+uv run ruff check src/langchain_dev_utils/path/to/file.py
+
+# Fix auto-fixable issues
+uv run ruff check --fix .
+
+# Format code
+uv run ruff format .
+
+# Format specific file
+uv run ruff format src/langchain_dev_utils/path/to/file.py
+```
+
+### Testing (pytest)
+```bash
+# Run all tests
+uv run pytest
+
+# Run with verbose output
+uv run pytest -v
+
+# Run specific test file
+uv run pytest tests/test_agent.py
+
+# Run specific test function
+uv run pytest tests/test_agent.py::test_prebuilt_agent
+
+# Run specific test class
+uv run pytest tests/test_chat_models.py::TestImageProcessing
+
+# Run with asyncio support (automatically configured)
+uv run pytest -s
+
+# Run tests matching pattern
+uv run pytest -k "test_load"
+```
+
+### Build
+```bash
+# Build package
+uv build
+
+# Build wheel only
+uv build --wheel
+```
+
+---
+
+## Code Style Guidelines
+
+### Imports
+- Use absolute imports for external packages
+- Use relative imports within the package (e.g., `from ..chat_models import ...`)
+- Group imports: stdlib → third-party → local
+- Ruff handles import sorting automatically
+
+### Type Hints
+- Use type hints for all function parameters and return types
+- Use `Optional[Type]` or `Type | None` for nullable types (both acceptable)
+- Use `Any` sparingly and only when necessary
+- Use `Sequence`, `Mapping` for generic collections
+- Use generics with TypeVars where appropriate
+
+### Naming Conventions
+- `snake_case` for functions, methods, variables
+- `PascalCase` for classes
+- `UPPER_CASE` for constants
+- `_leading_underscore` for private/internal functions
+- Leading double underscore for name mangling when needed
+
+### Docstrings
+- Use Google-style docstrings
+- Include Args, Returns, Raises sections for public functions
+- Include Examples section for complex functions
+- Keep docstrings under 100 characters per line when possible
+
+### Code Structure
+- Maximum line length: 88 characters (ruff enforces, E501 ignored)
+- Use trailing commas in multi-line structures
+- Two blank lines between top-level functions/classes
+- One blank line between methods
+
+### Error Handling
+- Use specific exception types, not bare `except:`
+- Provide descriptive error messages with f-strings
+- Use `raise ValueError(msg)` pattern with descriptive messages
+- Avoid bare `raise` statements
+
+### Async Code
+- Use `pytest.mark.asyncio` for async test functions
+- Use `async`/`await` consistently
+- Prefer `asyncio` primitives from standard library
+
+### Ruff Configuration
+- Enabled rules: E, F, I, PGH003, T201
+- Import sorting (I) is enforced
+- No print statements in production code (T201)
+
+---
+
+## Testing Guidelines
+
+- Tests live in `tests/` directory
+- Test files named `test_*.py`
+- Test functions named `test_*`
+- Use pytest fixtures for setup/teardown
+- Use `pytest.mark.asyncio` for async tests
+- Mock external API calls in unit tests
+- Integration tests use actual APIs (marked implicitly by test names)
+- Tests use `langchain-tests` for standard integration test suites
+
+---
+
+## Project Structure
+
+```
+langchain-dev-utils/
+├── src/langchain_dev_utils/      # Source code
+│   ├── agents/                   # Agent utilities
+│   ├── chat_models/              # Chat model utilities
+│   ├── embeddings/               # Embedding utilities
+│   ├── graph/                    # Graph utilities
+│   ├── message_convert/          # Message conversion
+│   ├── pipeline/                 # Pipeline utilities (deprecated since v1.4.0, will be removed in v1.5.0)
+│   └── tool_calling/             # Tool calling utilities
+├── tests/                        # Test files
+├── docs/                         # Documentation
+├── pyproject.toml                # Project configuration
+└── uv.lock                       # Dependency lock file
+```
+
+---
+
+## Dependencies
+
+- Core: langchain, langchain-core, langgraph
+- Optional: jinja2, json-repair, langchain-openai
+- Dev: ruff, dashscope, langchain-model-profiles
+- Test: python-dotenv, langchain-tests, langchain-deepseek, langchain-qwq, langchain-ollama, langchain-community
+
+Python version: >=3.11

{langchain_dev_utils-1.4.1 → langchain_dev_utils-1.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-dev-utils
-Version: 1.4.1
+Version: 1.4.2
 Summary: A practical utility library for LangChain and LangGraph development
 Project-URL: Source Code, https://github.com/TBice123123/langchain-dev-utils
 Project-URL: repository, https://github.com/TBice123123/langchain-dev-utils

{langchain_dev_utils-1.4.1 → langchain_dev_utils-1.4.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "langchain-dev-utils"
-version = "1.4.1"
+version = "1.4.2"
 description = "A practical utility library for LangChain and LangGraph development"
 readme = "README.md"
 authors = [{ name = "tiebingice", email = "tiebingice123@outlook.com" }]

langchain_dev_utils-1.4.2/src/langchain_dev_utils/__init__.py

@@ -0,0 +1 @@
+__version__ = "1.4.2"

{langchain_dev_utils-1.4.1 → langchain_dev_utils-1.4.2}/src/langchain_dev_utils/_utils.py

@@ -1,11 +1,48 @@
 from importlib import util
 from typing import Literal, Optional, cast
 
+from langchain_core.tools import BaseTool
 from langgraph.graph import StateGraph
 from langgraph.graph.state import StateNode
 from pydantic import BaseModel
 
 
+def _duplicate_tools(tools: list[BaseTool]) -> list[BaseTool]:
+    """Duplicate tools with the same name.
+
+    Args:
+        tools (list[BaseTool]): The list of tools.
+
+    Returns:
+        list[BaseTool]: The duplicated tools.
+    """
+    tool_name_set = set()
+    duplicated_tools = []
+    for tool_obj in tools:
+        if tool_obj.name not in tool_name_set:
+            duplicated_tools.append(tool_obj)
+            tool_name_set.add(tool_obj.name)
+    return duplicated_tools
+
+
+def _merge_tools(tools: list[BaseTool]) -> list[BaseTool]:
+    """Merge tools with the same name.
+
+    Args:
+        tools (list[BaseTool]): The list of tools.
+
+    Returns:
+        list[BaseTool]: The merged tools.
+    """
+    tool_name_set = set()
+    merged_tools = []
+    for tool_obj in tools:
+        if tool_obj.name not in tool_name_set:
+            merged_tools.append(tool_obj)
+            tool_name_set.add(tool_obj.name)
+    return merged_tools
+
+
 def _transform_node_to_tuple(
     node: StateNode | tuple[str, StateNode],
 ) -> tuple[str, StateNode]:

{langchain_dev_utils-1.4.1 → langchain_dev_utils-1.4.2}/src/langchain_dev_utils/agents/middleware/handoffs.py

@@ -1,4 +1,4 @@
-from typing import Any, Awaitable, Callable, Literal, cast
+from typing import Awaitable, Callable, Literal, cast
 
 from langchain.agents import AgentState
 from langchain.agents.middleware import AgentMiddleware, ModelRequest, ModelResponse
@@ -9,6 +9,7 @@ from langchain_core.messages import SystemMessage, ToolMessage
 from langgraph.types import Command
 from typing_extensions import NotRequired, Optional, TypedDict
 
+from langchain_dev_utils._utils import _duplicate_tools, _merge_tools
 from langchain_dev_utils.chat_models import load_chat_model
 
 
@@ -19,7 +20,7 @@ class MultiAgentState(AgentState):
 class AgentConfig(TypedDict):
     model: NotRequired[str | BaseChatModel]
     prompt: str | SystemMessage
-    tools: NotRequired[list[BaseTool | dict[str, Any]]]
+    tools: NotRequired[list[BaseTool]]
     default: NotRequired[bool]
     handoffs: list[str] | Literal["all"]
 
@@ -176,7 +177,16 @@ class HandoffAgentMiddleware(AgentMiddleware):
             agents_config,
             handoffs_tools,
         )
-        self.tools = handoffs_tools
+
+        all_tools = [
+            *handoffs_tools,
+        ]
+
+        for agent_name in agents_config:
+            tools = agents_config.get(agent_name, {}).get("tools", [])
+            all_tools.extend(tools)
+
+        self.tools = _merge_tools(all_tools)
 
     def _get_override_request(self, request: ModelRequest) -> ModelRequest:
         active_agent_name = request.state.get("active_agent", self.default_agent_name)
@@ -184,15 +194,14 @@ class HandoffAgentMiddleware(AgentMiddleware):
         _config = self.agents_config[active_agent_name]
 
         params = {}
-        if _config.get("model"):
-            model = _config.get("model")
+        if model := _config.get("model"):
             if isinstance(model, str):
                 model = load_chat_model(model)
             params["model"] = model
-        if _config.get("prompt"):
-            params["system_prompt"] = _config.get("prompt")
-        if _config.get("tools"):
-            params["tools"] = _config.get("tools")
+        if prompt := _config.get("prompt"):
+            params["system_prompt"] = prompt
+        if tools := _config.get("tools"):
+            params["tools"] = _duplicate_tools(tools)
 
         if params:
             return request.override(**params)

{langchain_dev_utils-1.4.1 → langchain_dev_utils-1.4.2}/src/langchain_dev_utils/agents/middleware/model_router.py

@@ -10,6 +10,7 @@ from langgraph.runtime import Runtime
 from pydantic import BaseModel, Field
 from typing_extensions import TypedDict
 
+from langchain_dev_utils._utils import _duplicate_tools, _merge_tools
 from langchain_dev_utils.chat_models import load_chat_model
 from langchain_dev_utils.message_convert import format_sequence
 
@@ -120,6 +121,13 @@ class ModelRouterMiddleware(AgentMiddleware):
 
         self.router_prompt = router_prompt
 
+        all_tools = []
+        for model in model_list:
+            if tools := model.get("tools"):
+                all_tools.extend(tools)
+
+        self.tools = _merge_tools(all_tools)
+
     def _select_model(self, messages: list[AnyMessage]):
         response = cast(
             SelectModel,
@@ -174,7 +182,7 @@ class ModelRouterMiddleware(AgentMiddleware):
                     model = load_chat_model(select_model_name)
             override_kwargs["model"] = model
             if model_values["tools"] is not None:
-                override_kwargs["tools"] = model_values["tools"]
+                override_kwargs["tools"] = _duplicate_tools(model_values["tools"])
             if model_values["system_prompt"] is not None:
                 override_kwargs["system_message"] = SystemMessage(
                     content=model_values["system_prompt"]

{langchain_dev_utils-1.4.1 → langchain_dev_utils-1.4.2}/src/langchain_dev_utils/agents/wrap.py

@@ -17,7 +17,7 @@ def _process_input(request: str, runtime: ToolRuntime) -> str:
 def _process_output(
     request: str, response: dict[str, Any], runtime: ToolRuntime
 ) -> Any:
-    return response["messages"][-1].content
+    return response["messages"][-1].text
 
 
 def get_subagent_name(runtime: ToolRuntime) -> str:

{langchain_dev_utils-1.4.1 → langchain_dev_utils-1.4.2}/tests/test_embedding.py

@@ -5,8 +5,8 @@ from langchain_tests.integration_tests.embeddings import EmbeddingsIntegrationTe
 
 from langchain_dev_utils.embeddings.adapters import create_openai_compatible_embedding
 
-SiliconFlowEmbeddings = create_openai_compatible_embedding(
-    "siliconflow", embedding_model_cls_name="SiliconFlowEmbeddings"
+ZAIEmbeddings = create_openai_compatible_embedding(
+    "zai", embedding_model_cls_name="ZAIEmbeddings"
 )
 
 
@@ -14,9 +14,9 @@ class TestStandard(EmbeddingsIntegrationTests):
     @property
     def embeddings_class(self) -> type[Embeddings]:
         """Embeddings class."""
-        return cast("type[Embeddings]", SiliconFlowEmbeddings)
+        return cast("type[Embeddings]", ZAIEmbeddings)
 
     @property
     def embedding_model_params(self) -> dict[str, Any]:
         """Embeddings model parameters."""
-        return {"model": "BAAI/bge-m3"}
+        return {"model": "embedding-3"}

{langchain_dev_utils-1.4.1 → langchain_dev_utils-1.4.2}/tests/test_handoffs_middleware.py

@@ -6,9 +6,7 @@ from langgraph.checkpoint.memory import InMemorySaver
 from langgraph.types import Command
 
 from langchain_dev_utils.agents import create_agent
-from langchain_dev_utils.agents.middleware import (
-    HandoffAgentMiddleware,
-)
+from langchain_dev_utils.agents.middleware import HandoffAgentMiddleware
 from langchain_dev_utils.agents.middleware.handoffs import AgentConfig
 from langchain_dev_utils.chat_models import load_chat_model
 
@@ -91,10 +89,6 @@ def test_handoffs_middleware():
                 handoffs_tool_overrides=handoffs_tool_map,
             )
         ],
-        tools=[
-            get_current_time,
-            run_code,
-        ],
         checkpointer=InMemorySaver(),
     )
 

{langchain_dev_utils-1.4.1 → langchain_dev_utils-1.4.2}/tests/test_load_embbeding.py

@@ -7,7 +7,6 @@ from langchain_dev_utils.embeddings import load_embeddings
 @pytest.fixture(
     params=[
         "dashscope:text-embedding-v4",
-        "siliconflow:BAAI/bge-m3",
         "ollama:bge-m3:latest",
     ]
 )

{langchain_dev_utils-1.4.1 → langchain_dev_utils-1.4.2}/tests/utils/register.py

@@ -27,10 +27,6 @@ def register_all_model_providers():
 def register_all_embeddings_providers():
     batch_register_embeddings_provider(
         [
-            {
-                "provider_name": "siliconflow",
-                "embeddings_model": "openai-compatible",
-            },
             {"provider_name": "dashscope", "embeddings_model": DashScopeEmbeddings},
         ]
     )

langchain_dev_utils-1.4.1/src/langchain_dev_utils/__init__.py

@@ -1 +0,0 @@
-__version__ = "1.4.1"