File tree Expand file tree Collapse file tree
surfsense_backend/app/routes Expand file tree Collapse file tree Original file line number Diff line number Diff 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 ,
You can’t perform that action at this time.
0 commit comments