langchain-dev-utils 1.2.14__py3-none-any.whl → 1.2.16__py3-none-any.whl

langchain_dev_utils/__init__.py

@@ -1 +1 @@
-__version__ = "1.2.14"
+__version__ = "1.2.16"

langchain_dev_utils/agents/factory.py

@@ -7,6 +7,7 @@ from langchain.agents.middleware.types import (
     ResponseT,
     _InputAgentState,
     _OutputAgentState,
+    StateT_co,
 )
 from langchain.agents.structured_output import ResponseFormat
 from langchain_core.messages import SystemMessage
@@ -25,8 +26,8 @@ def create_agent(  # noqa: PLR0915
     tools: Sequence[BaseTool | Callable | dict[str, Any]] | None = None,
     *,
     system_prompt: str | SystemMessage | None = None,
-    middleware: Sequence[AgentMiddleware[AgentState[ResponseT], ContextT]] = (),
     response_format: ResponseFormat[ResponseT] | type[ResponseT] | None = None,
+    middleware: Sequence[AgentMiddleware[StateT_co, ContextT]] = (),
     state_schema: type[AgentState[ResponseT]] | None = None,
     context_schema: type[ContextT] | None = None,
     checkpointer: Checkpointer | None = None,

langchain_dev_utils/agents/middleware/format_prompt.py

@@ -32,7 +32,7 @@ def format_prompt(request: ModelRequest) -> str:
         ...     {
         ...         "messages": [HumanMessage(content="Hello")],
         ...     },
-        ...     context=Context(name="assistant", user="tbice"),
+        ...     context=Context(name="assistant", user="Tom"),
         ... )
 
     """

langchain_dev_utils/agents/middleware/model_router.py

@@ -1,8 +1,8 @@
-from typing import Any, Awaitable, Callable, NotRequired, Optional, cast
+from typing import Annotated, Any, Awaitable, Callable, NotRequired, Optional, cast
 
 from langchain.agents import AgentState
 from langchain.agents.middleware import AgentMiddleware, ModelRequest, ModelResponse
-from langchain.agents.middleware.types import ModelCallResult
+from langchain.agents.middleware.types import ModelCallResult, OmitFromInput
 from langchain_core.language_models import BaseChatModel
 from langchain_core.messages import AnyMessage, SystemMessage
 from langchain_core.tools import BaseTool
@@ -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]
 
 
@@ -64,7 +65,7 @@ Strictly adhere to tool call requirements!
 
 
 class ModelRouterState(AgentState):
-    router_model_selection: str
+    router_model_selection: Annotated[str, OmitFromInput]
 
 
 class ModelRouterMiddleware(AgentMiddleware):
@@ -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.dist-info → langchain_dev_utils-1.2.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-dev-utils
-Version: 1.2.14
+Version: 1.2.16
 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.dist-info → langchain_dev_utils-1.2.16.dist-info}/RECORD

@@ -1,15 +1,15 @@
-langchain_dev_utils/__init__.py,sha256=mxzjGyB-ihJR05pHYcBiUe_XT5X9wj6cBHLxOXBhAeM,23
+langchain_dev_utils/__init__.py,sha256=7RIJ-Nh9kHJf5O5NvahUeP8DNXq6oIzbYcIt_yKv0lQ,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
-langchain_dev_utils/agents/factory.py,sha256=XdGjktksfTDys7X4SgfPrQz10HUo5fTNAWESDQenIlE,3728
+langchain_dev_utils/agents/factory.py,sha256=mr-Q3fxxV_yEN3l8p-Rk0uxXvKjDUGEZDtrg9HRwr-8,3732
 langchain_dev_utils/agents/file_system.py,sha256=Yk3eetREE26WNrnTWLoiDUpOyCJ-rhjlfFDk6foLa1E,8468
 langchain_dev_utils/agents/plan.py,sha256=WwhoiJBmVYVI9bT8HfjCzTJ_SIp9WFil0gOeznv2omQ,6497
 langchain_dev_utils/agents/wrap.py,sha256=RuchoH_VotPmKFuYEn2SXoSgNxZhSA9jKM0Iv_8oHLk,4718
 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/format_prompt.py,sha256=LzYiQXCRvkpfDhGPxhZwdepeU3j-HUWPJqcx3FWsfT4,2357
 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=qBspvj9ZoKfmC1pHWTO0EHHfxjgCUd-TuSbqvZl0kmg,7977
 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.14.dist-info/METADATA,sha256=QDvbo10fXRhdu1AP-afZ3aa9Tl39fKW1mMI06hY4GVI,4594
-langchain_dev_utils-1.2.14.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-langchain_dev_utils-1.2.14.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
-langchain_dev_utils-1.2.14.dist-info/RECORD,,
+langchain_dev_utils-1.2.16.dist-info/METADATA,sha256=2lobuT6h5sJqxw25CaeDtdEQrk2EvROnPLhuF-ny4mI,4516
+langchain_dev_utils-1.2.16.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+langchain_dev_utils-1.2.16.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
+langchain_dev_utils-1.2.16.dist-info/RECORD,,