Fix AttributeError: 'HumanMessage' object has no attribute

vintasoftware · Nov 19, 2024 · 0d1ae05 · 0d1ae05
1 parent 999980c
commit 0d1ae05
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/django_ai_assistant/helpers/assistants.py b/django_ai_assistant/helpers/assistants.py
@@ -459,7 +459,9 @@ def retriever(state: AgentState):
             # This is necessary for compatibility with Anthropic
             messages_to_summarize = state["messages"][1:-1]
             input_message = state["messages"][-1]
-            docs = retriever.invoke({"input": input_message, "history": messages_to_summarize})
+            docs = retriever.invoke(
+                {"input": input_message.content, "history": messages_to_summarize}
+            )
 
             document_separator = self.get_document_separator()
             document_prompt = self.get_document_prompt()