From a99706ec5e1b50ec4a48e2dbfda00a2dbe6517b0 Mon Sep 17 00:00:00 2001
From: Xi Yan <xiyan@meta.com>
Date: Thu, 27 Feb 2025 14:01:18 -0800
Subject: [PATCH] fix

---
 .../agents/meta_reference/agent_instance.py   | 24 -------------------
 1 file changed, 24 deletions(-)

diff --git a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py
index eb3257bce..6c02b81c4 100644
--- a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py
+++ b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py
@@ -17,7 +17,6 @@ from typing import Any, AsyncGenerator, Dict, List, Optional, Tuple
 from urllib.parse import urlparse
 
 import httpx
-from rich.pretty import pprint
 
 from llama_stack.apis.agents import (
     AgentConfig,
@@ -187,19 +186,9 @@ class ChatAgent(ShieldRunnerMixin):
 
             turns = await self.storage.get_session_turns(request.session_id)
 
-            print("!! create and execute turn turns", len(turns))
-            pprint(turns)
-
             messages = await self.get_messages_from_turns(turns)
-
-            print("!! create and execute turn messages", len(messages))
-            pprint(messages)
-
             messages.extend(request.messages)
 
-            print("!! create and execute turn messages extended", len(messages))
-            pprint(messages)
-
             turn_id = str(uuid.uuid4())
             span.set_attribute("turn_id", turn_id)
             start_time = datetime.now().astimezone().isoformat()
@@ -284,22 +273,15 @@ class ChatAgent(ShieldRunnerMixin):
             if len(turns) == 0:
                 raise ValueError("No turns found for session")
 
-            pprint("!! resume turn turns")
-            pprint(turns)
             messages = await self.get_messages_from_turns(turns)
             messages.extend(request.tool_responses)
 
-            print("!! resume turn")
-            pprint(messages)
-
             last_turn = turns[-1]
             last_turn_messages = self.turn_to_messages(last_turn)
             last_turn_messages = [
                 x for x in last_turn_messages if isinstance(x, UserMessage) or isinstance(x, ToolResponseMessage)
             ]
 
-            print("last turn messages")
-            pprint(last_turn_messages)
             # TODO: figure out whether we should add the tool responses to the last turn messages
             last_turn_messages.extend(request.tool_responses)
 
@@ -405,9 +387,6 @@ class ChatAgent(ShieldRunnerMixin):
         documents: Optional[List[Document]] = None,
         toolgroups_for_turn: Optional[List[AgentToolGroup]] = None,
     ) -> AsyncGenerator:
-        print("!!RUN input messages")
-
-        pprint(input_messages)
         # Doing async generators makes downstream code much simpler and everything amenable to
         # streaming. However, it also makes things complicated here because AsyncGenerators cannot
         # return a "final value" for the `yield from` statement. we simulate that by yielding a
@@ -452,9 +431,6 @@ class ChatAgent(ShieldRunnerMixin):
                 else:
                     yield res
 
-        pprint("!!RUN final response")
-        pprint(messages)
-
         yield final_response
 
     async def run_multiple_shields_wrapper(