From 9faf83b83b08ff05ff724197614fff6e2beb12e7 Mon Sep 17 00:00:00 2001
From: Pavan Kumar <66913595+ascender1729@users.noreply.github.com>
Date: Thu, 17 Apr 2025 17:16:58 +0000
Subject: [PATCH] fix: set model_name='gpt-4o-mini' for all agents to suppress
warnings
- Updated all agents in `multi_server_test.py` to explicitly define model name
- Eliminated LiteLLM fallback warnings during runtime
- Ensured proper agent responses in multi-agent MCP test environment
---
examples/mcp_example/multi_server_test.py | 11 ++++++-----
1 file changed, 6 insertions(+), 5 deletions(-)
diff --git a/examples/mcp_example/multi_server_test.py b/examples/mcp_example/multi_server_test.py
index d112d311..de1144f9 100644
--- a/examples/mcp_example/multi_server_test.py
+++ b/examples/mcp_example/multi_server_test.py
@@ -1,4 +1,3 @@
-
from swarms import Agent
from swarms.tools.mcp_integration import MCPServerSseParams
from swarms.prompts.finance_agent_sys_prompt import FINANCIAL_AGENT_SYS_PROMPT
@@ -37,7 +36,8 @@ finance_agent = Agent(
max_loops=1,
mcp_servers=[calc_server],
interactive=True,
- streaming_on=True
+ streaming_on=True,
+ model_name="gpt-4o-mini"
)
# Multi-server agent with access to all operations
@@ -48,7 +48,8 @@ super_agent = Agent(
max_loops=1,
mcp_servers=[math_server, calc_server],
interactive=True,
- streaming_on=True
+ streaming_on=True,
+ model_name="gpt-4o-mini"
)
def main():
@@ -58,7 +59,7 @@ def main():
while True:
try:
user_input = input("\nEnter your calculation request: ")
-
+
if user_input.lower() == 'exit':
break
@@ -80,4 +81,4 @@ def main():
print(f"Error processing request: {e}")
if __name__ == "__main__":
- main()
+ main()
\ No newline at end of file