From 5644d10c82f98ff1e61c251824182f98db0d81f6 Mon Sep 17 00:00:00 2001
From: Xi Yan <xiyan@meta.com>
Date: Thu, 20 Feb 2025 15:26:37 -0800
Subject: [PATCH] remove usermessages

---
 docs/_static/llama-stack-spec.html                  | 13 +++----------
 docs/_static/llama-stack-spec.yaml                  |  8 +++-----
 llama_stack/apis/agents/agents.py                   |  7 +------
 .../inline/agents/meta_reference/agents.py          |  7 +------
 4 files changed, 8 insertions(+), 27 deletions(-)

diff --git a/docs/_static/llama-stack-spec.html b/docs/_static/llama-stack-spec.html
index 32cbd60f5..d2b3529df 100644
--- a/docs/_static/llama-stack-spec.html
+++ b/docs/_static/llama-stack-spec.html
@@ -3954,23 +3954,16 @@
             "ContinueAgentTurnRequest": {
                 "type": "object",
                 "properties": {
-                    "new_messages": {
+                    "tool_responses": {
                         "type": "array",
                         "items": {
-                            "oneOf": [
-                                {
-                                    "$ref": "#/components/schemas/UserMessage"
-                                },
-                                {
-                                    "$ref": "#/components/schemas/ToolResponseMessage"
-                                }
-                            ]
+                            "$ref": "#/components/schemas/ToolResponseMessage"
                         }
                     }
                 },
                 "additionalProperties": false,
                 "required": [
-                    "new_messages"
+                    "tool_responses"
                 ],
                 "title": "ContinueAgentTurnRequest"
             },
diff --git a/docs/_static/llama-stack-spec.yaml b/docs/_static/llama-stack-spec.yaml
index 139f51507..121098a54 100644
--- a/docs/_static/llama-stack-spec.yaml
+++ b/docs/_static/llama-stack-spec.yaml
@@ -2573,15 +2573,13 @@ components:
     ContinueAgentTurnRequest:
       type: object
       properties:
-        new_messages:
+        tool_responses:
           type: array
           items:
-            oneOf:
-              - $ref: '#/components/schemas/UserMessage'
-              - $ref: '#/components/schemas/ToolResponseMessage'
+            $ref: '#/components/schemas/ToolResponseMessage'
       additionalProperties: false
       required:
-        - new_messages
+        - tool_responses
       title: ContinueAgentTurnRequest
     InferenceStep:
       type: object
diff --git a/llama_stack/apis/agents/agents.py b/llama_stack/apis/agents/agents.py
index 05ddf587a..c9554098f 100644
--- a/llama_stack/apis/agents/agents.py
+++ b/llama_stack/apis/agents/agents.py
@@ -354,12 +354,7 @@ class Agents(Protocol):
         agent_id: str,
         session_id: str,
         turn_id: str,
-        new_messages: List[
-            Union[
-                UserMessage,
-                ToolResponseMessage,
-            ]
-        ],
+        tool_responses: List[ToolResponseMessage],
     ) -> Union[Turn, AsyncIterator[AgentTurnResponseStreamChunk]]: ...
 
     @webmethod(
diff --git a/llama_stack/providers/inline/agents/meta_reference/agents.py b/llama_stack/providers/inline/agents/meta_reference/agents.py
index cb8192968..d34fd0b57 100644
--- a/llama_stack/providers/inline/agents/meta_reference/agents.py
+++ b/llama_stack/providers/inline/agents/meta_reference/agents.py
@@ -174,12 +174,7 @@ class MetaReferenceAgentsImpl(Agents):
         agent_id: str,
         session_id: str,
         turn_id: str,
-        new_messages: List[
-            Union[
-                UserMessage,
-                ToolResponseMessage,
-            ]
-        ],
+        tool_responses: List[ToolResponseMessage],
     ) -> AsyncGenerator:
         pass