Update agent.py

agent.py

@@ -149,9 +149,10 @@ def build_graph(provider: str = "huggingface"):
     elif provider == "huggingface":
         # Add huggingface endpoint
         #repo_id = "mistralai/Mistral-7B-Instruct-v0.2"
-        repo_id = "Qwen/Qwen2.5-Coder-32B-Instruct" # -> it doesn't reply well
+        #repo_id = "Qwen/Qwen2.5-Coder-32B-Instruct" # -> it doesn't reply well
         #repo_id = "deepseek-ai/DeepSeek-Coder-V2-Instruct" -> it doesn't work (error on StopIteration)
         #repo_id = "meta-llama/CodeLlama-34b-Instruct-hf" -> it doesn't work (error on StopIteration)
+        repo_id = "WizardLMTeam/WizardCoder-15B-V1.0"
         chat = ChatHuggingFace(
             #llm=HuggingFaceEndpoint(
             #    endpoint_url="https://api-inference.huggingface.co/models/Meta-DeepLearning/llama-2-7b-chat-hf",