langchain-dev-utils 1.1.13__tar.gz → 1.1.15__tar.gz

{langchain_dev_utils-1.1.13 → langchain_dev_utils-1.1.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-dev-utils
-Version: 1.1.13
+Version: 1.1.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

{langchain_dev_utils-1.1.13 → langchain_dev_utils-1.1.15}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "langchain-dev-utils"
-version = "1.1.13"
+version = "1.1.15"
 description = "A practical utility library for LangChain and LangGraph development"
 readme = "README.md"
 authors = [{ name = "tiebingice", email = "tiebingice123@outlook.com" }]
@@ -28,9 +28,11 @@ python_files = ["test_*.py"]
 python_functions = ["test_*"]
 
 [dependency-groups]
+dev = ["ruff>=0.14.5"]
 tests = [
     "python-dotenv>=1.1.1",
     "langchain-tests>=1.0.0",
     "langchain-deepseek>=1.0.0",
     "langchain-qwq>=0.3.0",
+    "langchain-ollama>=1.0.0",
 ]

langchain_dev_utils-1.1.15/src/langchain_dev_utils/__init__.py

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

{langchain_dev_utils-1.1.13 → langchain_dev_utils-1.1.15}/src/langchain_dev_utils/agents/factory.py

@@ -1,9 +1,14 @@
 from typing import Any, Sequence
-from typing import Callable, cast
+from typing import Callable
 
-from langchain.agents import AgentState, create_agent as _create_agent
-from langchain.agents.factory import ResponseT
-from langchain.agents.middleware import AgentMiddleware
+from langchain.agents import create_agent as _create_agent
+from langchain.agents.middleware.types import (
+    AgentMiddleware,
+    AgentState,
+    ResponseT,
+    _InputAgentState,
+    _OutputAgentState,
+)
 from langchain.agents.structured_output import ResponseFormat
 from langchain_core.tools import BaseTool
 from langgraph.cache.base import BaseCache
@@ -31,7 +36,9 @@ def create_agent(  # noqa: PLR0915
     debug: bool = False,
     name: str | None = None,
     cache: BaseCache | None = None,
-) -> CompiledStateGraph:
+) -> CompiledStateGraph[
+    AgentState[ResponseT], ContextT, _InputAgentState, _OutputAgentState[ResponseT]
+]:
     """
     Create a prebuilt agent with string-based model specification.
 
@@ -80,22 +87,19 @@ def create_agent(  # noqa: PLR0915
         ... })
         >>> response
     """
