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

langchain_dev_utils/__init__.py

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

langchain_dev_utils/agents/middleware/plan.py

@@ -244,9 +244,10 @@ def create_read_plan_tool(
 
 _PLAN_SYSTEM_PROMPT_NOT_READ_PLAN = """You can manage task plans using two simple tools:
 
-1. **write_plan**
+## write_plan
 - Use it to break complex tasks (3+ steps) into a clear, actionable list. Only include next steps to execute — the first becomes `"in_progress"`, the rest `"pending"`. Don’t use it for simple tasks (<3 steps).
-2. **finish_sub_plan**
+
+## finish_sub_plan
 - Call it **only when the current task is 100% done**. It automatically marks it `"done"` and promotes the next `"pending"` task to `"in_progress"`. No parameters needed. Never use it mid-task or if anything’s incomplete.
 Keep plans lean, update immediately, and never batch completions.
 """
@@ -277,9 +278,15 @@ class PlanMiddleware(AgentMiddleware):
 
     Args:
         system_prompt: Custom system prompt to guide the agent on using the plan tool.
-            If not provided, uses the default `_PLAN_MIDDLEWARE_SYSTEM_PROMPT`.
-        tools: List of tools to be added to the agent. The tools must be created by `create_write_plan_tool`, `create_finish_sub_plan_tool`, and `create_read_plan_tool`(optional).
-
+            If not provided, uses the default `_PLAN_SYSTEM_PROMPT` or `_PLAN_SYSTEM_PROMPT_NOT_READ_PLAN` based on the `use_read_plan_tool` parameter.
+        write_plan_tool_description: Description of the `write_plan` tool.
+            If not provided, uses the default `_DEFAULT_WRITE_PLAN_TOOL_DESCRIPTION`.
+        finish_sub_plan_tool_description: Description of the `finish_sub_plan` tool.
+            If not provided, uses the default `_DEFAULT_FINISH_SUB_PLAN_TOOL_DESCRIPTION`.
+        read_plan_tool_description: Description of the `read_plan` tool.
+            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`.
     Example:
         ```python
         from langchain_dev_utils.agents.middleware.plan import PlanMiddleware
@@ -304,6 +311,7 @@ class PlanMiddleware(AgentMiddleware):
         finish_sub_plan_tool_description: Optional[str] = None,
         read_plan_tool_description: Optional[str] = None,
         use_read_plan_tool: bool = True,
+        message_key: Optional[str] = None,
     ) -> None:
         super().__init__()
 
@@ -319,8 +327,12 @@ class PlanMiddleware(AgentMiddleware):
         )
 
         tools = [
-            create_write_plan_tool(description=write_plan_tool_description),
-            create_finish_sub_plan_tool(description=finish_sub_plan_tool_description),
+            create_write_plan_tool(
+                description=write_plan_tool_description, message_key=message_key
+            ),
+            create_finish_sub_plan_tool(
+                description=finish_sub_plan_tool_description, message_key=message_key
+            ),
         ]
 
         if use_read_plan_tool:

langchain_dev_utils/chat_models/adapters/openai_compatible.py

@@ -46,7 +46,7 @@ _DictOrPydantic = Union[dict, _BM]
 
 
 class _ModelProviderConfigType(BaseModel):
-    supported_tool_choice: ToolChoiceType = Field(default=[])
+    supported_tool_choice: ToolChoiceType = Field(default_factory=list)
     keep_reasoning_content: bool = Field(default=False)
     support_json_mode: bool = Field(default=False)
 

langchain_dev_utils/chat_models/base.py

@@ -1,10 +1,11 @@
-import os
 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 .types import ChatModelType, ToolChoiceType
+from pydantic import BaseModel
 
 _MODEL_PROVIDERS_DICT = {}
 
@@ -22,6 +23,34 @@ class ChatModelProvider(TypedDict):
     provider_config: NotRequired[ProviderConfig]
 
 
