AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-12 22:48:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add timestamp to messages (#4624)

Browse Source 
### What problem does this PR solve?


### Type of change

- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
Kevin Hu 2025-01-24 11:07:55 +08:00 committed by GitHub
parent f1d9f4290e
commit e9ccba0395
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/db/services/canvas_service.py
View File

@ -14,6 +14,7 @@
# limitations under the License. # limitations under the License.
# #
import json import json
import time
import traceback import traceback
from uuid import uuid4 from uuid import uuid4
from agent.canvas import Canvas from agent.canvas import Canvas
@ -80,7 +81,7 @@ def completion(tenant_id, agent_id, question, session_id=None, stream=True, **kw
"id": session_id, "id": session_id,
"dialog_id": cvs.id, "dialog_id": cvs.id,
"user_id": kwargs.get("user_id", "") if isinstance(kwargs, dict) else "", "user_id": kwargs.get("user_id", "") if isinstance(kwargs, dict) else "",
"message": [{"role": "assistant", "content": canvas.get_prologue()}], "message": [{"role": "assistant", "content": canvas.get_prologue(), "created_at": time.time()}],
"source": "agent", "source": "agent",
"dsl": cvs.dsl "dsl": cvs.dsl
} }
@ -134,7 +135,7 @@ def completion(tenant_id, agent_id, question, session_id=None, stream=True, **kw
yield "data:" + json.dumps({"code": 0, "message": "", "data": ans}, yield "data:" + json.dumps({"code": 0, "message": "", "data": ans},
ensure_ascii=False) + "\n\n" ensure_ascii=False) + "\n\n"
canvas.messages.append({"role": "assistant", "content": final_ans["content"], "id": message_id}) canvas.messages.append({"role": "assistant", "content": final_ans["content"], "created_at": time.time(), "id": message_id})
canvas.history.append(("assistant", final_ans["content"])) canvas.history.append(("assistant", final_ans["content"]))
if final_ans.get("reference"): if final_ans.get("reference"):
canvas.reference.append(final_ans["reference"]) canvas.reference.append(final_ans["reference"])

9
api/db/services/conversation_service.py
View File

@ -13,6 +13,7 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
# #
import time
from uuid import uuid4 from uuid import uuid4
from api.db import StatusEnum from api.db import StatusEnum
from api.db.db_models import Conversation, DB from api.db.db_models import Conversation, DB
@ -75,9 +76,9 @@ def structure_answer(conv, ans, message_id, session_id):
if not conv.message: if not conv.message:
conv.message = [] conv.message = []
if not conv.message or conv.message[-1].get("role", "") != "assistant": if not conv.message or conv.message[-1].get("role", "") != "assistant":
conv.message.append({"role": "assistant", "content": ans["answer"], "id": message_id}) conv.message.append({"role": "assistant", "content": ans["answer"], "created_at": time.time(), "id": message_id})
else: else:
conv.message[-1] = {"role": "assistant", "content": ans["answer"], "id": message_id} conv.message[-1] = {"role": "assistant", "content": ans["answer"], "created_at": time.time(), "id": message_id}
if conv.reference: if conv.reference:
conv.reference[-1] = reference conv.reference[-1] = reference
return ans return ans
@ -94,7 +95,7 @@ def completion(tenant_id, chat_id, question, name="New session", session_id=None
"id": session_id, "id": session_id,
"dialog_id": chat_id, "dialog_id": chat_id,
"name": name, "name": name,
"message": [{"role": "assistant", "content": dia[0].prompt_config.get("prologue")}], "message": [{"role": "assistant", "content": dia[0].prompt_config.get("prologue"), "created_at": time.time()}],
"user_id": kwargs.get("user_id", "") "user_id": kwargs.get("user_id", "")
} }
ConversationService.save(**conv) ConversationService.save(**conv)
@ -166,7 +167,7 @@ def iframe_completion(dialog_id, question, session_id=None, stream=True, **kwarg
"id": session_id, "id": session_id,
"dialog_id": dialog_id, "dialog_id": dialog_id,
"user_id": kwargs.get("user_id", ""), "user_id": kwargs.get("user_id", ""),
"message": [{"role": "assistant", "content": dia.prompt_config["prologue"]}] "message": [{"role": "assistant", "content": dia.prompt_config["prologue"], "created_at": time.time()}]
} }
API4ConversationService.save(**conv) API4ConversationService.save(**conv)
yield "data:" + json.dumps({"code": 0, "message": "", yield "data:" + json.dumps({"code": 0, "message": "",