langchain-dev-utils 1.2.8__py3-none-any.whl → 1.2.9__py3-none-any.whl

langchain_dev_utils/__init__.py

@@ -1 +1 @@
-__version__ = "1.2.8"
+__version__ = "1.2.9"

langchain_dev_utils/chat_models/adapters/openai_compatible.py

@@ -41,7 +41,7 @@ from typing_extensions import Self
 
 from ..types import (
     CompatibilityOptions,
-    ReasoningContentKeepType,
+    ReasoningKeepPolicy,
     ResponseFormatType,
     ToolChoiceType,
 )
@@ -119,7 +119,7 @@ class _BaseChatOpenAICompatible(BaseChatOpenAI):
     """Supported tool choice"""
     supported_response_format: ResponseFormatType = Field(default_factory=list)
     """Supported response format"""
-    reasoning_content_keep_type: ReasoningContentKeepType = Field(default="discard")
+    reasoning_keep_policy: ReasoningKeepPolicy = Field(default="never")
     """How to keep reasoning content in the messages"""
     include_usage: bool = Field(default=True)
     """Whether to include usage information in the output"""
@@ -159,7 +159,7 @@ class _BaseChatOpenAICompatible(BaseChatOpenAI):
 
         payload_messages = []
         last_human_index = -1
-        if self.reasoning_content_keep_type == "temp":
+        if self.reasoning_keep_policy == "current":
             last_human_index = _get_last_human_message_index(messages)
 
         for index, m in enumerate(messages):
@@ -167,15 +167,14 @@ class _BaseChatOpenAICompatible(BaseChatOpenAI):
                 msg_dict = _convert_message_to_dict(
                     _convert_from_v1_to_chat_completions(m)
                 )
