From 07f4c65bdea557078e74bd38363f71dc7b5431d0 Mon Sep 17 00:00:00 2001 From: Pavan Kumar <66913595+ascender1729@users.noreply.github.com> Date: Fri, 18 Apr 2025 15:39:31 +0000 Subject: [PATCH] style(agent): simplify output format by removing agent distinction in results --- examples/mcp_example/mock_multi_agent.py | 26 ++++++++++++++++-------- 1 file changed, 17 insertions(+), 9 deletions(-) diff --git a/examples/mcp_example/mock_multi_agent.py b/examples/mcp_example/mock_multi_agent.py index 17ffeb04..bed8eee0 100644 --- a/examples/mcp_example/mock_multi_agent.py +++ b/examples/mcp_example/mock_multi_agent.py @@ -137,23 +137,31 @@ class MultiAgentMathSystem: results = asyncio.run(self.process_task(user_input)) - print("\nResults:") - print("="*50) + responses = [] for result in results: - if result["response"] is not None: # Only show responses from relevant agents + if result["response"] is not None: if "error" in result: - print(f"Error: {result['error']}") + responses.append(f"Error: {result['error']}") else: - # Only show the actual calculation result, not the system information response = result["response"] if isinstance(response, str): - # Remove system information and keep only the calculation part + # Clean up calculation results if "=" in response: calculation = response.split("=")[-1].strip() - print(f"Result: {calculation}") + responses.append(calculation) else: - print(response) - print("="*50) + # Remove system/agent information + clean_response = response.split("System:")[0].strip() + clean_response = clean_response.split("Human:")[0].strip() + if clean_response: + responses.append(clean_response) + + if responses: + print("\nResult:") + print("-"*30) + for response in responses: + print(response) + print("-"*30) except Exception as e: print(f"System error: {str(e)}")