langchain-dev-utils 1.2.13__py3-none-any.whl → 1.2.15__py3-none-any.whl

langchain_dev_utils/__init__.py

@@ -1 +1 @@
-__version__ = "1.2.13"
+__version__ = "1.2.15"

langchain_dev_utils/agents/middleware/model_router.py

@@ -19,6 +19,7 @@ class ModelDict(TypedDict):
     model_description: str
     tools: NotRequired[list[BaseTool | dict[str, Any]]]
     model_kwargs: NotRequired[dict[str, Any]]
+    model_instance: NotRequired[BaseChatModel]
     model_system_prompt: NotRequired[str]
 
 
@@ -76,7 +77,7 @@ class ModelRouterMiddleware(AgentMiddleware):
             model name or a BaseChatModel instance
         model_list: List of available routing models, each containing model_name,
             model_description, tools(Optional), model_kwargs(Optional),
-            model_system_prompt(Optional)
+            model_instance(Optional), model_system_prompt(Optional)
         router_prompt: Routing prompt template, uses default template if None
 
     Examples:
@@ -155,6 +156,7 @@ class ModelRouterMiddleware(AgentMiddleware):
                 "tools": item.get("tools", None),
                 "kwargs": item.get("model_kwargs", None),
                 "system_prompt": item.get("model_system_prompt", None),
+                "model_instance": item.get("model_instance", None),
             }
             for item in self.model_list
         }
@@ -163,10 +165,13 @@ class ModelRouterMiddleware(AgentMiddleware):
         override_kwargs = {}
         if select_model_name != "default-model" and select_model_name in model_dict:
             model_values = model_dict.get(select_model_name, {})
-            if model_values["kwargs"] is not None:
-                model = load_chat_model(select_model_name, **model_values["kwargs"])
+            if model_values["model_instance"] is not None:
+                model = model_values["model_instance"]
             else:
-                model = load_chat_model(select_model_name)
+                if model_values["kwargs"] is not None:
+                    model = load_chat_model(select_model_name, **model_values["kwargs"])
+                else:
+                    model = load_chat_model(select_model_name)
             override_kwargs["model"] = model
             if model_values["tools"] is not None:
                 override_kwargs["tools"] = model_values["tools"]

langchain_dev_utils-1.2.15.dist-info/METADATA

