langchain-dev-utils 1.2.12__py3-none-any.whl → 1.2.13__py3-none-any.whl

langchain_dev_utils/__init__.py

@@ -1 +1 @@
-__version__ = "1.2.12"
+__version__ = "1.2.13"

langchain_dev_utils/chat_models/adapters/openai_compatible.py

@@ -20,7 +20,13 @@ from langchain_core.callbacks import (
     CallbackManagerForLLMRun,
 )
 from langchain_core.language_models import LangSmithParams, LanguageModelInput
-from langchain_core.messages import AIMessage, AIMessageChunk, BaseMessage, HumanMessage
+from langchain_core.messages import (
+    AIMessage,
+    AIMessageChunk,
+    BaseMessage,
+    HumanMessage,
+    ToolMessage,
+)
 from langchain_core.outputs import ChatGenerationChunk, ChatResult
 from langchain_core.runnables import Runnable
 from langchain_core.tools import BaseTool
@@ -63,6 +69,58 @@ def _get_last_human_message_index(messages: list[BaseMessage]) -> int:
     )
 
 
+def _transform_video_block(block: dict[str, Any]) -> dict:
+    """Transform video block to video_url block."""
+    if "url" in block:
+        return {
+            "type": "video_url",
+            "video_url": {
+                "url": block["url"],
+            },
+        }
+    if "base64" in block or block.get("source_type") == "base64":
+        if "mime_type" not in block:
+            error_message = "mime_type key is required for base64 data."
+            raise ValueError(error_message)
+        mime_type = block["mime_type"]
+        base64_data = block["data"] if "data" in block else block["base64"]
+        return {
+            "type": "video_url",
+            "video_url": {
+                "url": f"data:{mime_type};base64,{base64_data}",
+            },
+        }
+    error_message = "Unsupported source type. Only 'url' and 'base64' are supported."
+    raise ValueError(error_message)
+
+
+def _process_video_input(message: BaseMessage):
+    """
+    Process BaseMessage with video input.
+
+    Args:
+        message (BaseMessage): The HumanMessage instance to process.
+
+    Returns:
+        None: The method modifies the message in-place.
+    """
+    if not message.content:
+        return message
+    content = message.content
+
+    if not isinstance(content, list):
+        return message
+
+    formatted_content = []
+    for block in content:
+        if isinstance(block, dict) and block.get("type") == "video":
+            formatted_content.append(_transform_video_block(block))
+        else:
+            formatted_content.append(block)
+    message = message.model_copy(update={"content": formatted_content})
+    return message
+
+
 class _BaseChatOpenAICompatible(BaseChatOpenAI):
     """
     Base template class for OpenAI-compatible chat model implementations.
@@ -183,6 +241,10 @@ class _BaseChatOpenAICompatible(BaseChatOpenAI):
                     )
                 payload_messages.append(msg_dict)
             else:
+                if (
+                    isinstance(m, HumanMessage) or isinstance(m, ToolMessage)
+                ) and isinstance(m.content, list):
+                    m = _process_video_input(m)
                 payload_messages.append(_convert_message_to_dict(m))
 
         payload["messages"] = payload_messages

langchain_dev_utils/chat_models/base.py

@@ -11,7 +11,6 @@ from langchain_dev_utils._utils import (
 
 from .types import ChatModelProvider, ChatModelType, CompatibilityOptions
 
-
 _MODEL_PROVIDERS_DICT = {}
 
 

{langchain_dev_utils-1.2.12.dist-info → langchain_dev_utils-1.2.13.dist-info}/METADATA

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: langchain-dev-utils
-Version: 1.2.12
+Version: 1.2.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
-Project-URL: documentation, https://tbice123123.github.io/langchain-dev-utils-docs/en/
+Project-URL: documentation, https://tbice123123.github.io/langchain-dev-utils
 Author-email: tiebingice <tiebingice123@outlook.com>
 License-File: LICENSE
 Requires-Python: >=3.11

{langchain_dev_utils-1.2.12.dist-info → langchain_dev_utils-1.2.13.dist-info}/RECORD

@@ -1,4 +1,4 @@
-langchain_dev_utils/__init__.py,sha256=dbW85A2PinQCZabwD2DNDTfOE9315GDtQQKAsJP8IXk,23
+langchain_dev_utils/__init__.py,sha256=rQSlPcfj4yT4krIq6epTVQyBzIX4etVOgfupVkM-RnU,23
 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,10 +16,10 @@ langchain_dev_utils/agents/middleware/tool_call_repair.py,sha256=oZF0Oejemqs9kSn
 langchain_dev_utils/agents/middleware/tool_emulator.py,sha256=OgtPhqturaWzF4fRSJ3f_IXvIrYrrAjlpOC5zmLtrkY,2031
 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=t-tjNMX4BSyNxvBCLX-BKfJzJcnzhUzJY4NTHhVhnPA,11627
+langchain_dev_utils/chat_models/base.py,sha256=BzaoCIv145eE8b5wNDsbZDHn4EAxe4vdlptp7qXPWKk,11625
 langchain_dev_utils/chat_models/types.py,sha256=MD3cv_ZIe9fCdgwisNfuxAOhy-j4YSs1ZOQYyCjlNKs,927
 langchain_dev_utils/chat_models/adapters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langchain_dev_utils/chat_models/adapters/openai_compatible.py,sha256=v9ewQFn75Fl6urtVo4fZgg2RRrzRXGluKf-egYAI5A0,22658
+langchain_dev_utils/chat_models/adapters/openai_compatible.py,sha256=4iJgMGAReiJ668O9ZGAZbduxNY2PhEJJonDhQDBge44,24596
 langchain_dev_utils/embeddings/__init__.py,sha256=zbEOaV86TUi9Zrg_dH9dpdgacWg31HMJTlTQknA9EKk,244
 langchain_dev_utils/embeddings/base.py,sha256=BGoWY0L7nG9iRV3d4sSagXhECXrwvS1xA-A_OVltn3k,9406
 langchain_dev_utils/message_convert/__init__.py,sha256=ZGrHGXPKMrZ_p9MqfIVZ4jgbEyb7aC4Q7X-muuThIYU,457
@@ -32,7 +32,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=7Z_QO5OZUR6K8nLoIcafc6osnvX2IYNorOJcbx6bVso,9672
 langchain_dev_utils/tool_calling/utils.py,sha256=S4-KXQ8jWmpGTXYZitovF8rxKpaSSUkFruM8LDwvcvE,2765
-langchain_dev_utils-1.2.12.dist-info/METADATA,sha256=5MjLniCOhGN8a0L6cNLD69PUMNzf5K0nS4Cfifdl5CI,11853
-langchain_dev_utils-1.2.12.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-langchain_dev_utils-1.2.12.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
-langchain_dev_utils-1.2.12.dist-info/RECORD,,
+langchain_dev_utils-1.2.13.dist-info/METADATA,sha256=SUHFdkF4J5_VxYmZGL7Aqv_79JTXRBoCn--gMTrSjHw,11844
+langchain_dev_utils-1.2.13.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+langchain_dev_utils-1.2.13.dist-info/licenses/LICENSE,sha256=AWAOzNEcsvCEzHOF0qby5OKxviVH_eT9Yce1sgJTico,1084
+langchain_dev_utils-1.2.13.dist-info/RECORD,,