Update models.py

models.py

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