Skip to content

Commit aacf38f

Browse files
committed
feat: fix anonymous chat message history
1 parent c3b9d41 commit aacf38f

1 file changed

Lines changed: 10 additions & 6 deletions

File tree

surfsense_backend/app/routes/anonymous_chat_routes.py

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -349,7 +349,7 @@ async def stream_anonymous_chat(
349349
raise
350350

351351
async def _generate():
352-
from langchain_core.messages import HumanMessage
352+
from langchain_core.messages import AIMessage, HumanMessage
353353

354354
from app.agents.new_chat.chat_deepagent import create_surfsense_deep_agent
355355
from app.agents.new_chat.checkpointer import get_checkpointer
@@ -383,13 +383,17 @@ async def _generate():
383383
anon_session_id=session_id,
384384
)
385385

386+
langchain_messages = []
386387
user_query = ""
387-
for msg in reversed(body.messages):
388-
if msg.get("role") == "user":
389-
user_query = msg.get("content", "")
390-
break
388+
for msg in body.messages:
389+
role = msg.get("role", "")
390+
content = msg.get("content", "")
391+
if role == "user":
392+
langchain_messages.append(HumanMessage(content=content))
393+
user_query = content
394+
elif role == "assistant":
395+
langchain_messages.append(AIMessage(content=content))
391396

392-
langchain_messages = [HumanMessage(content=user_query)]
393397
input_state = {
394398
"messages": langchain_messages,
395399
"search_space_id": 0,

0 commit comments

Comments
 (0)