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
pull/819/head
Pavan Kumar 2 days ago committed by ascender1729
parent db3746131b
commit 9faf83b83b

@ -1,4 +1,3 @@
from swarms import Agent from swarms import Agent
from swarms.tools.mcp_integration import MCPServerSseParams from swarms.tools.mcp_integration import MCPServerSseParams
from swarms.prompts.finance_agent_sys_prompt import FINANCIAL_AGENT_SYS_PROMPT from swarms.prompts.finance_agent_sys_prompt import FINANCIAL_AGENT_SYS_PROMPT
@ -37,7 +36,8 @@ finance_agent = Agent(
max_loops=1, max_loops=1,
mcp_servers=[calc_server], mcp_servers=[calc_server],
interactive=True, interactive=True,
streaming_on=True streaming_on=True,
model_name="gpt-4o-mini"
) )
# Multi-server agent with access to all operations # Multi-server agent with access to all operations
@ -48,7 +48,8 @@ super_agent = Agent(
max_loops=1, max_loops=1,
mcp_servers=[math_server, calc_server], mcp_servers=[math_server, calc_server],
interactive=True, interactive=True,
streaming_on=True streaming_on=True,
model_name="gpt-4o-mini"
) )
def main(): def main():
@ -58,7 +59,7 @@ def main():
while True: while True:
try: try:
user_input = input("\nEnter your calculation request: ") user_input = input("\nEnter your calculation request: ")
if user_input.lower() == 'exit': if user_input.lower() == 'exit':
break break
@ -80,4 +81,4 @@ def main():
print(f"Error processing request: {e}") print(f"Error processing request: {e}")
if __name__ == "__main__": if __name__ == "__main__":
main() main()
Loading…
Cancel
Save