updated chat

App/Chat/utils/RAG.py

@@ -3,7 +3,7 @@ import asyncio
 import json,os
 import yaml
 import google.generativeai as palm
-from App.Embedding.utils.Initialize import search
+# from App.Embedding.utils.Initialize import search
 PALM_API = ""
 API_KEY = os.environ.get("PALM_API", PALM_API)
 palm.configure(api_key=API_KEY)
@@ -43,11 +43,11 @@ class GenerativeAIAssistant:
         '''
         return Template
 
-    async def generate_message(self,  messages,task_id):
+    async def generate_message(self,  messages,task_id='ok'):
         user_message=messages[-1]
-        latest_message = messages[-1]['content']['parts'][0]['text']
+        latest_message = messages[-1]['parts'][0]['text']
         latest_message={"content":self.generate_template(latest_message,task_id)}
-        user_message['content']['parts'][0]['text']=latest_message
+        user_message['parts'][0]['text']=latest_message
         messages[-1]=user_message
         url = f'https://generativelanguage.googleapis.com/v1beta/models/{self.model}:generateContent?key={self.api_key}'
 
@@ -104,7 +104,8 @@ class GenerativeAIAssistant:
                     temp=  await response.json()
                     return temp["candidates"][0]
                 except Exception as e:
-                    return f"Error ⚠️ {e}"
+                    return f"Error ⚠️ {e} {temp}"
                 
-    
+
+