Hello

App/Chat/utils/RAG.py

@@ -44,24 +44,62 @@ class GenerativeAIAssistant:
         return Template
 
     async def generate_message(self,  messages,task_id):
-        latest_message = messages[-1]['content']
+        user_message=messages[-1]
+        latest_message = messages[-1]['content']['parts'][0]['text']
         latest_message={"content":self.generate_template(latest_message,task_id)}
-        messages[-1]=latest_message
+        user_message['content']['parts'][0]['text']=latest_message
+        messages[-1]=user_message
         url = f'https://generativelanguage.googleapis.com/v1beta/models/{self.model}:generateContent?key={self.api_key}'
-        data = {
-            "prompt": {
-                "context": self.context,
-                "examples": self.examples,
-                "messages": messages
+
+
+        payload = {
+        "contents":messages,
+        "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"
             },
-            "temperature": self.temperature,
-            "top_k": self.top_k,
-            "top_p": self.top_p,
-            "candidate_count": self.candidate_count
-        }
+            {
+                "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+                "threshold": "BLOCK_ONLY_HIGH"
+            },
+            {
+                "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+                "threshold": "BLOCK_ONLY_HIGH"
+            }
+        ]
+    }
+
+
+
+
+
+
+        # data = {
+        #     "prompt": {
+        #         "context": self.context,
+        #         "examples": self.examples,
+        #         "messages": messages
+        #     },
+        #     "temperature": self.temperature,
+        #     "top_k": self.top_k,
+        #     "top_p": self.top_p,
+        #     "candidate_count": self.candidate_count
+        # }
 
         async with aiohttp.ClientSession() as session:
-            async with session.post(url, json=data, headers={'Content-Type': 'application/json'}) as response:
+            async with session.post(url, json=payload, headers={'Content-Type': 'application/json'}) as response:
                 try:
                     temp=  await response.json()
                     return temp["candidates"][0]["content"]['parts'][0]['text']

App/Chat/utils/Summarize.py

@@ -93,7 +93,7 @@ async def PalmTextModel(text, candidates=1):
 
 
     async with aiohttp.ClientSession() as session:
-        async with session.post(url, json=data, headers=headers) as response:
+        async with session.post(url, json=payload, headers=headers) as response:
             if response.status == 200:
                 result = await response.json()
                 # print(result)