mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-01-31 23:55:06 +08:00
Doc: memory http api (#12499)
### What problem does this PR solve? Use task save function for add_message api, and added http API document. ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) - [x] Documentation Update
This commit is contained in:
@ -31,34 +31,21 @@ async def add_message():
|
|||||||
|
|
||||||
req = await get_request_json()
|
req = await get_request_json()
|
||||||
memory_ids = req["memory_id"]
|
memory_ids = req["memory_id"]
|
||||||
agent_id = req["agent_id"]
|
|
||||||
session_id = req["session_id"]
|
|
||||||
user_id = req["user_id"] if req.get("user_id") else ""
|
|
||||||
user_input = req["user_input"]
|
|
||||||
agent_response = req["agent_response"]
|
|
||||||
|
|
||||||
res = []
|
message_dict = {
|
||||||
for memory_id in memory_ids:
|
"user_id": req.get("user_id"),
|
||||||
success, msg = await memory_message_service.save_to_memory(
|
"agent_id": req["agent_id"],
|
||||||
memory_id,
|
"session_id": req["session_id"],
|
||||||
{
|
"user_input": req["user_input"],
|
||||||
"user_id": user_id,
|
"agent_response": req["agent_response"],
|
||||||
"agent_id": agent_id,
|
}
|
||||||
"session_id": session_id,
|
|
||||||
"user_input": user_input,
|
|
||||||
"agent_response": agent_response
|
|
||||||
}
|
|
||||||
)
|
|
||||||
res.append({
|
|
||||||
"memory_id": memory_id,
|
|
||||||
"success": success,
|
|
||||||
"message": msg
|
|
||||||
})
|
|
||||||
|
|
||||||
if all([r["success"] for r in res]):
|
res, msg = await memory_message_service.queue_save_to_memory_task(memory_ids, message_dict)
|
||||||
return get_json_result(message="Successfully added to memories.")
|
|
||||||
|
|
||||||
return get_json_result(code=RetCode.SERVER_ERROR, message="Some messages failed to add.", data=res)
|
if res:
|
||||||
|
return get_json_result(message=msg)
|
||||||
|
|
||||||
|
return get_json_result(code=RetCode.SERVER_ERROR, message="Some messages failed to add. Detail:" + msg)
|
||||||
|
|
||||||
|
|
||||||
@manager.route("/<memory_id>:<message_id>", methods=["DELETE"]) # noqa: F821
|
@manager.route("/<memory_id>:<message_id>", methods=["DELETE"]) # noqa: F821
|
||||||
|
|||||||
File diff suppressed because it is too large
Load Diff
Reference in New Issue
Block a user