+def _get_base_url_field_name(model_cls: type[BaseModel]) -> str | None:
+    """
+    Return 'base_url' if the model has a field named or aliased as 'base_url',
+    else return 'api_base' if it has a field named or aliased as 'api_base',
+    else return None.
+    The return value is always either 'base_url', 'api_base', or None.
+    """
+    model_fields = model_cls.model_fields
+
+    # try model_fields first
+    if "base_url" in model_fields:
+        return "base_url"
+
+    if "api_base" in model_fields:
+        return "api_base"
+
+    # then try aliases
+    for field_info in model_fields.values():
+        if field_info.alias == "base_url":
+            return "base_url"
+
+    for field_info in model_fields.values():
+        if field_info.alias == "api_base":
+            return "api_base"
+
+    return None
+
+
 def _parse_model(model: str, model_provider: Optional[str]) -> tuple[str, str]:
     """Parse model string and provider.
 
@@ -71,6 +100,11 @@ def _load_chat_model_helper(
             "provider_config"
         ):
             kwargs.update({"provider_config": provider_config})
+
+        if base_url := _MODEL_PROVIDERS_DICT[model_provider].get("base_url"):
+            url_key = _get_base_url_field_name(chat_model)
+            if url_key:
+                kwargs.update({url_key: base_url})
         return chat_model(model=model, **kwargs)
 
     return _init_chat_model_helper(model, model_provider=model_provider, **kwargs)
@@ -91,7 +125,7 @@ def register_model_provider(
     Args:
         provider_name: Name of the provider to register
         chat_model: Either a BaseChatModel class or a string identifier for a supported provider
-        base_url: Optional base URL for API endpoints (Optional parameter;effective only when `chat_model` is a string and is "openai-compatible".)
+        base_url: The API address of the model provider (optional, valid for both types of `chat_model`, but mainly used when `chat_model` is a string and is "openai-compatible")
         provider_config: The configuration of the model provider (Optional parameter;effective only when `chat_model` is a string and is "openai-compatible".)
            It can be configured to configure some related parameters of the provider, such as whether to support json_mode structured output mode, the list of supported tool_choice
     Raises:
@@ -113,6 +147,7 @@ def register_model_provider(
         >>> model = load_chat_model(model="vllm:qwen3-4b")
         >>> model.invoke("Hello")
     """
+    base_url = base_url or from_env(f"{provider_name.upper()}_API_BASE", default=None)()
     if isinstance(chat_model, str):
         try:
             from .adapters.openai_compatible import _create_openai_compatible_model
