["candidates"][0]["content"]['parts'][0]['text']

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/v1beta/models/{self.model}:generateMessage?key={self.api_key}'
+        url = f'https://generativelanguage.googleapis.com/v1beta/models/{self.model}:generateContent?key={self.api_key}'
         data = {
             "prompt": {
                 "context": self.context,
@@ -64,7 +64,7 @@ class GenerativeAIAssistant:
             async with session.post(url, json=data, headers={'Content-Type': 'application/json'}) as response:
                 try:
                     temp=  await response.json()
-                    return temp['candidates']
+                    return temp["candidates"][0]["content"]['parts'][0]['text']
                 except Exception as e:
                     return f"Error ⚠️ {e}"
                 

App/Chat/utils/Summarize.py

@@ -52,33 +52,55 @@ async def PalmTextModel(text, candidates=1):
         "Content-Type": "application/json",
     }
 
-    data = {
-        "prompt": {"text": text},
-        "temperature": 0.95,
-        "top_k": 100,
-        "top_p": 0.95,
-        "candidate_count": candidates,
-        "max_output_tokens": 1024,
-        "stop_sequences": ["</output>"],
-        "safety_settings": [
-            {"category": "HARM_CATEGORY_DEROGATORY", "threshold": 4},
-            {"category": "HARM_CATEGORY_TOXICITY", "threshold": 4},
-            {"category": "HARM_CATEGORY_VIOLENCE", "threshold": 4},
-            {"category": "HARM_CATEGORY_SEXUAL", "threshold": 4},
-            {"category": "HARM_CATEGORY_MEDICAL", "threshold": 4},
-            {"category": "HARM_CATEGORY_DANGEROUS", "threshold": 4},
+    payload = {
+        "contents": [
+            {
+                "role": "user",
+                "parts": [
+                    {
+                        "text": text
+                    }
+                ]
+            }
         ],
+        "generationConfig": {
+            "temperature": 0.9,
+            "topK": 1,
+            "topP": 1,
+            "maxOutputTokens": 2048,
+            "stopSequences": []
+        },
+        "safetySettings": [
+            {
+                "category": "HARM_CATEGORY_HARASSMENT",
+                "threshold": "BLOCK_ONLY_HIGH"
+            },
+            {
+                "category": "HARM_CATEGORY_HATE_SPEECH",
+                "threshold": "BLOCK_ONLY_HIGH"
+            },
+            {
+                "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+                "threshold": "BLOCK_ONLY_HIGH"
+            },
+            {
+                "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+                "threshold": "BLOCK_ONLY_HIGH"
+            }
+        ]
     }
 
+
+
     async with aiohttp.ClientSession() as session:
         async with session.post(url, json=data, headers=headers) as response:
             if response.status == 200:
                 result = await response.json()
                 # print(result)
                 if candidates > 1:
-                    temp = [candidate["output"] for candidate in result["candidates"]]
+                    temp = [candidate["content"]['parts'][0]['text'] for candidate in result["candidates"]]
                     return temp
-                temp = result["candidates"][0]["output"]
+                temp = result["candidates"][0]["content"]['parts'][0]['text']
                 return temp
             else:
                 print(f"Error: {response.status}\n{await response.text()}")