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 126c2e193..f08bdb032 100644 --- a/llama_stack/providers/inline/agents/meta_reference/agent_instance.py +++ b/llama_stack/providers/inline/agents/meta_reference/agent_instance.py @@ -185,9 +185,9 @@ class ChatAgent(ShieldRunnerMixin): stream=request.stream, ): if isinstance(chunk, CompletionMessage): - log.info( - f"{chunk.role.capitalize()}: {chunk.content}", - ) + # log.info( + # f"{chunk.role.capitalize()}: {chunk.content}", + # ) output_message = chunk continue @@ -405,11 +405,11 @@ class ChatAgent(ShieldRunnerMixin): n_iter = 0 while True: msg = input_messages[-1] - if len(str(msg)) > 1000: - msg_str = f"{str(msg)[:500]}......{str(msg)[-500:]}" - else: - msg_str = str(msg) - log.info(f"{msg_str}") + # if len(str(msg)) > 1000: + # msg_str = f"{str(msg)[:500]}......{str(msg)[-500:]}" + # else: + # msg_str = str(msg) + # log.info(f"{msg_str}") step_id = str(uuid.uuid4()) yield AgentTurnResponseStreamChunk( @@ -514,12 +514,12 @@ class ChatAgent(ShieldRunnerMixin): ) if n_iter >= self.agent_config.max_infer_iters: - log.info("Done with MAX iterations, exiting.") + # log.info("Done with MAX iterations, exiting.") yield message break if stop_reason == StopReason.out_of_tokens: - log.info("Out of token budget, exiting.") + # log.info("Out of token budget, exiting.") yield message break @@ -533,10 +533,10 @@ class ChatAgent(ShieldRunnerMixin): message.content = [message.content] + attachments yield message else: - log.info(f"Partial message: {str(message)}") + # log.info(f"Partial message: {str(message)}") input_messages = input_messages + [message] else: - log.info(f"{str(message)}") + # log.info(f"{str(message)}") try: tool_call = message.tool_calls[0] @@ -800,7 +800,7 @@ async def attachment_message(tempdir: str, urls: List[URL]) -> ToolResponseMessa path = urlparse(uri).path basename = os.path.basename(path) filepath = f"{tempdir}/{make_random_string() + basename}" - log.info(f"Downloading {url} -> {filepath}") + # log.info(f"Downloading {url} -> {filepath}") async with httpx.AsyncClient() as client: r = await client.get(uri)