Update agent.py

agent.py

@@ -463,20 +463,16 @@ from langchain_community.llms import HuggingFaceHub
 hf_token = os.getenv("HF_TOKEN")
 
 # Load the Qwen2.5-Coder-32B-Instruct model using HuggingFaceHub
-qwen_model = HuggingFaceHub(
-    repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",  # Specify the model from Hugging Face
-    api_key=hf_token,  # Pass the API token for access
-    model_kwargs={"temperature": 0.7}  # Adjust temperature as needed
+from langchain_huggingface import HuggingFaceEndpoint
+
+# Initialize the HuggingFaceEndpoint with the desired model and parameters
+llm = HuggingFaceEndpoint(
+    repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
+    task="text-generation",
+    huggingfacehub_api_token=hf_token,
+    model_kwargs={"temperature": 0.7, "max_length": 1024}
 )
 
-# Example usage with LangChain
-tools = [
-    Tool(
-        name="code_tool",
-        func=qwen_model.run,  # Use the run method for inference
-        description="Use this tool for code generation or similar tasks"
-    )
-]
 
 # Initialize the LangChain agent with the tool(s) and the model
 agent = initialize_agent(