gemini-pro:generateContent

App/Chat/utils/RAG.py

@@ -47,7 +47,7 @@ class GenerativeAIAssistant:
         latest_message = messages[-1]['content']
         latest_message={"content":self.generate_template(latest_message,task_id)}
         messages[-1]=latest_message
-        url = f'https://generativelanguage.googleapis.com/v1beta3/models/{self.model}:generateMessage?key={self.api_key}'
+        url = f'https://generativelanguage.googleapis.com/v1beta/models/{self.model}:generateMessage?key={self.api_key}'
         data = {
             "prompt": {
                 "context": self.context,

App/Chat/utils/Summarize.py

@@ -46,7 +46,7 @@ def count_tokens(text):
 
 
 async def PalmTextModel(text, candidates=1):
-    url = f"https://generativelanguage.googleapis.com/v1beta2/models/text-bison-001:generateText?key={API_KEY}"
+    url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key={API_KEY}"
 
     headers = {
         "Content-Type": "application/json",