Update app.py

app.py

@@ -7,7 +7,8 @@ from langchain_core.documents import Document
 import base64
 from langchain_core.messages import HumanMessage
 from langchain_openai import ChatOpenAI
-from langchain_community.tools.ddg_search.tool import DuckDuckGoSearchTool
+from langchain_community.tools import DuckDuckGoSearchResults
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -41,7 +42,7 @@ tools = [
     add,
     subtract,
     search,
-    DuckDuckGoSearchTool()
+    DuckDuckGoSearchResults()
 ]
 ###### state and behavior
 from typing import TypedDict, Annotated, Optional
@@ -52,7 +53,51 @@ from langgraph.graph.message import add_messages
 class AgentState(TypedDict):
     # The input document
     messages: Annotated[list[AnyMessage], add_messages]
-    
+
+from langchain_core.messages import HumanMessage, SystemMessage
+
+# from langchain_core.tools import tool # non necessary with toolNode
+
+# @tool
+# def multiply(a: int, b: int) -> int:
+#     """Multiply a and b."""
+#     return a * b
+
+def assistant(state: AgentState):
+    # System message
+    textual_description_of_tool = """
+DuckDuckGoSearchResults():
+    Search for information in the internet through DuckDuckGo engine
+
+    Returns:
+        Snipnet of information from search engine.
+        
+divide(a: int|float, b: int|float) -> float:
+    Divide a and b.
+    Returns:
+        The result of A divided by B if B!=0
+        
+multiply(a: int|float, b: int|float) -> float:
+    Multiply a with b.
+    Returns:
+        The result of A multiplied by B
+        
+add(a: int|float, b: int|float) -> float:
+    Add a and b.
+    Returns:
+        The result of A adding B.
+        
+Substract(a: int|float, b: int|float) -> float:
+    Substract a with b.
+    Returns:
+        The result of a minus b.
+        
+"""
+    previous_message = state["messages"]
+    sys_msg = SystemMessage(content=f"You are an helpful agent that can analyse some images and run some computatio without provided tools :\n{textual_description_of_tool} \n You have access to some otpional images. Currently the loaded images is : {image}")
+
+    return {"messages": [llm_with_tools.invoke([sys_msg] + state["messages"])], "input_file": state["input_file"]}
+
 ################ state
 class BasicAgent:
     def __init__(self):