-                if (
-                    self.reasoning_content_keep_type == "retain"
-                    and m.additional_kwargs.get("reasoning_content")
+                if self.reasoning_keep_policy == "all" and m.additional_kwargs.get(
+                    "reasoning_content"
                 ):
                     msg_dict["reasoning_content"] = m.additional_kwargs.get(
                         "reasoning_content"
                     )
                 elif (
-                    self.reasoning_content_keep_type == "temp"
+                    self.reasoning_keep_policy == "current"
                     and index > last_human_index
                     and m.additional_kwargs.get("reasoning_content")
                 ):
@@ -558,13 +557,9 @@ def _create_openai_compatible_model(
             ToolChoiceType,
             Field(default=compatibility_options.get("supported_tool_choice", ["auto"])),
         ),
-        reasoning_content_keep_type=(
-            ReasoningContentKeepType,
-            Field(
-                default=compatibility_options.get(
-                    "reasoning_content_keep_type", "discard"
-                )
-            ),
+        reasoning_keep_policy=(
+            ReasoningKeepPolicy,
+            Field(default=compatibility_options.get("reasoning_keep_policy", "never")),
         ),
         supported_response_format=(
             ResponseFormatType,

langchain_dev_utils/chat_models/types.py

@@ -9,11 +9,11 @@ ToolChoiceType = list[Literal["auto", "none", "required", "specific"]]
 
 ResponseFormatType = list[Literal["json_schema", "json_mode"]]
 
-ReasoningContentKeepType = Literal["discard", "temp", "retain"]
+ReasoningKeepPolicy = Literal["never", "current", "all"]
 
 
 class CompatibilityOptions(TypedDict):
     supported_tool_choice: NotRequired[ToolChoiceType]
     supported_response_format: NotRequired[ResponseFormatType]
-    reasoning_content_keep_type: NotRequired[ReasoningContentKeepType]
+    reasoning_keep_policy: NotRequired[ReasoningKeepPolicy]
     include_usage: NotRequired[bool]

langchain_dev_utils/embeddings/base.py

@@ -218,30 +218,30 @@ 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")):
-            api_key = secret_from_env(f"{provider.upper()}_API_KEY", default=None)()
-            if not api_key:
-                raise ValueError(
-                    f"API key for {provider} not found. Please set it in the environment."
-                )
-            kwargs["api_key"] = api_key
-            if embeddings == "openai-compatible":
-                kwargs["check_embedding_ctx_length"] = False
-                embeddings = "openai"
-        return init_embeddings(
-            model=model,
-            provider=embeddings,
-            base_url=_EMBEDDINGS_PROVIDERS_DICT[provider]["base_url"],
-            **kwargs,
-        )
+    if provider in _EMBEDDINGS_PROVIDERS_DICT:
+        embeddings = _EMBEDDINGS_PROVIDERS_DICT[provider]["embeddings_model"]
+        if isinstance(embeddings, str):
+            if not (api_key := kwargs.get("api_key")):
+                api_key = secret_from_env(f"{provider.upper()}_API_KEY", default=None)()
+                if not api_key:
+                    raise ValueError(
+                        f"API key for {provider} not found. Please set it in the environment."
+                    )
+                kwargs["api_key"] = api_key
+                if embeddings == "openai-compatible":
+                    kwargs["check_embedding_ctx_length"] = False
+                    embeddings = "openai"
+            return init_embeddings(
+                model=model,
+                provider=embeddings,
+                base_url=_EMBEDDINGS_PROVIDERS_DICT[provider]["base_url"],
+                **kwargs,
+            )
+        else:
+            if base_url := _EMBEDDINGS_PROVIDERS_DICT[provider].get("base_url"):
+                url_key = _get_base_url_field_name(embeddings)
+                if url_key is not None:
+                    kwargs.update({url_key: base_url})
+            return embeddings(model=model, **kwargs)
     else:
-        if base_url := _EMBEDDINGS_PROVIDERS_DICT[provider].get("base_url"):
-            url_key = _get_base_url_field_name(embeddings)
-            if url_key is not None:
-                kwargs.update({url_key: base_url})
-        return embeddings(model=model, **kwargs)
+        return init_embeddings(model, provider=provider, **kwargs)

langchain_dev_utils/message_convert/content.py

@@ -52,8 +52,10 @@ def convert_reasoning_content_for_ai_message(
     reasoning_content = _get_reasoning_content(model_response)
 
     if reasoning_content:
-        model_response.content = (
-            f"{think_tag[0]}{reasoning_content}{think_tag[1]}{model_response.content}"
+        return model_response.model_copy(
+            update={
+                "content": f"{think_tag[0]}{reasoning_content}{think_tag[1]}{model_response.content}"
+            }
         )
     return model_response
 
@@ -99,12 +101,16 @@ def convert_reasoning_content_for_chunk_iterator(
             reasoning_content = _get_reasoning_content(chunk)
             if reasoning_content:
                 if isfirst:
-                    chunk.content = f"{think_tag[0]}{reasoning_content}"
+                    chunk = chunk.model_copy(
+                        update={"content": f"{think_tag[0]}{reasoning_content}"}
+                    )
                     isfirst = False
                 else:
-                    chunk.content = reasoning_content
+                    chunk = chunk.model_copy(update={"content": reasoning_content})
             elif chunk.content and isend and not isfirst:
-                chunk.content = f"{think_tag[1]}{chunk.content}"
+                chunk = chunk.model_copy(
+                    update={"content": f"{think_tag[1]}{chunk.content}"}
+                )
                 isend = False
         yield chunk
 
@@ -149,12 +155,16 @@ async def aconvert_reasoning_content_for_chunk_iterator(
             reasoning_content = _get_reasoning_content(chunk)
             if reasoning_content:
                 if isfirst:
-                    chunk.content = f"{think_tag[0]}{reasoning_content}"
+                    chunk = chunk.model_copy(
+                        update={"content": f"{think_tag[0]}{reasoning_content}"}
+                    )
                     isfirst = False
                 else:
-                    chunk.content = reasoning_content
+                    chunk = chunk.model_copy(update={"content": reasoning_content})
             elif chunk.content and isend and not isfirst:
-                chunk.content = f"{think_tag[1]}{chunk.content}"
+                chunk = chunk.model_copy(
+                    update={"content": f"{think_tag[1]}{chunk.content}"}
+                )
                 isend = False
         yield chunk
 

langchain_dev_utils/tool_calling/utils.py

@@ -62,20 +62,20 @@ def parse_tool_calling(
         ...     tool_calls = parse_tool_calling(response)
     """
 
-    tool_call = None
+    tool_calls = None
 
     tool_call_blocks = [
         block for block in message.content_blocks if block["type"] == "tool_call"
     ]
     if tool_call_blocks:
-        tool_call = tool_call_blocks
+        tool_calls = tool_call_blocks
 
-    if not tool_call:
-        tool_call = message.tool_calls
+    if not tool_calls:
+        tool_calls = message.tool_calls
 
-    if not tool_call:
+    if not tool_calls:
         raise ValueError("No tool call found in message")
 
     if first_tool_call_only:
-        return (tool_call[0]["name"], tool_call[0]["args"])
-    return [(tool_call["name"], tool_call["args"]) for tool_call in tool_call]
+        return (tool_calls[0]["name"], tool_calls[0]["args"])
+    return [(tool_call["name"], tool_call["args"]) for tool_call in tool_calls]

{langchain_dev_utils-1.2.8.dist-info → langchain_dev_utils-1.2.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-dev-utils
-Version: 1.2.8
+Version: 1.2.9
 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
@@ -192,52 +192,66 @@ def get_current_time() -> str:
 
 ### 4. **Agent Development**
 
-Includes the following features:
+Includes the following capabilities:
 
-- Predefined agent factory functions
-- Common middleware components
+- Multi-agent construction  
+- Commonly used middleware components  
 
-#### 4.1 Agent Factory Functions
+#### 4.1 Multi-Agent Construction
 
-In LangChain v1, the official `create_agent` function can be used to create a single agent; its `model` parameter accepts either a BaseChatModel instance or a specific string (when a string is provided, only models supported by `init_chat_model` are allowed). To extend the flexibility of specifying models via string, this library provides an equivalent `create_agent` function that lets you designate any model supported by `load_chat_model` (registration required beforehand).
+Wrapping an agent as a tool is a common implementation pattern in multi-agent systems, as elaborated in the official LangChain documentation. To support this pattern, this library provides a pre-built utility function `wrap_agent_as_tool`, which encapsulates an agent instance into a tool that can be invoked by other agents.
 
-Usage example:
+**Usage Example**:
 
 ```python
-from langchain_dev_utils.agents import create_agent
+import datetime
+from langchain_dev_utils.agents import create_agent, wrap_agent_as_tool
 from langchain.agents import AgentState
 
+
+@tool
+def get_current_time() -> str:
+    """Get the current time."""
+    return datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+
+
 agent = create_agent("vllm:qwen3-4b", tools=[get_current_time], name="time-agent")
-response = agent.invoke({"messages": [{"role": "user", "content": "What time is it?"}]})
+call_time_agent_tool = wrap_agent_as_tool(agent)
+response = call_time_agent_tool.invoke(
+    {"messages": [{"role": "user", "content": "What time is it now?"}]}
+)
 print(response)
 ```
 
 #### 4.2 Middleware
 
-Provides some commonly used middleware components. Below, we illustrate with `ToolCallRepairMiddleware` and `PlanMiddleware`.
-
-`ToolCallRepairMiddleware` is used to repair `invalid_tool_calls` generated by large language models.
+Provides several commonly used middleware components. Below are examples using `ToolCallRepairMiddleware` and `PlanMiddleware`.
 
-`PlanMiddleware` is used for agent planning.
+- `ToolCallRepairMiddleware` automatically repairs malformed tool calls found in the model's `invalid_tool_calls` output.
+- `PlanMiddleware` enables task planning capabilities for agents.
 
 ```python
 from langchain_dev_utils.agents.middleware import (
-    ToolcallRepairMiddleware,
+    ToolCallRepairMiddleware,
     PlanMiddleware,
 )
 
 agent = create_agent(
     "vllm:qwen3-4b",
     name="plan-agent",
-    middleware=[ToolCallRepairMiddleware(), PlanMiddleware(
-        use_read_plan_tool=False
-    )]
+    middleware=[
+        ToolCallRepairMiddleware(),
+        PlanMiddleware(use_read_plan_tool=False)
+    ]
 )
-response = agent.invoke({"messages": [{"role": "user", "content": "Give me a travel plan to New York"}]})
+response = agent.invoke({"messages": [{"role": "user", "content": "Give me a travel plan for visiting New York."}]})
 print(response)
 ```
 
-**For more information about agent development and all built-in middleware, please refer to**: [Pre-built Agent Functions](https://tbice123123.github.io/langchain-dev-utils-docs/en/agent-development/prebuilt.html), [Middleware](https://tbice123123.github.io/langchain-dev-utils-docs/en/agent-development/middleware.html)
+**For more details on agent development and a complete list of built-in middleware, please refer to**:  
+[Multi-Agent Construction](https://tbice123123.github.io/langchain-dev-utils-docs/en/agent-development/multi-agent.html),  
+[Middleware](https://tbice123123.github.io/langchain-dev-utils-docs/en/agent-development/middleware.html)
+
 
 ### 5. **State Graph Orchestration**
 

{langchain_dev_utils-1.2.8.dist-info → langchain_dev_utils-1.2.9.dist-info}/RECORD

@@ -1,4 +1,4 @@
-langchain_dev_utils/__init__.py,sha256=CfVXm0wwlKPW0khOcwhWw61TpgtZiLijCePsAIOK3aU,22
+langchain_dev_utils/__init__.py,sha256=Oh3Y6CIypkhAjW-aquBTyP3_cA-gKgKTwq9EpcWpjps,22
 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
@@ -16,13 +16,13 @@ langchain_dev_utils/agents/middleware/tool_emulator.py,sha256=OgtPhqturaWzF4fRSJ
 langchain_dev_utils/agents/middleware/tool_selection.py,sha256=dRH5ejR6N02Djwxt6Gd63MYkg6SV5pySlzaRt53OoZk,3113
 langchain_dev_utils/chat_models/__init__.py,sha256=YSLUyHrWEEj4y4DtGFCOnDW02VIYZdfAH800m4Klgeg,224
 langchain_dev_utils/chat_models/base.py,sha256=CVMfgqMRnIKv8z4babusa2c4RKVuiWTL39mPD8cHAf4,11880
-langchain_dev_utils/chat_models/types.py,sha256=w9Zu2I_HtpWQ1jNEUE9QkEunxD6UUtIh0hGJVb7b5gk,690
+langchain_dev_utils/chat_models/types.py,sha256=kVLbT-IbvNtWPVmyVmh58le5r8XCqrEwuFB9-TWCBJk,672
 langchain_dev_utils/chat_models/adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langchain_dev_utils/chat_models/adapters/openai_compatible.py,sha256=hH713hs4LRfqUbYlqJKR0geJUjXkQAbU6-segYyuLCE,21599
+langchain_dev_utils/chat_models/adapters/openai_compatible.py,sha256=YgC8ups0owOVmY-Fwi7oxiktsC9LG6UZka8XNTCBW9g,21457
 langchain_dev_utils/embeddings/__init__.py,sha256=zbEOaV86TUi9Zrg_dH9dpdgacWg31HMJTlTQknA9EKk,244
-langchain_dev_utils/embeddings/base.py,sha256=l4uCB5ecr3GAkfYGpYxqamOPIM6fkP1H_QK-277YEic,9295
+langchain_dev_utils/embeddings/base.py,sha256=BGoWY0L7nG9iRV3d4sSagXhECXrwvS1xA-A_OVltn3k,9406
 langchain_dev_utils/message_convert/__init__.py,sha256=ZGrHGXPKMrZ_p9MqfIVZ4jgbEyb7aC4Q7X-muuThIYU,457
-langchain_dev_utils/message_convert/content.py,sha256=LhrFXL1zYkkpp4ave6SBorDLig5xnllQ2VYCgFz-eR4,7681
+langchain_dev_utils/message_convert/content.py,sha256=2V1g21byg3iLv5RjUW8zv3jwYwV7IH2hNim7jGRsIes,8096
 langchain_dev_utils/message_convert/format.py,sha256=1TOcJ09atH7LRtn_IIuBshKDXAyqoy3Q9b0Po-S-F9g,2377
 langchain_dev_utils/pipeline/__init__.py,sha256=eE6WktaLHDkqMeXDIDaLtm-OPTwtsX_Av8iK9uYrceo,186
 langchain_dev_utils/pipeline/parallel.py,sha256=nwZWbdSNeyanC9WufoJBTceotgT--UnPOfStXjgNMOc,5271
@@ -30,8 +30,8 @@ langchain_dev_utils/pipeline/sequential.py,sha256=sYJXQzVHDKUc-UV-HMv38JTPnse1A7
 langchain_dev_utils/pipeline/types.py,sha256=T3aROKKXeWvd0jcH5XkgMDQfEkLfPaiOhhV2q58fDHs,112
 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=W2ZRRMhn7SHHZxFfCXVaPIh2uFkY2XkO6EWrdRuv6VE,2757
-langchain_dev_utils-1.2.8.dist-info/METADATA,sha256=gzvPmy60STe1Yg-qZHJVMBRPRwXZ8_3uabCEdZjeXmE,13100
-langchain_dev_utils-1.2.8.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-langchain_dev_utils-1.2.8.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
-langchain_dev_utils-1.2.8.dist-info/RECORD,,
+langchain_dev_utils/tool_calling/utils.py,sha256=S4-KXQ8jWmpGTXYZitovF8rxKpaSSUkFruM8LDwvcvE,2765
+langchain_dev_utils-1.2.9.dist-info/METADATA,sha256=zzBTgY8EUmuz08ofh1t1letaKywRXuqQgfPdqWBVw2Q,13279
+langchain_dev_utils-1.2.9.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+langchain_dev_utils-1.2.9.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
+langchain_dev_utils-1.2.9.dist-info/RECORD,,