From d30ca92421287e4e17eb1fab537359fad98597b4 Mon Sep 17 00:00:00 2001
From: Pavan Kumar <66913595+ascender1729@users.noreply.github.com>
Date: Thu, 17 Apr 2025 16:55:17 +0000
Subject: [PATCH] chore: set model name to 'gpt-4o-mini' in server and test
 configs

- Updated `math_server.py` and `test_integration.py` to explicitly use 'gpt-4o-mini'
- Ensures consistent model configuration across test and runtime environments
---
 examples/mcp_example/math_server.py      | 4 +++-
 examples/mcp_example/test_integration.py | 3 ++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/examples/mcp_example/math_server.py b/examples/mcp_example/math_server.py
index ca7eaf2e..4f3b111b 100644
--- a/examples/mcp_example/math_server.py
+++ b/examples/mcp_example/math_server.py
@@ -36,4 +36,6 @@ def calculate_percentage(part: float, whole: float) -> float:
 
 if __name__ == "__main__":
     print("Starting Math Server on port 6274...")
-    mcp.run(transport="sse", host="0.0.0.0", port=6274)
+    # Initialize LiteLLM with specific model
+llm = LiteLLM(model_name="gpt-4o-mini")
+mcp.run(transport="sse", host="0.0.0.0", port=6274)
diff --git a/examples/mcp_example/test_integration.py b/examples/mcp_example/test_integration.py
index 0a521499..edce69db 100644
--- a/examples/mcp_example/test_integration.py
+++ b/examples/mcp_example/test_integration.py
@@ -21,7 +21,8 @@ def setup_agent(name: str, description: str, servers: list) -> Agent:
         system_prompt="You are a math assistant. Process mathematical operations using the provided MCP tools.",
         max_loops=1,
         mcp_servers=servers,
-        streaming_on=False
+        streaming_on=False,
+        model_name="gpt-4o-mini" # Added model_name here
     )
 
 def main():