langchain-dev-utils 1.2.14__tar.gz → 1.2.15__tar.gz

{langchain_dev_utils-1.2.14 → langchain_dev_utils-1.2.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-dev-utils
-Version: 1.2.14
+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
@@ -54,38 +54,38 @@ Tired of writing repetitive code in LangChain development? `langchain-dev-utils`
 
 ## ⚡ Quick Start
 
-1. Install `langchain-dev-utils`:
+**1. Install `langchain-dev-utils`**
 
-   ```bash
-   pip install -U langchain[openai] langchain-dev-utils
-   ```
+```bash
+pip install -U "langchain-dev-utils[standard]"
+```
 
-2. Start using:
+**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
+```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", "http://localhost:8000/v1")
+# 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}"
+@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)
+# 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)
-   ```
+# 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/)**
 

{langchain_dev_utils-1.2.14 → langchain_dev_utils-1.2.15}/README.md

@@ -37,38 +37,38 @@ Tired of writing repetitive code in LangChain development? `langchain-dev-utils`
 
 ## ⚡ Quick Start
 
-1. Install `langchain-dev-utils`:
-
-   ```bash
-   pip install -U langchain[openai] langchain-dev-utils
-   ```
-
-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", "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)
-   ```
+**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/)**
 

{langchain_dev_utils-1.2.14 → langchain_dev_utils-1.2.15}/README_cn.md

@@ -37,38 +37,38 @@
 
 ## ⚡ 快速开始
 
-1. 安装 `langchain-dev-utils`：
-
-   ```bash
-   pip install -U langchain[openai] langchain-dev-utils
-   ```
-
-2. 开始使用：
-
-   ```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("vllm", "openai-compatible", "http://localhost:8000/v1")
-
-    @tool
-    def get_current_weather(location: str) -> str:
-        """获取指定地点的当前天气"""
-        return f"25度，{location}"
-
-    # 使用字符串动态加载模型
-    model = load_chat_model("vllm:qwen3-4b")
-    response = model.invoke("你好")
-    print(response)
-
-    # 创建智能体
-    agent = create_agent("vllm:qwen3-4b", tools=[get_current_weather])
-    response = agent.invoke({"messages": [HumanMessage(content="今天纽约的天气如何？")]})
-    print(response)
-   ```
+**1. 安装 `langchain-dev-utils`**
+
+```bash
+pip install -U "langchain-dev-utils[standard]"
+```
+
+**2. 开始使用**
+
+```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("vllm", "openai-compatible", base_url="http://localhost:8000/v1")
+
+@tool
+def get_current_weather(location: str) -> str:
+    """获取指定地点的当前天气"""
+    return f"25度，{location}"
+
+# 使用字符串动态加载模型
+model = load_chat_model("vllm:qwen3-4b")
+response = model.invoke("你好")
+print(response)
+
+# 创建智能体
+agent = create_agent("vllm:qwen3-4b", tools=[get_current_weather])
+response = agent.invoke({"messages": [HumanMessage(content="今天纽约的天气如何？")]})
+print(response)
+```
 
 **获取更多的本库功能，请访问[完整文档](https://tbice123123.github.io/langchain-dev-utils/zh/)**
 

{langchain_dev_utils-1.2.14 → langchain_dev_utils-1.2.15}/pyproject.toml

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

langchain_dev_utils-1.2.15/src/langchain_dev_utils/__init__.py

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

{langchain_dev_utils-1.2.14 → langchain_dev_utils-1.2.15}/src/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.14/src/langchain_dev_utils/__init__.py

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