Update agent.py

agent.py

@@ -6,7 +6,7 @@ from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage, AIM
 from langgraph.graph import StateGraph, START, END, MessagesState
 from langgraph.prebuilt import ToolNode, tools_condition
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
-
+from langchain_google_genai import ChatGoogleGenerativeAI
 
 HUGGINGFACEHUB_API_TOKEN = getenv("HUGGINGFACEHUB_API_TOKEN")
 
@@ -16,12 +16,7 @@ HUGGINGFACEHUB_API_TOKEN = getenv("HUGGINGFACEHUB_API_TOKEN")
 #    huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
 #)
 
-
-
-llm = HuggingFaceEndpoint(
-    url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
-)
-
+llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
 
 chat = ChatHuggingFace(llm=llm, verbose=True)
 tools = [add,