Update models.py

models.py

@@ -72,6 +72,16 @@ class ChatgptCustom(BaseModel):
     query: str
 
 class GeminiPro(BaseModel):
+    query: str
+    bard_api_key: Optional[str] = None
+    mongo_url: Optional[str] = None
+    user_id: Optional[int] = None
+    version: Optional[str] = "v1beta"
+    gemini_api_key: Optional[str] = None
+    is_login: Optional[bool] = False
+    is_multi_chat: Optional[bool] = False
+
+class GeminiOracle(BaseModel):
     query: str
     bard_api_key: Optional[str] = None
     mongo_url: Optional[str] = None