bla commited on
Commit
db84a0f
·
verified ·
1 Parent(s): 41d120a

Update App/Chat/utils/RAG.py

Browse files
Files changed (1) hide show
  1. App/Chat/utils/RAG.py +1 -2
App/Chat/utils/RAG.py CHANGED
@@ -46,8 +46,7 @@ class GenerativeAIAssistant:
46
  async def generate_message(self, messages,task_id='ok'):
47
  user_message=messages[-1]
48
  latest_message = messages[-1]['parts'][0]['text']
49
- latest_message={"content":self.generate_template(latest_message,task_id)}
50
- user_message['parts'][0]['text']=latest_message
51
  messages[-1]=user_message
52
  url = f'https://generativelanguage.googleapis.com/v1beta/models/{self.model}:generateContent?key={self.api_key}'
53
 
 
46
  async def generate_message(self, messages,task_id='ok'):
47
  user_message=messages[-1]
48
  latest_message = messages[-1]['parts'][0]['text']
49
+ user_message['parts'][0]['text']=self.generate_template(latest_message,task_id)
 
50
  messages[-1]=user_message
51
  url = f'https://generativelanguage.googleapis.com/v1beta/models/{self.model}:generateContent?key={self.api_key}'
52