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

langchain_dev_utils-1.2.14/PKG-INFO

@@ -0,0 +1,102 @@
+Metadata-Version: 2.4
+Name: langchain-dev-utils
+Version: 1.2.14
+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[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)
+   ```
+
+**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.14/README.md

@@ -0,0 +1,85 @@
+# 🦜️🧰 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[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)
+   ```
+
+**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.14/README_cn.md

@@ -0,0 +1,86 @@
+# 🦜️🧰 langchain-dev-utils
+
+<p align="center">
+    <em>🚀 专为 LangChain 和 LangGraph 开发者打造的高效工具库</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/zh/)
+
+> 当前为中文版，英文版请访问[English Version](https://github.com/TBice123123/langchain-dev-utils/blob/master/README.md)
+
+## ✨ 为什么选择 langchain-dev-utils？
+
+厌倦了在 LangChain 开发中编写重复代码？`langchain-dev-utils` 正是您需要的解决方案！这个轻量但功能强大的工具库专为提升 LangChain 和 LangGraph 开发体验而设计，帮助您：
+
+- ⚡ **提升开发效率** - 减少样板代码，让您专注于核心功能
+- 🧩 **简化复杂流程** - 轻松管理多模型、多工具和多智能体应用
+- 🔧 **增强代码质量** - 提高一致性和可读性，减少维护成本
+- 🎯 **加速原型开发** - 快速实现想法，更快迭代验证
+
+
+## 🎯 核心功能
+
+- **🔌 统一的模型管理** - 通过字符串指定模型提供商，轻松切换和组合不同模型
+- **💬 灵活的消息处理** - 支持思维链拼接、流式处理和消息格式化
+- **🛠️ 强大的工具调用** - 内置工具调用检测、参数解析和人工审核功能
+- **🤖 高效的 Agent 开发** - 简化智能体创建流程，扩充更多的常用中间件
+- **📊 灵活的状态图组合** - 支持串行和并行方式组合多个 StateGraph
+
+## ⚡ 快速开始
+
+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)
+   ```
+
+**获取更多的本库功能，请访问[完整文档](https://tbice123123.github.io/langchain-dev-utils/zh/)**
+
+
+## 🛠️ GitHub 仓库
+
+访问 [GitHub 仓库](https://github.com/TBice123123/langchain-dev-utils) 查看源代码和问题。
+
+
+---
+
+<div align="center">
+  <p>由 ❤️ 和 ☕ 驱动开发</p>
+  <p>如果这个项目对您有帮助，请给我们一个 ⭐️</p>
+</div>

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

@@ -1,6 +1,6 @@
 [project]
 name = "langchain-dev-utils"
-version = "1.2.12"
+version = "1.2.14"
 description = "A practical utility library for LangChain and LangGraph development"
 readme = "README.md"
 authors = [{ name = "tiebingice", email = "tiebingice123@outlook.com" }]
@@ -10,7 +10,7 @@ dependencies = ["langchain>=1.1.0", "langgraph>=1.0.0"]
 [project.urls]
 "Source Code" = "https://github.com/TBice123123/langchain-dev-utils"
 repository = "https://github.com/TBice123123/langchain-dev-utils"
-documentation = "https://tbice123123.github.io/langchain-dev-utils-docs/en/"
+documentation = "https://tbice123123.github.io/langchain-dev-utils"
 
 
 [project.optional-dependencies]
@@ -20,9 +20,6 @@ standard = ["json-repair>=0.53.1", "langchain-openai"]
 requires = ["hatchling"]
 build-backend = "hatchling.build"
 
-[tool.uv.build-backend]
-source-exclude = ["/data", "/docs", "mkdocs.yml"]
-
 [tool.hatch.build]
 exclude = ["/data", "/docs", "mkdocs.yml"]
 

langchain_dev_utils-1.2.14/src/langchain_dev_utils/__init__.py

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

{langchain_dev_utils-1.2.12 → langchain_dev_utils-1.2.14}/src/langchain_dev_utils/chat_models/adapters/openai_compatible.py

@@ -20,7 +20,13 @@ from langchain_core.callbacks import (
     CallbackManagerForLLMRun,
 )
 from langchain_core.language_models import LangSmithParams, LanguageModelInput
-from langchain_core.messages import AIMessage, AIMessageChunk, BaseMessage, HumanMessage
+from langchain_core.messages import (
+    AIMessage,
+    AIMessageChunk,
+    BaseMessage,
+    HumanMessage,
+    ToolMessage,
+)
 from langchain_core.outputs import ChatGenerationChunk, ChatResult
 from langchain_core.runnables import Runnable
 from langchain_core.tools import BaseTool
@@ -63,6 +69,58 @@ def _get_last_human_message_index(messages: list[BaseMessage]) -> int:
     )
 
 
+def _transform_video_block(block: dict[str, Any]) -> dict:
+    """Transform video block to video_url block."""
+    if "url" in block:
+        return {
+            "type": "video_url",
+            "video_url": {
+                "url": block["url"],
+            },
+        }
+    if "base64" in block or block.get("source_type") == "base64":
+        if "mime_type" not in block:
+            error_message = "mime_type key is required for base64 data."
+            raise ValueError(error_message)
+        mime_type = block["mime_type"]
+        base64_data = block["data"] if "data" in block else block["base64"]
+        return {
+            "type": "video_url",
+            "video_url": {
+                "url": f"data:{mime_type};base64,{base64_data}",
+            },
+        }
+    error_message = "Unsupported source type. Only 'url' and 'base64' are supported."
+    raise ValueError(error_message)
+
+
+def _process_video_input(message: BaseMessage):
+    """
+    Process BaseMessage with video input.
+
+    Args:
+        message (BaseMessage): The HumanMessage instance to process.
+
+    Returns:
+        None: The method modifies the message in-place.
+    """
+    if not message.content:
+        return message
+    content = message.content
+
+    if not isinstance(content, list):
+        return message
+
+    formatted_content = []
+    for block in content:
+        if isinstance(block, dict) and block.get("type") == "video":
+            formatted_content.append(_transform_video_block(block))
+        else:
+            formatted_content.append(block)
+    message = message.model_copy(update={"content": formatted_content})
+    return message
+
+
 class _BaseChatOpenAICompatible(BaseChatOpenAI):
     """
     Base template class for OpenAI-compatible chat model implementations.
@@ -183,6 +241,10 @@ class _BaseChatOpenAICompatible(BaseChatOpenAI):
                     )
                 payload_messages.append(msg_dict)
             else:
+                if (
+                    isinstance(m, HumanMessage) or isinstance(m, ToolMessage)
+                ) and isinstance(m.content, list):
+                    m = _process_video_input(m)
                 payload_messages.append(_convert_message_to_dict(m))
 
         payload["messages"] = payload_messages

{langchain_dev_utils-1.2.12 → langchain_dev_utils-1.2.14}/src/langchain_dev_utils/chat_models/base.py

@@ -11,7 +11,6 @@ from langchain_dev_utils._utils import (
 
 from .types import ChatModelProvider, ChatModelType, CompatibilityOptions
 
-
 _MODEL_PROVIDERS_DICT = {}
 
 

langchain_dev_utils-1.2.14/tests/test_chat_models.py

@@ -0,0 +1,185 @@
+import base64
+from typing import Any, cast
+
+import httpx
+import pytest
+from langchain_core.language_models import BaseChatModel
+from langchain_core.messages import AIMessage, HumanMessage, ToolMessage
+from langchain_tests.integration_tests.chat_models import ChatModelIntegrationTests
+
+from langchain_dev_utils.chat_models.base import load_chat_model
+
+
+class TestStandard(ChatModelIntegrationTests):
+    @pytest.fixture
+    def model(self, request: Any) -> BaseChatModel:
+        """Model fixture."""
+        extra_init_params = getattr(request, "param", None) or {}
+        if extra_init_params.get("output_version") == "v1":
+            pytest.skip("Output version v1 is not supported")
+        return self.chat_model_class(
+            **{
+                **self.standard_chat_model_params,
+                **self.chat_model_params,
+                **extra_init_params,
+            },
+        )
+
+    @property
+    def chat_model_class(self) -> type[BaseChatModel]:
+        return cast("type[BaseChatModel]", load_chat_model)
+
+    @property
+    def chat_model_params(self) -> dict:
+        return {
+            "model": "zai:glm-4.5",
+            "extra_body": {
+                "thinking": {
+                    "type": "disabled",
+                }
+            },
+        }
+
+    @property
+    def has_tool_calling(self) -> bool:
+        return True
+
+    @property
+    def has_structured_output(self) -> bool:
+        return True
+
+    @property
+    def has_tool_choice(self) -> bool:
+        return False
+
+    @property
+    def supports_image_tool_message(self) -> bool:
+        return False
+
+    @property
+    def supports_json_mode(self) -> bool:
+        """(bool) whether the chat model supports JSON mode."""
+        return False
+
+
+class TestImageProcessing:
+    @pytest.fixture(autouse=True)
+    def setup(self) -> None:
+        self.model = load_chat_model("zai:glm-4.6v")
+        self.image_url = "https://cloudcovert-1305175928.cos.ap-guangzhou.myqcloud.com/%E5%9B%BE%E7%89%87grounding.PNG"
+        self.image_data = base64.b64encode(httpx.get(self.image_url).content).decode(
+            "utf-8"
+        )
+
+    def test_image_url_format(self) -> None:
+        message = HumanMessage(
+            content=[
+                {"type": "text", "text": "Give a concise description of this image."},
+                {
+                    "type": "image_url",
+                    "image_url": {"url": f"data:image/png;base64,{self.image_data}"},
+                },
+            ],
+        )
+        _ = self.model.invoke([message])
+
+    def test_image_base64_format(self) -> None:
+        message = HumanMessage(
+            content=[
+                {"type": "text", "text": "Give a concise description of this image."},
+                {
+                    "type": "image",
+                    "base64": self.image_data,
+                    "mime_type": "image/png",
+                },
+            ],
+        )
+        _ = self.model.invoke([message])
+
+    def test_image_url_direct_format(self) -> None:
+        message = HumanMessage(
+            content=[
+                {
+                    "type": "text",
+                    "text": "Give a concise description of this image.  answer in Chinese",
+                },
+                {
+                    "type": "image",
+                    "url": self.image_url,
+                },
+            ],
+        )
+        _ = self.model.invoke([message])
+
+    def test_oai_format_tool_message(self) -> None:
+        oai_format_message = ToolMessage(
+            content=[
+                {
+                    "type": "image_url",
+                    "image_url": {"url": f"data:image/png;base64,{self.image_data}"},
+                },
+            ],
+            tool_call_id="1",
+            name="random_image",
+        )
+
+        messages = [
+            HumanMessage(
+                "get a random diagram using the tool and give it a concise description，answer in Chinese"
+            ),
+            AIMessage(
+                [],
+                tool_calls=[
+                    {
+                        "type": "tool_call",
+                        "id": "1",
+                        "name": "random_image",
+                        "args": {},
+                    }
+                ],
+            ),
+            oai_format_message,
+        ]
+
+        def random_image() -> str:
+            """Return a random image."""
+            return ""
+
+        _ = self.model.bind_tools([random_image]).invoke(messages)
+
+    def test_standard_format_tool_message(self) -> None:
+        standard_format_message = ToolMessage(
+            content=[
+                {
+                    "type": "image",
+                    "base64": self.image_data,
+                    "mime_type": "image/png",
+                },
+            ],
+            tool_call_id="1",
+            name="random_image",
+        )
+
+        messages = [
+            HumanMessage(
+                "get a random diagram using the tool and give it a concise description，answer in Chinese"
+            ),
+            AIMessage(
+                [],
+                tool_calls=[
+                    {
+                        "type": "tool_call",
+                        "id": "1",
+                        "name": "random_image",
+                        "args": {},
+                    }
+                ],
+            ),
+            standard_format_message,
+        ]
+
+        def random_image() -> str:
+            """Return a random image."""
+            return ""
+
+        _ = self.model.bind_tools([random_image]).invoke(messages)