openai: fix allowed block types (#20636)

langchain-ai · Apr 26, 2024 · 336947c · 336947c
1 parent ad75820
commit 336947c
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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)