Update app.py

app.py

@@ -39,16 +39,22 @@ class GeminiLLM(LLM):
             "contents": [
                 {
                     "role": "system",
-                    "text": "You are an agent. Please respond concisely."
+                    "content": {
+                        "parts": ["You are an agent. Please respond concisely."]
+                    }
                 },
                 {
-                    "parts": [{"text": prompt}]
+                    "role": "user",
+                    "content": {
+                        "parts": [prompt]
+                    }
                 }
             ],
             "generationConfig": {
                 "temperature": self.temperature
             }
         }
+
         return requests.post(url, headers=headers, json=data)
 
     def _call(self, prompt: str, stop: Optional[List[str]] = None) -> str: