langchain-dev-utils 1.1.13__py3-none-any.whl → 1.1.14__py3-none-any.whl

langchain_dev_utils/__init__.py

@@ -1 +1 @@
-__version__ = "1.1.13"
+__version__ = "1.1.14"

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/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/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.dist-info → langchain_dev_utils-1.1.14.dist-info}/METADATA

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

{langchain_dev_utils-1.1.13.dist-info → langchain_dev_utils-1.1.14.dist-info}/RECORD

@@ -1,19 +1,19 @@
-langchain_dev_utils/__init__.py,sha256=fHD3CDKZLpB_vekISdHB54mQUzaieloAPZEREmkbiRQ,23
+langchain_dev_utils/__init__.py,sha256=mqdNVCi4IalSmo5XOmgqpcssoVNsIsPLW08ykYdHXYU,23
 langchain_dev_utils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langchain_dev_utils/agents/__init__.py,sha256=e17SMQdJIQngbUCr2N1tY-yw0tD3tEnH7PSvyDmVPeQ,127
-langchain_dev_utils/agents/factory.py,sha256=pQeqz_ZlU43Os5gKlRu5-iCLTslPWEqWJzuGxpKhcRo,3904
+langchain_dev_utils/agents/factory.py,sha256=h4uAkid2NJMMs4qV6RYFexew-ixfhEvpa254eDeDEcU,3912
 langchain_dev_utils/agents/file_system.py,sha256=S6RUEmQI2eerW0gBQp0IP0X5ak5FwvqgIGRiycr2iyw,8468
 langchain_dev_utils/agents/plan.py,sha256=ydJuJLlNydheQvLPl2uCc3TBVv42YxGzPhKgtldIdIk,6497
 langchain_dev_utils/agents/wrap.py,sha256=4BWksU9DRz8c3ZHQiUi4GHwGhNysDLNs8pmLWV7BeAI,5165
 langchain_dev_utils/agents/middleware/__init__.py,sha256=cjrb8Rue5uukl9pKPF7CjSrHtcYsUBj3Mdvv2szlp7E,679
 langchain_dev_utils/agents/middleware/model_fallback.py,sha256=cvTj_sOw3r4B4ErMAVdsrniMImWnUpLMECmQErxdsUU,1688
 langchain_dev_utils/agents/middleware/model_router.py,sha256=YkaPpYmIZaGj--YlUjm7dVcNzRt3Au317eor4SDYsQs,8799
-langchain_dev_utils/agents/middleware/plan.py,sha256=pVABuihOo-TGuPwJA_AdpBa6eodbdZalXozl_YcMsHc,15198
+langchain_dev_utils/agents/middleware/plan.py,sha256=ew9AiJLgLawnlBpiqBPXxyT9CZ1i4aqtwlbi_Qbly4w,15282
 langchain_dev_utils/agents/middleware/summarization.py,sha256=Ws-_cxSQQfa5rn5Spq1gSLpgIleUCno3QmWRvN4-u9E,2213
 langchain_dev_utils/agents/middleware/tool_emulator.py,sha256=u9rV24yUB-dyc1uUfUe74B1wOGVI3TZRwxkE1bvGm18,2025
 langchain_dev_utils/agents/middleware/tool_selection.py,sha256=ZqdyK4Yhp2u3GM6B_D6U7Srca9vy1o7s6N_LrV24-dQ,3107
 langchain_dev_utils/chat_models/__init__.py,sha256=YSLUyHrWEEj4y4DtGFCOnDW02VIYZdfAH800m4Klgeg,224
-langchain_dev_utils/chat_models/base.py,sha256=BagUNjqWwTZ2vJ-uHPQ0vyC6nYXOdFJidV_73jlPFG8,11232
+langchain_dev_utils/chat_models/base.py,sha256=l2833l9RuTnS3OecoRL2RWeYc17kLWoC6nuWWUpo5FU,11232
 langchain_dev_utils/chat_models/types.py,sha256=oPXFsfho9amnwek5v3ey8LcnsfKVzecWSJcKVBG4ETc,261
 langchain_dev_utils/chat_models/adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langchain_dev_utils/chat_models/adapters/openai_compatible.py,sha256=6ZTRCFqgW8fk8nbZs0OarmuHP5M6wr-0mbFogZuLTWY,18409
@@ -29,7 +29,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=nbaON9806pv5tpMRQUA_Ch3HJA5HBFgzZR7kQRf6PiY,9819
 langchain_dev_utils/tool_calling/utils.py,sha256=3cNv_Zx32KxdsGn8IkxjWUzxYEEwVJeJgTZTbfSg0pA,2751
-langchain_dev_utils-1.1.13.dist-info/METADATA,sha256=m205M6P2wNSDHHHNCVuSE-1SBMD1oyiMHRYYwHbJyEA,16264
-langchain_dev_utils-1.1.13.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langchain_dev_utils-1.1.13.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
-langchain_dev_utils-1.1.13.dist-info/RECORD,,
+langchain_dev_utils-1.1.14.dist-info/METADATA,sha256=VqNoa-IeJpZsKauWp1An3t1amgCCypyDH2xPiUwoVsk,16264
+langchain_dev_utils-1.1.14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langchain_dev_utils-1.1.14.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
+langchain_dev_utils-1.1.14.dist-info/RECORD,,