-    return cast(
-        CompiledStateGraph,
-        _create_agent(
-            model=load_chat_model(model),
-            tools=tools,
-            system_prompt=system_prompt,
-            middleware=middleware,
-            response_format=response_format,
-            state_schema=state_schema,
-            context_schema=context_schema,
-            checkpointer=checkpointer,
-            store=store,
-            interrupt_before=interrupt_before,
-            interrupt_after=interrupt_after,
-            debug=debug,
-            name=name,
-            cache=cache,
-        ),
+    return _create_agent(
+        model=load_chat_model(model),
+        tools=tools,
+        system_prompt=system_prompt,
+        middleware=middleware,
+        response_format=response_format,
+        state_schema=state_schema,
+        context_schema=context_schema,
+        checkpointer=checkpointer,
+        store=store,
+        interrupt_before=interrupt_before,
+        interrupt_after=interrupt_after,
+        debug=debug,
+        name=name,
+        cache=cache,
     )

{langchain_dev_utils-1.1.13 → langchain_dev_utils-1.1.15}/src/langchain_dev_utils/agents/middleware/plan.py

@@ -287,6 +287,7 @@ class PlanMiddleware(AgentMiddleware):
             If not provided, uses the default `_DEFAULT_READ_PLAN_TOOL_DESCRIPTION`.
         use_read_plan_tool: Whether to use the `read_plan` tool.
             If not provided, uses the default `True`.
+        message_key: The key of the message to be updated. Defaults to "messages".
     Example:
         ```python
         from langchain_dev_utils.agents.middleware.plan import PlanMiddleware

{langchain_dev_utils-1.1.13 → langchain_dev_utils-1.1.15}/src/langchain_dev_utils/chat_models/base.py

@@ -3,9 +3,9 @@ from typing import Any, NotRequired, Optional, TypedDict, cast
 from langchain.chat_models.base import _SUPPORTED_PROVIDERS, _init_chat_model_helper
 from langchain_core.language_models.chat_models import BaseChatModel
 from langchain_core.utils import from_env
+from pydantic import BaseModel
 
 from .types import ChatModelType, ToolChoiceType
-from pydantic import BaseModel
 
 _MODEL_PROVIDERS_DICT = {}
 

{langchain_dev_utils-1.1.13 → langchain_dev_utils-1.1.15}/src/langchain_dev_utils/embeddings/base.py

@@ -223,6 +223,9 @@ def load_embeddings(
     ):
         raise ValueError(f"Provider {provider} not registered")
 
+    if provider in _SUPPORTED_PROVIDERS:
+        return init_embeddings(model, provider=provider, **kwargs)
+
     embeddings = _EMBEDDINGS_PROVIDERS_DICT[provider]["embeddings_model"]
     if isinstance(embeddings, str):
         if not (api_key := kwargs.get("api_key")):

{langchain_dev_utils-1.1.13 → langchain_dev_utils-1.1.15}/tests/test_agent.py

@@ -33,6 +33,7 @@ def test_prebuilt_agent():
     response = agent.invoke(
         {"messages": [HumanMessage("What's the weather in New York?")]}
     )
+
     assert len(response["messages"]) == 4
 
     assert response["messages"][1].tool_calls[0]["name"] == "get_current_weather"

{langchain_dev_utils-1.1.13 → langchain_dev_utils-1.1.15}/tests/test_load_embbeding.py

@@ -22,7 +22,13 @@ batch_register_embeddings_provider(
 )
 
 
-@pytest.fixture(params=["dashscope:text-embedding-v4", "siliconflow:BAAI/bge-m3"])
+@pytest.fixture(
+    params=[
+        "dashscope:text-embedding-v4",
+        "siliconflow:BAAI/bge-m3",
+        "ollama:bge-m3:latest",
+    ]
+)
 def embbeding_model(request: pytest.FixtureRequest) -> Embeddings:
     params = request.param
     return load_embeddings(params)

{langchain_dev_utils-1.1.13 → langchain_dev_utils-1.1.15}/uv.lock

@@ -418,7 +418,7 @@ wheels = [
 
 [[package]]
 name = "langchain-dev-utils"
-version = "1.1.12"
+version = "1.1.14"
 source = { editable = "." }
 dependencies = [
     { name = "langchain" },
@@ -431,8 +431,12 @@ standard = [
 ]
 
 [package.dev-dependencies]
+dev = [
+    { name = "ruff" },
+]
 tests = [
     { name = "langchain-deepseek" },
+    { name = "langchain-ollama" },
     { name = "langchain-qwq" },
     { name = "langchain-tests" },
     { name = "python-dotenv" },
@@ -447,13 +451,28 @@ requires-dist = [
 provides-extras = ["standard"]
 
 [package.metadata.requires-dev]
+dev = [{ name = "ruff", specifier = ">=0.14.5" }]
 tests = [
     { name = "langchain-deepseek", specifier = ">=1.0.0" },
+    { name = "langchain-ollama", specifier = ">=1.0.0" },
     { name = "langchain-qwq", specifier = ">=0.3.0" },
     { name = "langchain-tests", specifier = ">=1.0.0" },
     { name = "python-dotenv", specifier = ">=1.1.1" },
 ]
 
+[[package]]
+name = "langchain-ollama"
+version = "1.0.0"
+source = { registry = "https://pypi.mirrors.ustc.edu.cn/simple/" }
+dependencies = [
+    { name = "langchain-core" },
+    { name = "ollama" },
+]
+sdist = { url = "https://mirrors.ustc.edu.cn/pypi/packages/8b/45/d996b4c0e3e7155ff7a110bf24dcb5043fc1405559a2089c11fe97511cc2/langchain_ollama-1.0.0.tar.gz", hash = "sha256:2ea9ad1b0f0ab319d600b9193d1124a8925523a3b943d75a967718e24ec09a8a", size = 151042, upload-time = "2025-10-17T15:41:50.277Z" }
+wheels = [
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/91/08/7be292aee722692b13a93316247b57eefb83d4309f5fdfe636cc47786efe/langchain_ollama-1.0.0-py3-none-any.whl", hash = "sha256:5828523fcbd137847490841110a6aedf96b68534e7fe2735715ecf3e835b2391", size = 29006, upload-time = "2025-10-17T15:41:49.497Z" },
+]
+
 [[package]]
 name = "langchain-openai"
 version = "1.0.3"
@@ -797,6 +816,19 @@ wheels = [
     { url = "https://mirrors.ustc.edu.cn/pypi/packages/95/8e/2844c3959ce9a63acc7c8e50881133d86666f0420bcde695e115ced0920f/numpy-2.3.4-pp311-pypy311_pp73-win_amd64.whl", hash = "sha256:81b3a59793523e552c4a96109dde028aa4448ae06ccac5a76ff6532a85558a7f", size = 12973130, upload-time = "2025-10-15T16:18:09.397Z" },
 ]
 
+[[package]]
+name = "ollama"
+version = "0.6.1"
+source = { registry = "https://pypi.mirrors.ustc.edu.cn/simple/" }
+dependencies = [
+    { name = "httpx" },
+    { name = "pydantic" },
+]
+sdist = { url = "https://mirrors.ustc.edu.cn/pypi/packages/9d/5a/652dac4b7affc2b37b95386f8ae78f22808af09d720689e3d7a86b6ed98e/ollama-0.6.1.tar.gz", hash = "sha256:478c67546836430034b415ed64fa890fd3d1ff91781a9d548b3325274e69d7c6", size = 51620, upload-time = "2025-11-13T23:02:17.416Z" }
+wheels = [
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/47/4f/4a617ee93d8208d2bcf26b2d8b9402ceaed03e3853c754940e2290fed063/ollama-0.6.1-py3-none-any.whl", hash = "sha256:fc4c984b345735c5486faeee67d8a265214a31cbb828167782dc642ce0a2bf8c", size = 14354, upload-time = "2025-11-13T23:02:16.292Z" },
+]
+
 [[package]]
 name = "openai"
 version = "2.6.1"
@@ -1470,6 +1502,32 @@ wheels = [
     { url = "https://mirrors.ustc.edu.cn/pypi/packages/25/7a/b0178788f8dc6cafce37a212c99565fa1fe7872c70c6c9c1e1a372d9d88f/rich-14.2.0-py3-none-any.whl", hash = "sha256:76bc51fe2e57d2b1be1f96c524b890b816e334ab4c1e45888799bfaab0021edd", size = 243393, upload-time = "2025-10-09T14:16:51.245Z" },
 ]
 
+[[package]]
+name = "ruff"
+version = "0.14.5"
+source = { registry = "https://pypi.mirrors.ustc.edu.cn/simple/" }
+sdist = { url = "https://mirrors.ustc.edu.cn/pypi/packages/82/fa/fbb67a5780ae0f704876cb8ac92d6d76da41da4dc72b7ed3565ab18f2f52/ruff-0.14.5.tar.gz", hash = "sha256:8d3b48d7d8aad423d3137af7ab6c8b1e38e4de104800f0d596990f6ada1a9fc1", size = 5615944, upload-time = "2025-11-13T19:58:51.155Z" }
+wheels = [
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/68/31/c07e9c535248d10836a94e4f4e8c5a31a1beed6f169b31405b227872d4f4/ruff-0.14.5-py3-none-linux_armv6l.whl", hash = "sha256:f3b8248123b586de44a8018bcc9fefe31d23dda57a34e6f0e1e53bd51fd63594", size = 13171630, upload-time = "2025-11-13T19:57:54.894Z" },
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/8e/5c/283c62516dca697cd604c2796d1487396b7a436b2f0ecc3fd412aca470e0/ruff-0.14.5-py3-none-macosx_10_12_x86_64.whl", hash = "sha256:f7a75236570318c7a30edd7f5491945f0169de738d945ca8784500b517163a72", size = 13413925, upload-time = "2025-11-13T19:57:59.181Z" },
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/b6/f3/aa319f4afc22cb6fcba2b9cdfc0f03bbf747e59ab7a8c5e90173857a1361/ruff-0.14.5-py3-none-macosx_11_0_arm64.whl", hash = "sha256:6d146132d1ee115f8802356a2dc9a634dbf58184c51bff21f313e8cd1c74899a", size = 12574040, upload-time = "2025-11-13T19:58:02.056Z" },
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/f9/7f/cb5845fcc7c7e88ed57f58670189fc2ff517fe2134c3821e77e29fd3b0c8/ruff-0.14.5-py3-none-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:e2380596653dcd20b057794d55681571a257a42327da8894b93bbd6111aa801f", size = 13009755, upload-time = "2025-11-13T19:58:05.172Z" },
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/21/d2/bcbedbb6bcb9253085981730687ddc0cc7b2e18e8dc13cf4453de905d7a0/ruff-0.14.5-py3-none-manylinux_2_17_armv7l.manylinux2014_armv7l.whl", hash = "sha256:2d1fa985a42b1f075a098fa1ab9d472b712bdb17ad87a8ec86e45e7fa6273e68", size = 12937641, upload-time = "2025-11-13T19:58:08.345Z" },
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/a4/58/e25de28a572bdd60ffc6bb71fc7fd25a94ec6a076942e372437649cbb02a/ruff-0.14.5-py3-none-manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:88f0770d42b7fa02bbefddde15d235ca3aa24e2f0137388cc15b2dcbb1f7c7a7", size = 13610854, upload-time = "2025-11-13T19:58:11.419Z" },
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/7d/24/43bb3fd23ecee9861970978ea1a7a63e12a204d319248a7e8af539984280/ruff-0.14.5-py3-none-manylinux_2_17_ppc64.manylinux2014_ppc64.whl", hash = "sha256:3676cb02b9061fee7294661071c4709fa21419ea9176087cb77e64410926eb78", size = 15061088, upload-time = "2025-11-13T19:58:14.551Z" },
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/23/44/a022f288d61c2f8c8645b24c364b719aee293ffc7d633a2ca4d116b9c716/ruff-0.14.5-py3-none-manylinux_2_17_ppc64le.manylinux2014_ppc64le.whl", hash = "sha256:b595bedf6bc9cab647c4a173a61acf4f1ac5f2b545203ba82f30fcb10b0318fb", size = 14734717, upload-time = "2025-11-13T19:58:17.518Z" },
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/58/81/5c6ba44de7e44c91f68073e0658109d8373b0590940efe5bd7753a2585a3/ruff-0.14.5-py3-none-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:f55382725ad0bdb2e8ee2babcbbfb16f124f5a59496a2f6a46f1d9d99d93e6e2", size = 14028812, upload-time = "2025-11-13T19:58:20.533Z" },
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/ad/ef/41a8b60f8462cb320f68615b00299ebb12660097c952c600c762078420f8/ruff-0.14.5-py3-none-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:7497d19dce23976bdaca24345ae131a1d38dcfe1b0850ad8e9e6e4fa321a6e19", size = 13825656, upload-time = "2025-11-13T19:58:23.345Z" },
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/7c/00/207e5de737fdb59b39eb1fac806904fe05681981b46d6a6db9468501062e/ruff-0.14.5-py3-none-manylinux_2_31_riscv64.whl", hash = "sha256:410e781f1122d6be4f446981dd479470af86537fb0b8857f27a6e872f65a38e4", size = 13959922, upload-time = "2025-11-13T19:58:26.537Z" },
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/bc/7e/fa1f5c2776db4be405040293618846a2dece5c70b050874c2d1f10f24776/ruff-0.14.5-py3-none-musllinux_1_2_aarch64.whl", hash = "sha256:c01be527ef4c91a6d55e53b337bfe2c0f82af024cc1a33c44792d6844e2331e1", size = 12932501, upload-time = "2025-11-13T19:58:29.822Z" },
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/67/d8/d86bf784d693a764b59479a6bbdc9515ae42c340a5dc5ab1dabef847bfaa/ruff-0.14.5-py3-none-musllinux_1_2_armv7l.whl", hash = "sha256:f66e9bb762e68d66e48550b59c74314168ebb46199886c5c5aa0b0fbcc81b151", size = 12927319, upload-time = "2025-11-13T19:58:32.923Z" },
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/ac/de/ee0b304d450ae007ce0cb3e455fe24fbcaaedae4ebaad6c23831c6663651/ruff-0.14.5-py3-none-musllinux_1_2_i686.whl", hash = "sha256:d93be8f1fa01022337f1f8f3bcaa7ffee2d0b03f00922c45c2207954f351f465", size = 13206209, upload-time = "2025-11-13T19:58:35.952Z" },
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/33/aa/193ca7e3a92d74f17d9d5771a765965d2cf42c86e6f0fd95b13969115723/ruff-0.14.5-py3-none-musllinux_1_2_x86_64.whl", hash = "sha256:c135d4b681f7401fe0e7312017e41aba9b3160861105726b76cfa14bc25aa367", size = 13953709, upload-time = "2025-11-13T19:58:39.002Z" },
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/cc/f1/7119e42aa1d3bf036ffc9478885c2e248812b7de9abea4eae89163d2929d/ruff-0.14.5-py3-none-win32.whl", hash = "sha256:c83642e6fccfb6dea8b785eb9f456800dcd6a63f362238af5fc0c83d027dd08b", size = 12925808, upload-time = "2025-11-13T19:58:42.779Z" },
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/3b/9d/7c0a255d21e0912114784e4a96bf62af0618e2190cae468cd82b13625ad2/ruff-0.14.5-py3-none-win_amd64.whl", hash = "sha256:9d55d7af7166f143c94eae1db3312f9ea8f95a4defef1979ed516dbb38c27621", size = 14331546, upload-time = "2025-11-13T19:58:45.691Z" },
+    { url = "https://mirrors.ustc.edu.cn/pypi/packages/e5/80/69756670caedcf3b9be597a6e12276a6cf6197076eb62aad0c608f8efce0/ruff-0.14.5-py3-none-win_arm64.whl", hash = "sha256:4b700459d4649e2594b31f20a9de33bc7c19976d4746d8d0798ad959621d64a4", size = 13433331, upload-time = "2025-11-13T19:58:48.434Z" },
+]
+
 [[package]]
 name = "sniffio"
 version = "1.3.1"

langchain_dev_utils-1.1.13/src/langchain_dev_utils/__init__.py

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