Update Agent.py

Agent.py

@@ -5,9 +5,8 @@ from langchain.chat_models import ChatOpenAI
 from langchain.memory import ConversationBufferMemory
 from langchain_community.tools import DuckDuckGoSearchRun
 from langchain.schema import SystemMessage
-from langchain_community.chat_models import ChatHuggingFace
 from langchain.schema.messages import HumanMessage
-
+from langchain_huggingface import ChatHuggingFace
 GAIA_SYSTEM_PROMPT = """
 You are an advanced autonomous agent designed to pass the GAIA benchmark.
 You must:
@@ -30,11 +29,12 @@ llm = ChatHuggingFace(
     repo_id="HuggingFaceH4/zephyr-7b-beta",
     model_kwargs={"temperature": 0.0, "max_new_tokens": 1024}
 )
+
 messages = [
     SystemMessage(content=GAIA_SYSTEM_PROMPT)    
 ]
 
-response = llm(messages)
+response =  llm.invoke(messages)  
 
 def calculator_tool(input: str) -> str:
     try: