Update app.py

app.py

@@ -7,8 +7,7 @@ app = FastAPI()
 
 # Define the request model
 class ChatRequest(BaseModel):
-    system_prompt: str
-    user_query: str
+    messages: list = []
     model: str = "gemini-1.5-pro-latest"
     temperature: float = 1.0
     top_p: float = 0.8
@@ -26,16 +25,7 @@ headers = {
 async def chat(request: ChatRequest):
     # Define the payload
     payload = {
-        "messages": [
-            {
-                "role": "system",
-                "content": request.system_prompt
-            },
-            {
-                "role": "user",
-                "content": request.user_query
-            }
-        ],
+        "messages": request.messages,
         "stream": True,
         "model": request.model,
         "temperature": request.temperature,