diff --git a/libs/partners/openai/tests/integration_tests/chat_models/test_base.py b/libs/partners/openai/tests/integration_tests/chat_models/test_base.py
index 4b02676e2a8..0417bbc32ed 100644
--- a/libs/partners/openai/tests/integration_tests/chat_models/test_base.py
+++ b/libs/partners/openai/tests/integration_tests/chat_models/test_base.py
@@ -473,6 +473,7 @@ def test_manual_tool_call_msg(use_responses_api: bool) -> None:
                     name="GenerateUsername",
                     args={"name": "Sally", "hair_color": "green"},
                     id="foo",
+                    type="tool_call",
                 )
             ],
         ),
@@ -494,6 +495,7 @@ def test_manual_tool_call_msg(use_responses_api: bool) -> None:
                     name="GenerateUsername",
                     args={"name": "Sally", "hair_color": "green"},
                     id="bar",
+                    type="tool_call",
                 )
             ],
         ),