From 3baca53eba0ff7259b6c7597e58a40453302ef65 Mon Sep 17 00:00:00 2001
From: Yuval Turgeman <yturgema@redhat.com>
Date: Thu, 18 Sep 2025 16:33:25 -0400
Subject: [PATCH] Extend OpenAIResponseInput with MCP types

When posting chained resposnes with MCP tools, llamastack fails
to validate OpenAIResponseInput.

Signed-off-by: Yuval Turgeman <yturgema@redhat.com>
---
 llama_stack/apis/agents/openai_responses.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/llama_stack/apis/agents/openai_responses.py b/llama_stack/apis/agents/openai_responses.py
index 591992479..9efb3d211 100644
--- a/llama_stack/apis/agents/openai_responses.py
+++ b/llama_stack/apis/agents/openai_responses.py
@@ -724,6 +724,8 @@ OpenAIResponseInput = Annotated[
     OpenAIResponseOutputMessageWebSearchToolCall
     | OpenAIResponseOutputMessageFileSearchToolCall
     | OpenAIResponseOutputMessageFunctionToolCall
+    | OpenAIResponseOutputMessageMCPCall
+    | OpenAIResponseOutputMessageMCPListTools
     | OpenAIResponseInputFunctionToolCallOutput
     |
     # Fallback to the generic message type as a last resort