Update main.py

main.py

@@ -8,28 +8,27 @@ from langchain_core.output_parsers import PydanticOutputParser
 from langchain.agents import create_tool_calling_agent, AgentExecutor
 from tools import search_tool, wiki_tool, save_tool
 
-os.getenv("HUGGINGFACEHUB_API_TOKEN")
-
-# Define the response format
-class ResearchResponse(BaseModel):
-    topic: str
-    summary: str
-    sources: list[str]
-    tools_used: list[str]
-
-    
-
-# ✅ Use HuggingFace Mistral model (free)
+# ✅ Get the token from environment variables
+hf_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
 
+# ✅ Check and raise an error if it's missing
 if hf_token is None:
     raise ValueError("Missing Hugging Face API token. Please set the 'HUGGINGFACEHUB_API_TOKEN' secret in your HF space settings.")
 
+# ✅ Use HuggingFace Mistral model (free), Initialize your LLM using the token
 llm = HuggingFaceHub(
     repo_id="mistralai/Mistral-7B-Instruct-v0.3",
     huggingfacehub_api_token=hf_token,
     model_kwargs={"temperature": 0.7, "max_new_tokens": 500},
 )
 
+# Define the response format
+class ResearchResponse(BaseModel):
+    topic: str
+    summary: str
+    sources: list[str]
+    tools_used: list[str]
+
 # Create parser
 parser = PydanticOutputParser(pydantic_object=ResearchResponse)