Update agent.py

agent.py

@@ -18,6 +18,7 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 
 #load_dotenv()
 google_api_key = os.environ["GOOGLE_API_KEY"]
+hf_api_key = os.environ["HF_API_KEY"]
 
 @tool
 def add(a: int, b: int) -> int:
@@ -111,6 +112,7 @@ def build_graph(provider: str = "huggingface"):
             llm=HuggingFaceEndpoint(
                 url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",
                 temperature=0,
+                huggingfacehub_api_token=hf_api_key
             ),
         )
     def assistant(state: MessagesState):