Upload main.py

main.py

@@ -566,9 +566,7 @@ def v1beta2_google_ai(
         data = {
             "prompt": {
                 "messages": [
-                    {
-                        "content": query
-                    }
+                    {"content": query}
                 ]
             },
             "temperature": temperature,