retry

app.py

@@ -7,7 +7,7 @@ from typing_extensions import TypedDict, Annotated
 import gradio as gr
 from langchain.tools import tool
 from langchain_huggingface import HuggingFacePipeline
-from langchain.messages import HumanMessage, AIMessage, ToolMessage, AnyMessage
+from langchain_core.messages import HumanMessage, AIMessage, ToolMessage, BaseMessage
 from langgraph.graph import StateGraph, START, END
 
 # 1) Model
@@ -40,7 +40,7 @@ tools_by_name = {t.name: t for t in tools}
 
 # 3) State
 class MessagesState(TypedDict):
-    messages: Annotated[list[AnyMessage], operator.add]
+    messages: Annotated[list[BaseMessage], operator.add]
     llm_calls: int
 
 SYSTEM = """You are an arithmetic tool user.
@@ -56,7 +56,7 @@ If there IS a tool result already, output the final answer:
 No extra text. Use double quotes if possible.
 """.strip()
 
-def _format_for_phi(messages: list[AnyMessage]) -> str:
+def _format_for_phi(messages: list[BaseMessage]) -> str:
     parts = [SYSTEM, ""]
     for m in messages:
         if isinstance(m, HumanMessage):

requirements.txt

@@ -6,3 +6,6 @@ langchain==0.2.14
 langgraph==0.2.33
 langchain-huggingface==0.0.3
 huggingface_hub==0.24.6
+langchain-core==0.2.35
+
+