langchain-dev-utils 1.2.5__py3-none-any.whl → 1.2.6__py3-none-any.whl

langchain_dev_utils/__init__.py

@@ -1 +1 @@
-__version__ = "1.2.5"
+__version__ = "1.2.6"

langchain_dev_utils/_utils.py

@@ -0,0 +1,39 @@
+from importlib import util
+
+from pydantic import BaseModel
+
+
+def _check_langchain_openai_install() -> None:
+    if not util.find_spec("langchain_openai"):
+        msg = (
+            "Please install langchain_dev_utils[standard],when use 'openai-compatible'"
+        )
+        raise ImportError(msg)
+
+
+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

langchain_dev_utils/chat_models/base.py

@@ -3,7 +3,11 @@ 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 langchain_dev_utils._utils import (
+    _check_langchain_openai_install,
+    _get_base_url_field_name,
+)
 
 from .types import ChatModelType, CompatibilityOptions
 
@@ -18,34 +22,6 @@ class ChatModelProvider(TypedDict):
     compatibility_options: NotRequired[CompatibilityOptions]
 
 
-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.
 
@@ -89,7 +65,7 @@ def _load_chat_model_helper(
         BaseChatModel: Initialized chat model instance
     """
     model, model_provider = _parse_model(model, model_provider)
-    if model_provider in _MODEL_PROVIDERS_DICT.keys():
+    if model_provider in _MODEL_PROVIDERS_DICT:
         chat_model = _MODEL_PROVIDERS_DICT[model_provider]["chat_model"]
         if base_url := _MODEL_PROVIDERS_DICT[model_provider].get("base_url"):
             url_key = _get_base_url_field_name(chat_model)
@@ -98,7 +74,7 @@ def _load_chat_model_helper(
         if model_profiles := _MODEL_PROVIDERS_DICT[model_provider].get(
             "model_profiles"
         ):
-            if model in model_profiles:
+            if model in model_profiles and "profile" not in kwargs:
                 kwargs.update({"profile": model_profiles[model]})
         return chat_model(model=model, **kwargs)
 
@@ -145,12 +121,9 @@ def register_model_provider(
     """
     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
-        except ImportError:
-            raise ImportError(
-                "Please install langchain_dev_utils[standard],when chat_model is a 'openai-compatible'"
-            )
+        _check_langchain_openai_install()
+        from .adapters.openai_compatible import _create_openai_compatible_model
+
         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"

langchain_dev_utils/embeddings/base.py

@@ -2,7 +2,11 @@ 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
+
+from langchain_dev_utils._utils import (
+    _check_langchain_openai_install,
+    _get_base_url_field_name,
+)
 
 _EMBEDDINGS_PROVIDERS_DICT = {}
 
@@ -15,34 +19,6 @@ 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.
 
@@ -119,6 +95,8 @@ def register_embeddings_provider(
                 "when embeddings_model is a string, the value must be 'openai-compatible'"
             )
 
+        _check_langchain_openai_install()
+
         _EMBEDDINGS_PROVIDERS_DICT.update(
             {
                 provider_name: {
@@ -238,7 +216,6 @@ def load_embeddings(
             if embeddings == "openai-compatible":
                 kwargs["check_embedding_ctx_length"] = False
                 embeddings = "openai"
-
         return init_embeddings(
             model=model,
             provider=embeddings,

{langchain_dev_utils-1.2.5.dist-info → langchain_dev_utils-1.2.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langchain-dev-utils
-Version: 1.2.5
+Version: 1.2.6
 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
@@ -157,6 +157,8 @@ For stream responses obtained using `stream()` and `astream()`, you can use `mer
 | `chunks` | List[AIMessageChunk] | Yes | - | List of AIMessageChunk objects |
 
 ```python
+from langchain_dev_utils.message_convert import merge_ai_message_chunk
+
 chunks = list(model.stream("Hello"))
 merged = merge_ai_message_chunk(chunks)
 ```
@@ -174,6 +176,7 @@ For a list, you can use `format_sequence` to format it.
 | `with_num` | bool | No | False | If True, add a numeric prefix to each item (e.g., "1. Hello") |
 
 ```python
+from langchain_dev_utils.message_convert import format_sequence
 text = format_sequence([
     "str1",
     "str2",
@@ -205,7 +208,7 @@ Includes the following features:
 | Parameter | Type | Required | Default | Description |
 |-----------|------|----------|---------|-------------|
 | `message` | AIMessage | Yes | - | AIMessage object |
-| `first_tool_call_only` | bool | No | False | Whether to only check the first tool call |
+| `first_tool_call_only` | bool | No | False | Whether to only parse the first tool call |
 
 ```python
 import datetime
@@ -234,7 +237,7 @@ if has_tool_calling(response):
 Both can accept a `handler` parameter for custom breakpoint return and response handling logic.
 
 ```python
-from langchain_dev_utils import human_in_the_loop
+from langchain_dev_utils.tool_calling import human_in_the_loop
 from langchain_core.tools import tool
 import datetime
 
@@ -262,7 +265,7 @@ In LangChain v1, the officially provided `create_agent` function can be used to
 
 | Parameter | Type | Required | Default | Description |
 |-----------|------|----------|---------|-------------|
-| `model` | str \| BaseChatModel | Yes | - | Model name or model instance. Can be a string identifier for a model registered with `register_model_provider` or a BaseChatModel instance. |
+| `model` | str  | Yes | - | Model name or model instance. Can be a string identifier for a model registered with `register_model_provider` or a BaseChatModel instance. |
 | Other parameters | Various | No | - | All other parameters are the same as in `langchain.agents.create_agent` |
 
 Usage example:
@@ -317,7 +320,7 @@ Uses `create_sequential_pipeline`, supported parameters:
 
 | Parameter | Type | Required | Default | Description |
 |-----------|------|----------|---------|-------------|
-| `sub_graphs` | List[StateGraph] | Yes | - | List of state graphs to combine (must be StateGraph instances) |
+| `sub_graphs` | List[StateGraph\|CompiledStateGraph] | Yes | - | List of state graphs to combine (must be StateGraph instances or CompiledStateGraph instances) |
 | `state_schema` | type | Yes | - | State Schema for the final generated graph |
 | `graph_name` | str | No | - | Name of the final generated graph |
 | `context_schema` | type | No | - | Context Schema for the final generated graph |
@@ -378,7 +381,7 @@ Uses `create_parallel_pipeline`, supported parameters:
 
 | Parameter | Type | Required | Default | Description |
 |-----------|------|----------|---------|-------------|
-| `sub_graphs` | List[StateGraph] | Yes | - | List of state graphs to combine |
+| `sub_graphs` | List[StateGraph\|CompiledStateGraph] | Yes | - | List of state graphs to combine (must be StateGraph instances or CompiledStateGraph instances) |
 | `state_schema` | type | Yes | - | State Schema for the final generated graph |
 | `branches_fn` | Callable | Yes | - | Parallel branch function, returns a list of Send objects to control parallel execution |
 | `graph_name` | str | No | - | Name of the final generated graph |

{langchain_dev_utils-1.2.5.dist-info → langchain_dev_utils-1.2.6.dist-info}/RECORD

@@ -1,4 +1,5 @@
-langchain_dev_utils/__init__.py,sha256=jBmZf3HLbiQlWiolOsAA6J5-BbxXD2bqFqEqDH3lfqo,22
+langchain_dev_utils/__init__.py,sha256=vMQK58X8_YZGKzRm0ThvPAKFtpfyejGmUnDrY9RQ13w,22
+langchain_dev_utils/_utils.py,sha256=5bFs4cf3HvkMNkv35V8Sowu4YSXmfF5VNwmv_eHfkgQ,1151
 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=JjdJwPTJpQwAlwQlBalbuGej5Jcpy2Fz6lH3EwEaxQo,3979
@@ -13,12 +14,12 @@ langchain_dev_utils/agents/middleware/summarization.py,sha256=BtWPJcQBssGAT0nb1c
 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=MeVFt16ytJMTISnW1YbzNvaW_iesxH1nWr8FHG-8CL8,12550
+langchain_dev_utils/chat_models/base.py,sha256=AYRcGViGJYsquqru_www3zt8-ZCkfzPCrw-dFF6HDts,11661
 langchain_dev_utils/chat_models/types.py,sha256=M0iCGWgXmX1f1vkymH-jNGdFQlsJS5JqpmgHctUS9jw,512
 langchain_dev_utils/chat_models/adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langchain_dev_utils/chat_models/adapters/openai_compatible.py,sha256=4Q8ySa7jS2_AFo0oxLoqeY_aQyPppvV-DAMLt2rmGoE,20192
 langchain_dev_utils/embeddings/__init__.py,sha256=zbEOaV86TUi9Zrg_dH9dpdgacWg31HMJTlTQknA9EKk,244
-langchain_dev_utils/embeddings/base.py,sha256=25ebUEaf7075h8NARgTHjAvwK6JddhHor5upiucJqu0,9686
+langchain_dev_utils/embeddings/base.py,sha256=lGZWbi6G1M0OcAO_d_k1QAFJm9z9gM0L4UAZ6xFtEoQ,8973
 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 +30,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.2.5.dist-info/METADATA,sha256=LB3TTtW0Hb6vopD23t0wiziRUsSFroD0-wsp1r8ipjg,18822
-langchain_dev_utils-1.2.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-langchain_dev_utils-1.2.5.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
-langchain_dev_utils-1.2.5.dist-info/RECORD,,
+langchain_dev_utils-1.2.6.dist-info/METADATA,sha256=PhJoxRlERmnMzRumGzLW3kwU58Cg3Tt6MiFibsfIi8U,19090
+langchain_dev_utils-1.2.6.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+langchain_dev_utils-1.2.6.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
+langchain_dev_utils-1.2.6.dist-info/RECORD,,