Update App/Chat/utils/RAG.py
Browse files- 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 |
-
|
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 |
|