Skip to content

Commit 2e486cb

Browse files
committed
better format for chat history
1 parent f2fbf99 commit 2e486cb

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

backend/graph.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -219,8 +219,10 @@ def route_to_retriever(
219219
def get_chat_history(messages: Sequence[BaseMessage]) -> Sequence[BaseMessage]:
220220
chat_history = []
221221
for message in messages:
222-
if isinstance(message, AIMessage) or isinstance(message, HumanMessage):
223-
chat_history.append(message)
222+
if (isinstance(message, AIMessage) and not message.tool_calls) or isinstance(
223+
message, HumanMessage
224+
):
225+
chat_history.append({"content": message.content, "role": message.type})
224226
return chat_history
225227

226228

0 commit comments

Comments
 (0)