Update agent/agent.py

agent/agent.py

@@ -2,11 +2,11 @@ import os
 from typing import TypedDict, List, Dict, Any, Optional
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage, AIMessage
+from langgraph.graph.message import add_messages
 from langgraph.graph import START, StateGraph
 from langgraph.prebuilt import ToolNode, tools_condition
 
-from tools.search_tool import search_tool
-
+from agent.tools.search_tool import search_tool
 
 # --- zBottaAgent Definition ---
 
@@ -26,3 +26,4 @@ class zBottaAgent:
         response = self.chat_with_tools.invoke(messages)
         print(f"Agent returning response: {response}")
         return response
+