Final_Assignment_Template1

Sleeping

Mehedi2 commited on Oct 1, 2025

Commit

4fef00e

verified ·

1 Parent(s): 107e713

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -15,7 +15,7 @@ load_dotenv()
 # Initialize the Gemini model using Google AI Studio API key
 model = LiteLLMModel(
-    model_id="vertex_ai/gemini/google/gemini-2.5-flash-preview-09-2025",
     api_key=os.getenv("GEMINI_API_KEY"),
 )

 # Initialize the Gemini model using Google AI Studio API key
 model = LiteLLMModel(
+    model_id="google_ai/gemini-2.5-flash-preview-09-2025",
     api_key=os.getenv("GEMINI_API_KEY"),
 )