openai: fix allowed block types (#20636)

6 months ago · 6d530481c1
parent 764871f97d
commit 6d530481c1
1 changed files with 5 additions and 1 deletions
--- a/libs/partners/openai/langchain_openai/chat_models/base.py
+++ b/libs/partners/openai/langchain_openai/chat_models/base.py
@ -154,7 +154,11 @@ def _format_message_content(content: Any) -> Any:
        # Remove unexpected block types
        formatted_content = []
        for block in content:
-            if isinstance(block, dict) and "type" in block and block["type"] != "text":
+            if (
+                isinstance(block, dict)
+                and "type" in block
+                and block["type"] == "tool_use"
+            ):
                continue
            else:
                formatted_content.append(block)