@@ -120,8 +155,6 @@ def register_model_provider(
             raise ImportError(
                 "Please install langchain_dev_utils[standard],when chat_model is a 'openai-compatible'"
             )
-
-        base_url = base_url or os.getenv(f"{provider_name.upper()}_API_BASE")
         if base_url is None:
             raise ValueError(
                 f"base_url must be provided or set {provider_name.upper()}_API_BASE environment variable when chat_model is a string"
@@ -140,11 +173,17 @@ def register_model_provider(
                 provider_name: {
                     "chat_model": chat_model,
                     "provider_config": provider_config,
+                    "base_url": base_url,
                 }
             }
         )
     else:
-        _MODEL_PROVIDERS_DICT.update({provider_name: {"chat_model": chat_model}})
+        if base_url is not None:
+            _MODEL_PROVIDERS_DICT.update(
+                {provider_name: {"chat_model": chat_model, "base_url": base_url}}
+            )
+        else:
+            _MODEL_PROVIDERS_DICT.update({provider_name: {"chat_model": chat_model}})
 
 
 def batch_register_model_provider(
@@ -159,7 +198,7 @@ def batch_register_model_provider(
         providers: List of ChatModelProvider dictionaries, each containing:
             - provider_name: Name of the provider to register
             - chat_model: Either a BaseChatModel class or a string identifier for a supported provider
-            - base_url: Optional base URL for API endpoints(Optional parameter; effective only when `chat_model` is a string and is "openai-compatible".)
+            - base_url: The API address of the model provider (optional, valid for both types of `chat_model`, but mainly used when `chat_model` is a string and is "openai-compatible")
             - provider_config: The configuration of the model provider(Optional parameter; effective only when `chat_model` is a string and is "openai-compatible".)
                 It can be configured to configure some related parameters of the provider, such as whether to support json_mode structured output mode, the list of supported tool_choice
 

langchain_dev_utils/embeddings/base.py

@@ -1,7 +1,8 @@
-import os
 from typing import Any, Literal, NotRequired, Optional, TypedDict, Union
 
 from langchain.embeddings.base import Embeddings, _SUPPORTED_PROVIDERS, init_embeddings
+from langchain_core.utils import from_env, secret_from_env
+from pydantic import BaseModel
 
 _EMBEDDINGS_PROVIDERS_DICT = {}
 
@@ -14,6 +15,34 @@ class EmbeddingProvider(TypedDict):
     base_url: NotRequired[str]
 
 
+def _get_base_url_field_name(model_cls: type[BaseModel]) -> str | None:
+    """
+    Return 'base_url' if the model has a field named or aliased as 'base_url',
+    else return 'api_base' if it has a field named or aliased as 'api_base',
+    else return None.
+    The return value is always either 'base_url', 'api_base', or None.
+    """
+    model_fields = model_cls.model_fields
+
+    # try model_fields first
+    if "base_url" in model_fields:
+        return "base_url"
+
+    if "api_base" in model_fields:
+        return "api_base"
+
+    # then try aliases
+    for field_info in model_fields.values():
+        if field_info.alias == "base_url":
+            return "base_url"
+
+    for field_info in model_fields.values():
+        if field_info.alias == "api_base":
+            return "api_base"
+
+    return None
+
+
 def _parse_model_string(model_name: str) -> tuple[str, str]:
     """Parse model string into provider and model name.
 
@@ -56,7 +85,7 @@ def register_embeddings_provider(
     Args:
         provider_name: Name of the provider to register
         embeddings_model: Either an Embeddings class or a string identifier for a supported provider
-        base_url: Optional base URL for API endpoints (required when embeddings_model is a string)
+        base_url: The API address of the Embedding model provider (optional, valid for both types of `embeddings_model`, but mainly used when `embeddings_model` is a string and is "openai-compatible")
 
     Raises:
         ValueError: If base_url is not provided when embeddings_model is a string
@@ -77,8 +106,9 @@ def register_embeddings_provider(
         >>> embeddings = load_embeddings("vllm:qwen3-embedding-4b")
         >>> embeddings.embed_query("hello world")
     """
+
+    base_url = base_url or from_env(f"{provider_name.upper()}_API_BASE", default=None)()
     if isinstance(embeddings_model, str):
-        base_url = base_url or os.getenv(f"{provider_name.upper()}_API_BASE")
         if base_url is None:
             raise ValueError(
                 f"base_url must be provided or set {provider_name.upper()}_API_BASE environment variable when embeddings_model is a string"
@@ -98,9 +128,19 @@ def register_embeddings_provider(
             }
         )
     else:
-        _EMBEDDINGS_PROVIDERS_DICT.update(
-            {provider_name: {"embeddings_model": embeddings_model}}
-        )
+        if base_url is not None:
+            _EMBEDDINGS_PROVIDERS_DICT.update(
+                {
+                    provider_name: {
+                        "embeddings_model": embeddings_model,
+                        "base_url": base_url,
+                    }
+                }
+            )
+        else:
+            _EMBEDDINGS_PROVIDERS_DICT.update(
+                {provider_name: {"embeddings_model": embeddings_model}}
+            )
 
 
 def batch_register_embeddings_provider(
@@ -115,7 +155,7 @@ def batch_register_embeddings_provider(
         providers: List of EmbeddingProvider dictionaries, each containing:
             - provider_name: str - Provider name
             - embeddings_model: Union[Type[Embeddings], str] - Model class or provider string
-            - base_url: Optional[str] - Base URL for API endpoints
+            - base_url: The API address of the Embedding model provider (optional, valid for both types of `embeddings_model`, but mainly used when `embeddings_model` is a string and is "openai-compatible")
 
     Raises:
         ValueError: If any of the providers are invalid
@@ -186,7 +226,7 @@ def load_embeddings(
     embeddings = _EMBEDDINGS_PROVIDERS_DICT[provider]["embeddings_model"]
     if isinstance(embeddings, str):
         if not (api_key := kwargs.get("api_key")):
-            api_key = os.getenv(f"{provider.upper()}_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."
@@ -203,4 +243,8 @@ def load_embeddings(
             **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)

{langchain_dev_utils-1.1.12.dist-info → langchain_dev_utils-1.1.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-dev-utils
-Version: 1.1.12
+Version: 1.1.13
 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
@@ -61,7 +61,7 @@ Mainly consists of the following two functions:
 
 - `provider_name`: Model provider name, used as an identifier for subsequent model loading
 - `chat_model`: Chat model, can be a ChatModel or a string (currently supports "openai-compatible")
-- `base_url`: API address of the model provider (optional, valid when `chat_model` is a string and is "openai-compatible")
+- `base_url`: The API address of the model provider (optional, valid for both types of `chat_model`, but mainly used when `chat_model` is a string and is "openai-compatible")
 - `provider_config`: Relevant configuration for the model provider (optional, valid when `chat_model` is a string and is "openai-compatible"), can configure some provider-related parameters, such as whether to support structured output in json_mode, list of supported tool_choices, etc.
 
 `load_chat_model` parameter description:
@@ -101,7 +101,7 @@ Mainly consists of the following two functions:
 
 - `provider_name`: Embedding model provider name, used as an identifier for subsequent model loading
 - `embeddings_model`: Embedding model, can be Embeddings or a string (currently supports "openai-compatible")
-- `base_url`: API address of the model provider (optional, valid when `embeddings_model` is a string and is "openai-compatible")
+- `base_url`: The API address of the Embedding model provider (optional, valid for both types of `embeddings_model`, but mainly used when `embeddings_model` is a string and is "openai-compatible")
 
 `load_embeddings` parameter description:
 

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

@@ -1,4 +1,4 @@
-langchain_dev_utils/__init__.py,sha256=r2af4_JOawT2T11XCdpXkvokHT7xemJLdLkba1ehMoo,23
+langchain_dev_utils/__init__.py,sha256=fHD3CDKZLpB_vekISdHB54mQUzaieloAPZEREmkbiRQ,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
@@ -8,17 +8,17 @@ langchain_dev_utils/agents/wrap.py,sha256=4BWksU9DRz8c3ZHQiUi4GHwGhNysDLNs8pmLWV
 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=cFC2ZWOAUEFB1s6oRkm64cTAFpk8Af5hI-hoLcsBmSw,14535
+langchain_dev_utils/agents/middleware/plan.py,sha256=pVABuihOo-TGuPwJA_AdpBa6eodbdZalXozl_YcMsHc,15198
 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=IHappv8dTQjJe5Cb3JIJsNBxP26r48M69J9XN5NZTnQ,9802
+langchain_dev_utils/chat_models/base.py,sha256=BagUNjqWwTZ2vJ-uHPQ0vyC6nYXOdFJidV_73jlPFG8,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=gvVs8_ymzkbK0Ngz8xN6mFRv8VrNvm5prZP73EVtSbk,18399
+langchain_dev_utils/chat_models/adapters/openai_compatible.py,sha256=6ZTRCFqgW8fk8nbZs0OarmuHP5M6wr-0mbFogZuLTWY,18409
 langchain_dev_utils/embeddings/__init__.py,sha256=zbEOaV86TUi9Zrg_dH9dpdgacWg31HMJTlTQknA9EKk,244
-langchain_dev_utils/embeddings/base.py,sha256=pLFfhgBM3A1uKfvF8eghl-oiVH5W8JiAiALRTPBvC90,7816
+langchain_dev_utils/embeddings/base.py,sha256=OFXgaLO6DsadSITUmtrDvJg_-042lrrDwY5vnS9_do8,9574
 langchain_dev_utils/message_convert/__init__.py,sha256=xwjaQ1oJoc80xy70oQI4uW3gAmgV5JymJd5hgnA6s3g,458
 langchain_dev_utils/message_convert/content.py,sha256=ApmQ7fUUBO3Ihjm2hYSWd4GrU_CvrjbWla-MA7DAFRc,7758
 langchain_dev_utils/message_convert/format.py,sha256=fh4GyyuZBTMrHeCEwdu9fOh5n8tdli1vDF44jK1i-tI,2373
@@ -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.12.dist-info/METADATA,sha256=qYjcWvrmpBta1286FierB8IIqKOKPQMuek0r6ZQO2Ts,16144
-langchain_dev_utils-1.1.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langchain_dev_utils-1.1.12.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
-langchain_dev_utils-1.1.12.dist-info/RECORD,,
+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,,