Update app.py

app.py

@@ -263,12 +263,26 @@ async def chat_completion(request: ChatRequest, authorization: str = Header(None
                     return response.model_dump()
             else:
                 gemini_messages = convert_messages_to_gemini_format(request.messages)
-                payload = {
-                    "contents": gemini_messages,
-                    "generationConfig": {
-                        "temperature": request.temperature,
+                payload = {}
+                if request.model == "gemini-2.0-flash-exp-search":
+                    payload = {
+                        "contents": gemini_messages,
+                        "generationConfig": {
+                            "temperature": request.temperature,
+                        },
+                        "tools": [
+                              {
+                                  "google_search": {}
+                              }
+                        ]
+                    }
+                else:
+                    payload = {
+                        "contents": gemini_messages,
+                        "generationConfig": {
+                            "temperature": request.temperature,
+                        }
                     }
-                }
 
                 if request.stream:
                     logger.info("Streaming response enabled")