Update agent.py

agent.py

@@ -170,7 +170,7 @@ def build_graph(provider: str = "huggingface", huggingface_model: str = "mistral
         else:
             raise ValueError("Unsupported Hugging Face model")
 
-        hf_token = os.getenv("HUGGINGFACE_API_TOKEN")
+        hf_token = os.getenv("HF_TOKEN")
         headers = {"Authorization": f"Bearer {hf_token}"} if hf_token else {}
 
         llm = ChatHuggingFace(
@@ -216,13 +216,3 @@ def build_graph(provider: str = "huggingface", huggingface_model: str = "mistral
     # Compile graph
     return builder.compile()
 
-# test
-if __name__ == "__main__":
-    question = "When was a picture of St. Thomas Aquinas first added to the Wikipedia page on the Principle of double effect?"
-    # Build the graph
-    graph = build_graph(provider="groq")
-    # Run the graph
-    messages = [HumanMessage(content=question)]
-    messages = graph.invoke({"messages": messages})
-    for m in messages["messages"]:
-        m.pretty_print()