Update model.py

model.py

@@ -52,11 +52,12 @@ class LiteLLMModel:
             
             # Call LiteLLM
             response = litellm.completion(
-                model=self.model_id,
-                messages=messages,
-                tools=formatted_tools,
-                temperature=0.1
-            )
+    model=f"google/{self.model_id}" if "gemini" in self.model_id.lower() else self.model_id,
+    messages=messages,
+    tools=formatted_tools,
+    temperature=0.1
+)
+
             
             message = response.choices[0].message