vrr/SurfSense
2
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2025-09-02 02:29:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: bad chat history logic

Browse source
This commit is contained in:
DESKTOP-RTLN3BA\$punk 2025-07-10 15:06:25 -07:00
parent 24587d796d
commit a19a8af4ff
1 changed files with 8 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
surfsense_backend/app/routes/chats_routes.py
View file

@ -54,10 +54,14 @@ async def handle_chat_data(
if message['role'] == "user": if message['role'] == "user":
langchain_chat_history.append(HumanMessage(content=message['content'])) langchain_chat_history.append(HumanMessage(content=message['content']))
elif message['role'] == "assistant": elif message['role'] == "assistant":
# Last annotation type will always be "ANSWER" here # Find the last "ANSWER" annotation specifically
answer_annotation = message['annotations'][-1] answer_annotation = None
answer_text = "" for annotation in reversed(message['annotations']):
if answer_annotation['type'] == "ANSWER": if annotation['type'] == "ANSWER":
answer_annotation = annotation
break
if answer_annotation:
answer_text = answer_annotation['content'] answer_text = answer_annotation['content']
# If content is a list, join it into a single string # If content is a list, join it into a single string
if isinstance(answer_text, list): if isinstance(answer_text, list):