@@ -0,0 +1,102 @@
+Metadata-Version: 2.4
+Name: langchain-dev-utils
+Version: 1.2.15
+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
+Project-URL: documentation, https://tbice123123.github.io/langchain-dev-utils
+Author-email: tiebingice <tiebingice123@outlook.com>
+License-File: LICENSE
+Requires-Python: >=3.11
+Requires-Dist: langchain>=1.1.0
+Requires-Dist: langgraph>=1.0.0
+Provides-Extra: standard
+Requires-Dist: json-repair>=0.53.1; extra == 'standard'
+Requires-Dist: langchain-openai; extra == 'standard'
+Description-Content-Type: text/markdown
+
+# 🦜️🧰 langchain-dev-utils
+
+<p align="center">
+    <em>🚀 High-efficiency toolkit designed for LangChain and LangGraph developers</em>
+</p>
+
+<p align="center">
+  📚 <a href="https://tbice123123.github.io/langchain-dev-utils/">English</a> • 
+  <a href="https://tbice123123.github.io/langchain-dev-utils/zh/">中文</a>
+</p>
+
+[![PyPI](https://img.shields.io/pypi/v/langchain-dev-utils.svg?color=%2334D058&label=pypi%20package)](https://pypi.org/project/langchain-dev-utils/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![Python](https://img.shields.io/badge/python-3.11|3.12|3.13|3.14-%2334D058)](https://www.python.org/downloads)
+[![Downloads](https://static.pepy.tech/badge/langchain-dev-utils/month)](https://pepy.tech/project/langchain-dev-utils)
+[![Documentation](https://img.shields.io/badge/docs-latest-blue)](https://tbice123123.github.io/langchain-dev-utils)
+
+> This is the English version. For the Chinese version, please visit [中文版本](https://github.com/TBice123123/langchain-dev-utils/blob/master/README_cn.md)
+
+## ✨ Why choose langchain-dev-utils?
+
+Tired of writing repetitive code in LangChain development? `langchain-dev-utils` is the solution you need! This lightweight yet powerful toolkit is designed to enhance the development experience of LangChain and LangGraph, helping you:
+
+- ⚡ **Boost development efficiency** - Reduce boilerplate code, allowing you to focus on core functionality
+- 🧩 **Simplify complex workflows** - Easily manage multi-model, multi-tool, and multi-agent applications
+- 🔧 **Enhance code quality** - Improve consistency and readability, reducing maintenance costs
+- 🎯 **Accelerate prototype development** - Quickly implement ideas, iterate and validate faster
+
+
+## 🎯 Core Features
+
+- **🔌 Unified model management** - Specify model providers through strings, easily switch and combine different models
+- **💬 Flexible message handling** - Support for chain-of-thought concatenation, streaming processing, and message formatting
+- **🛠️ Powerful tool calling** - Built-in tool call detection, parameter parsing, and human review functionality
+- **🤖 Efficient Agent development** - Simplify agent creation process, expand more common middleware
+- **📊 Flexible state graph composition** - Support for serial and parallel composition of multiple StateGraphs
+
+## ⚡ Quick Start
+
+**1. Install `langchain-dev-utils`**
+
+```bash
+pip install -U "langchain-dev-utils[standard]"
+```
+
+**2. Start using**
+
+```python
+from langchain.tools import tool
+from langchain_core.messages import HumanMessage
+from langchain_dev_utils.chat_models import register_model_provider, load_chat_model
+from langchain_dev_utils.agents import create_agent
+
+# Register model provider
+register_model_provider("vllm", "openai-compatible", base_url="http://localhost:8000/v1")
+
+@tool
+def get_current_weather(location: str) -> str:
+    """Get the current weather for the specified location"""
+    return f"25 degrees, {location}"
+
+# Dynamically load model using string
+model = load_chat_model("vllm:qwen3-4b")
+response = model.invoke("Hello")
+print(response)
+
+# Create agent
+agent = create_agent("vllm:qwen3-4b", tools=[get_current_weather])
+response = agent.invoke({"messages": [HumanMessage(content="What's the weather like in New York today?")]})
+print(response)
+```
+
+**For more features of this library, please visit the [full documentation](https://tbice123123.github.io/langchain-dev-utils/)**
+
+
+## 🛠️ GitHub Repository
+
+Visit the [GitHub repository](https://github.com/TBice123123/langchain-dev-utils) to view the source code and report issues.
+
+---
+
+<div align="center">
+  <p>Developed with ❤️ and ☕</p>
+  <p>If this project helps you, please give us a ⭐️</p>
+</div>

{langchain_dev_utils-1.2.13.dist-info → langchain_dev_utils-1.2.15.dist-info}/RECORD

@@ -1,4 +1,4 @@
-langchain_dev_utils/__init__.py,sha256=rQSlPcfj4yT4krIq6epTVQyBzIX4etVOgfupVkM-RnU,23
+langchain_dev_utils/__init__.py,sha256=Q6rDLuL8XHKQggYBtRCtxzpPQJgFYWn4x0gcVlH7H4g,23
 langchain_dev_utils/_utils.py,sha256=MFEzR1BjXMj6HEVwt2x2omttFuDJ_rYAEbNqe99r9pM,1338
 langchain_dev_utils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langchain_dev_utils/agents/__init__.py,sha256=PJ-lSDZv_AXMYA3H4fx-HzJa14tPbkGmq1HX8LNfaPo,125
@@ -9,7 +9,7 @@ langchain_dev_utils/agents/wrap.py,sha256=RuchoH_VotPmKFuYEn2SXoSgNxZhSA9jKM0Iv_
 langchain_dev_utils/agents/middleware/__init__.py,sha256=EECbcYcHXQAMA-guJNRGwCVi9jG957d0nOaoIuyIKC0,832
 langchain_dev_utils/agents/middleware/format_prompt.py,sha256=rfii98tmOqkjaNHxWy7hovhEYKXrF0CdzsMLO54_CDI,2359
 langchain_dev_utils/agents/middleware/model_fallback.py,sha256=nivtXXF4cwyOBv6p7RW12nXtNg87wjTWxO3BKIYiroI,1674
-langchain_dev_utils/agents/middleware/model_router.py,sha256=pOK-4PNTLrmjaQA9poHoQnsaVwoX0JeJrLVysulv9iU,7631
+langchain_dev_utils/agents/middleware/model_router.py,sha256=sBXp1D1u_mv9g_JN_RcwKp9DONEY3XMWQVb4nGHPgkw,7925
 langchain_dev_utils/agents/middleware/plan.py,sha256=0qDCmenxgY_zrwMfOyYlgLfhYNw-HszNLeeOkfj14NA,16002
 langchain_dev_utils/agents/middleware/summarization.py,sha256=IoZ2PM1OC3AXwf0DWpfreuPOAipeiYu0KPmAABWXuY0,3087
 langchain_dev_utils/agents/middleware/tool_call_repair.py,sha256=oZF0Oejemqs9kSn8xbW79FWyVVarL4IGCz0gpqYBkFM,3529
@@ -32,7 +32,7 @@ langchain_dev_utils/pipeline/types.py,sha256=T3aROKKXeWvd0jcH5XkgMDQfEkLfPaiOhhV
 langchain_dev_utils/tool_calling/__init__.py,sha256=mu_WxKMcu6RoTf4vkTPbA1WSBSNc6YIqyBtOQ6iVQj4,322
 langchain_dev_utils/tool_calling/human_in_the_loop.py,sha256=7Z_QO5OZUR6K8nLoIcafc6osnvX2IYNorOJcbx6bVso,9672
 langchain_dev_utils/tool_calling/utils.py,sha256=S4-KXQ8jWmpGTXYZitovF8rxKpaSSUkFruM8LDwvcvE,2765
-langchain_dev_utils-1.2.13.dist-info/METADATA,sha256=SUHFdkF4J5_VxYmZGL7Aqv_79JTXRBoCn--gMTrSjHw,11844
-langchain_dev_utils-1.2.13.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-langchain_dev_utils-1.2.13.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
-langchain_dev_utils-1.2.13.dist-info/RECORD,,
+langchain_dev_utils-1.2.15.dist-info/METADATA,sha256=f5J-j-1HdBe7bor7rXmgFWaLEXYXci0CpppO8hJaJ2U,4516
+langchain_dev_utils-1.2.15.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+langchain_dev_utils-1.2.15.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
+langchain_dev_utils-1.2.15.dist-info/RECORD,,

langchain_dev_utils-1.2.13.dist-info/METADATA

@@ -1,345 +0,0 @@
-Metadata-Version: 2.4
-Name: langchain-dev-utils
-Version: 1.2.13
-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
-Project-URL: documentation, https://tbice123123.github.io/langchain-dev-utils
-Author-email: tiebingice <tiebingice123@outlook.com>
-License-File: LICENSE
-Requires-Python: >=3.11
-Requires-Dist: langchain>=1.1.0
-Requires-Dist: langgraph>=1.0.0
-Provides-Extra: standard
-Requires-Dist: json-repair>=0.53.1; extra == 'standard'
-Requires-Dist: langchain-openai; extra == 'standard'
-Description-Content-Type: text/markdown
-
-# 🦜️🧰 langchain-dev-utils
-
-<p align="center">
-    <em>A utility library for LangChain and LangGraph development.</em>
-</p>
-
-<p align="center">
-  📚 <a href="https://tbice123123.github.io/langchain-dev-utils/">English</a> • 
-  <a href="https://tbice123123.github.io/langchain-dev-utils/zh/">中文</a>
-</p>
-
-[![PyPI](https://img.shields.io/pypi/v/langchain-dev-utils.svg?color=%2334D058&label=pypi%20package)](https://pypi.org/project/langchain-dev-utils/)
-[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
-[![Python](https://img.shields.io/badge/python-3.11|3.12|3.13|3.14-%2334D058)](https://www.python.org/downloads)
-[![Downloads](https://static.pepy.tech/badge/langchain-dev-utils/month)](https://pepy.tech/project/langchain-dev-utils)
-[![Documentation](https://img.shields.io/badge/docs-latest-blue)](https://tbice123123.github.io/langchain-dev-utils/)
-
-> This is the English version. For the Chinese version, please visit [中文文档](https://github.com/TBice123123/langchain-dev-utils/blob/master/README_cn.md)
-
-**langchain-dev-utils** is a utility library focused on enhancing the development experience of LangChain and LangGraph. It provides a series of ready-to-use utility functions that can reduce repetitive code writing and improve code consistency and readability. By simplifying the development workflow, this library can help you build prototypes faster, iterate more smoothly, and create clearer and more reliable AI applications based on large language models.
-
-## 🚀 Installation
-
-```bash
-pip install -U langchain-dev-utils
-
-# Install full-featured version:
-pip install -U langchain-dev-utils[standard]
-```
-
-## 📦 Core Features
-
-### 1. **Model Management**
-
-In `langchain`, the `init_chat_model`/`init_embeddings` functions can be used to initialize chat model instances/embedding model instances, but they support a limited number of model providers. This module provides registration functions (`register_model_provider`/`register_embeddings_provider`) to easily register any model provider for later use with `load_chat_model` / `load_embeddings` for model loading.
-
-#### 1.1 Chat Model Management
-
-There are two main functions:
-
-- `register_model_provider`: Register a chat model provider
-- `load_chat_model`: Load a chat model
-
-Assuming you want to use the qwen3-4b model deployed with `vllm`, the reference code is as follows:
-
-```python
-from langchain_dev_utils.chat_models import (
-    register_model_provider,
-    load_chat_model,
-)
-
-# Register model provider
-register_model_provider(
-    provider_name="vllm",
-    chat_model="openai-compatible",
-    base_url="http://localhost:8000/v1",
-)
-
-# Load model
-model = load_chat_model("vllm:qwen3-4b")
-print(model.invoke("Hello"))
-```
-
-#### 1.2 Embedding Model Management
-
-There are two main functions:
-
-- `register_embeddings_provider`: Register an embedding model provider
-- `load_embeddings`: Load an embedding model
-
-Assuming you want to use the qwen3-embedding-4b model deployed with `vllm`, the reference code is as follows:
-
-```python
-from langchain_dev_utils.embeddings import register_embeddings_provider, load_embeddings
-
-# Register embedding model provider
-register_embeddings_provider(
-    provider_name="vllm",
-    embeddings_model="openai-compatible",
-    base_url="http://localhost:8000/v1",
-)
-
-# Load embedding model
-embeddings = load_embeddings("vllm:qwen3-embedding-4b")
-emb = embeddings.embed_query("Hello")
-print(emb)
-```
-
-
-### 2. **Message Conversion**
-
-Includes the following features:
-
-- Merge chain-of-thought content into final responses
-- Stream content merging
-- Content formatting tools
-
-#### 2.1 Stream Content Merging
-
-For streaming responses obtained using `stream()` and `astream()`, you can use `merge_ai_message_chunk` to merge them into a final AIMessage.
-
-```python
-from langchain_dev_utils.message_convert import merge_ai_message_chunk
-chunks = list(model.stream("Hello"))
-merged = merge_ai_message_chunk(chunks)
-```
-
-#### 2.2 Format List Content
-
-For a list, you can use `format_sequence` to format it.
-
-```python
-from langchain_dev_utils.message_convert import format_sequence
-text = format_sequence([
-    "str1",
-    "str2",
-    "str3"
-], separator="\n", with_num=True)
-```
-
-
-### 3. **Tool Calling**
-
-Includes the following features:
-
-- Check and parse tool calls
-- Add human-in-the-loop functionality
-
-#### 3.1 Check and Parse Tool Calls
-
-`has_tool_calling` and `parse_tool_calling` are used to check and parse tool calls.
-
-```python
-import datetime
-from langchain_core.tools import tool
-from langchain_dev_utils.tool_calling import has_tool_calling, parse_tool_calling
-
-@tool
-def get_current_time() -> str:
-    """Get current timestamp"""
-    return str(datetime.datetime.now().timestamp())
-
-response = model.bind_tools([get_current_time]).invoke("What time is it now?")
-
-if has_tool_calling(response):
-    name, args = parse_tool_calling(
-        response, first_tool_call_only=True
-    )
-    print(name, args)
-```
-
-#### 3.2 Add Human-in-the-Loop Functionality
-
-- `human_in_the_loop`: For synchronous tool functions
-- `human_in_the_loop_async`: For asynchronous tool functions
-
-Both can accept a `handler` parameter for custom breakpoint return and response handling logic.
-
-```python
-from langchain_dev_utils.tool_calling import human_in_the_loop
-from langchain_core.tools import tool
-import datetime
-
-@human_in_the_loop
-@tool
-def get_current_time() -> str:
-    """Get current timestamp"""
-    return str(datetime.datetime.now().timestamp())
-```
-
-
-### 4. **Agent Development**
-
-Includes the following features:
-
-- Multi-agent construction
-- Common middleware components
-
-#### 4.1 Multi-Agent Construction
-
-Wrapping agents as tools is a common implementation pattern in multi-agent systems, which is detailed in the official LangChain documentation. To this end, this library provides a pre-built function `wrap_agent_as_tool` to implement this pattern, which can wrap an agent instance into a tool that can be called by other agents.
-
-Usage example:
-
-```python
-import datetime
-from langchain_dev_utils.agents import create_agent, wrap_agent_as_tool
-from langchain.agents import AgentState
-
-@tool
-def get_current_time() -> str:
-    """Get current time"""
-    return datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-
-time_agent = create_agent("vllm:qwen3-4b", tools=[get_current_time], name="time-agent")
-call_time_agent_tool = wrap_agent_as_tool(time_agent)  
-
-agent = create_agent(
-    "vllm:qwen3-4b",
-    name="agent",
-    tools=[call_time_agent_tool],
-)
-response = agent.invoke(
-    {"messages": [{"role": "user", "content": "What time is it now?"}]}
-)
-print(response)
-```
-
-#### 4.2 Middleware
-
-Provides some common middleware components. Below are examples using `ToolCallRepairMiddleware` and `PlanMiddleware`.
-
-`ToolCallRepairMiddleware` is used to fix `invaild_tool_calls` content from large models.
-
-`PlanMiddleware` is used for agent planning.
-
-```python
-from langchain_dev_utils.agents.middleware import (
-    ToolCallRepairMiddleware,
-    PlanMiddleware,
-)
-
-agent=create_agent(
-    "vllm:qwen3-4b",
-    name="plan-agent",
-    middleware=[ToolCallRepairMiddleware(), PlanMiddleware(
-        use_read_plan_tool=False
-    )]
-)
-response = agent.invoke({"messages": [{"role": "user", "content": "Give me a travel plan to New York"}]}))
-print(response)
-```
-
-
-### 5. **State Graph Orchestration**
-
-Includes the following features:
-
-- Sequential graph orchestration
-- Parallel graph orchestration
-
-#### 5.1 Sequential Graph Orchestration
-
-Using `create_sequential_pipeline`, you can orchestrate multiple subgraphs in sequence:
-
-```python
-from langchain.agents import AgentState
-from langchain_core.messages import HumanMessage
-from langchain_dev_utils.agents import create_agent
-from langchain_dev_utils.pipeline import create_sequential_pipeline
-from langchain_dev_utils.chat_models import register_model_provider
-
-register_model_provider(
-    provider_name="vllm",
-    chat_model="openai-compatible",
-    base_url="http://localhost:8000/v1",
-)
-
-# Build sequential pipeline (all subgraphs execute in sequence)
-graph = create_sequential_pipeline(
-    sub_graphs=[
-        create_agent(
-            model="vllm:qwen3-4b",
-            tools=[get_current_time],
-            system_prompt="You are a time query assistant, you can only answer the current time. If this question is not related to time, please directly answer that you cannot answer",
-            name="time_agent",
-        ),
-        create_agent(
-            model="vllm:qwen3-4b",
-            tools=[get_current_weather],
-            system_prompt="You are a weather query assistant, you can only answer the current weather. If this question is not related to weather, please directly answer that you cannot answer",
-            name="weather_agent",
-        ),
-        create_agent(
-            model="vllm:qwen3-4b",
-            tools=[get_current_user],
-            system_prompt="You are a user query assistant, you can only answer the current user. If this question is not related to users, please directly answer that you cannot answer",
-            name="user_agent",
-        ),
-    ],
-    state_schema=AgentState,
-)
-
-response = graph.invoke({"messages": [HumanMessage("Hello")]})
-print(response)
-```
-
-#### 5.2 Parallel Graph Orchestration
-
-Using `create_parallel_pipeline`, you can orchestrate multiple subgraphs in parallel:
-
-```python
-from langchain_dev_utils.pipeline import create_parallel_pipeline
-
-# Build parallel pipeline (all subgraphs execute in parallel)
-graph = create_parallel_pipeline(
-    sub_graphs=[
-        create_agent(
-            model="vllm:qwen3-4b",
-            tools=[get_current_time],
-            system_prompt="You are a time query assistant, you can only answer the current time. If this question is not related to time, please directly answer that you cannot answer",
-            name="time_agent",
-        ),
-        create_agent(
-            model="vllm:qwen3-4b",
-            tools=[get_current_weather],
-            system_prompt="You are a weather query assistant, you can only answer the current weather. If this question is not related to weather, please directly answer that you cannot answer",
-            name="weather_agent",
-        ),
-        create_agent(
-            model="vllm:qwen3-4b",
-            tools=[get_current_user],
-            system_prompt="You are a user query assistant, you can only answer the current user. If this question is not related to users, please directly answer that you cannot answer",
-            name="user_agent",
-        ),
-    ],
-    state_schema=AgentState,
-)
-response = graph.invoke({"messages": [HumanMessage("Hello")]})
-print(response)
-```
-
-
-## 💬 Join the Community
-
-- [GitHub Repository](https://github.com/TBice123123/langchain-dev-utils) — Browse source code, submit Pull Requests
-- [Issue Tracker](https://github.com/TBice123123/langchain-dev-utils/issues) — Report bugs or suggest improvements
-- We welcome all forms of contributions — whether code, documentation, or usage examples. Let's build a more powerful and practical LangChain development ecosystem together