Update app.py

app.py

@@ -38,16 +38,7 @@ class GeminiLLM(LLM):
         data = {
             "contents": [
                 {
-                    "role": "system",
-                    "content": {
-                        "parts": ["You are an agent. Please respond concisely."]
-                    }
-                },
-                {
-                    "role": "user",
-                    "content": {
-                        "parts": [prompt]
-                    }
+                    "parts": [{"text": prompt}]
                 }
             ],
             "generationConfig": {