Spaces:
Sleeping
Sleeping
Commit
·
d68a7c6
1
Parent(s):
cbcabed
fix: improve mongo db role
Browse files- core/chat/bot_service.py +2 -2
core/chat/bot_service.py
CHANGED
@@ -153,13 +153,13 @@ class ChatCompletionService:
|
|
153 |
def _save_chat_history_to_db(self, response, metadata_collection):
|
154 |
chat_history_db = [
|
155 |
ChatMessage(
|
156 |
-
role=MessageRole.
|
157 |
content=self.user_request,
|
158 |
timestamp=datetime.now(),
|
159 |
payment="free" if self.type_bot == "general" else None,
|
160 |
),
|
161 |
ChatMessage(
|
162 |
-
role=MessageRole.
|
163 |
content=response,
|
164 |
metadata=metadata_collection,
|
165 |
timestamp=datetime.now(),
|
|
|
153 |
def _save_chat_history_to_db(self, response, metadata_collection):
|
154 |
chat_history_db = [
|
155 |
ChatMessage(
|
156 |
+
role=MessageRole.USER,
|
157 |
content=self.user_request,
|
158 |
timestamp=datetime.now(),
|
159 |
payment="free" if self.type_bot == "general" else None,
|
160 |
),
|
161 |
ChatMessage(
|
162 |
+
role=MessageRole.ASSISTANT,
|
163 |
content=response,
|
164 |
metadata=metadata_collection,
|
165 |
timestamp=datetime.now(),
|