Humanlearning commited on
Commit
6387e77
·
1 Parent(s): ca915e7

increased timeout

Browse files
Files changed (1) hide show
  1. agents.py +1 -1
agents.py CHANGED
@@ -60,7 +60,7 @@ class LlamaIndexAgent:
60
  print("DEBUG: Types in all_tools:", [type(t) for t in all_tools])
61
  # LLM and agent workflow
62
  # self.llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-Coder-32B-Instruct")
63
- self.llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-Coder-32B-Instruct")
64
  self.alfred = AgentWorkflow.from_tools_or_functions(
65
  all_tools,
66
  llm=self.llm,
 
60
  print("DEBUG: Types in all_tools:", [type(t) for t in all_tools])
61
  # LLM and agent workflow
62
  # self.llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-Coder-32B-Instruct")
63
+ self.llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-Coder-32B-Instruct", streaming=False, client_kwargs={"timeout": 60})
64
  self.alfred = AgentWorkflow.from_tools_or_functions(
65
  all_tools,
66
  llm=self.llm,