Update agent.py

agent.py

@@ -3,7 +3,7 @@ from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
 
 from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 from langchain_community.document_loaders import WikipediaLoader
-from llama_index.core.schema import Metadata
+from llama_index.core.tools.types import ToolMetadata
 from llama_index.core.schema import Document
 from llama_index.core.tools import FunctionTool
 from langchain_core.tools import tool
@@ -12,10 +12,6 @@ from llama_index.core.agent.workflow import AgentWorkflow
 
 llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-VL-32B-Instruct")
 
-class ToolMetadata(Metadata):
-    name: str
-    description: str
-    
 @tool
 def web_search(query: str) -> list:
     """Search Tavily for a query and return a maximum 3 results as LlamaIndex Documents.
@@ -62,8 +58,9 @@ def wiki_search(query: str) -> list:
 
     return llama_docs
 
+
 web_search = FunctionTool(
-    web_search, 
+    web_search,
     metadata=ToolMetadata(name="web_search", description="Tavily 3-hit search")
 )
 
@@ -71,6 +68,7 @@ wiki_search = FunctionTool(
     wiki_search,
     metadata=ToolMetadata(name="wiki_search", description="Wikipedia 2-hit search")
 )
+
 tools = [web_search, wiki_search]
 
 agent = AgentWorkflow.from_tools_or_functions(tools, llm=llm)