lucaslingle commited on
Commit
93170ef
·
verified ·
1 Parent(s): 727c6f6

Update app.py

Browse files

lengthen the timeout

Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -87,11 +87,17 @@ class EnhancedAgent:
87
  def __init__(self):
88
  self.model_id = "Qwen/Qwen3-235B-A22B-Thinking-2507"
89
  self.provider = "auto"
 
90
  self.tools = [read_file_as_text, reverse_string, web_search, visit_webpage, final_answer]
91
  self.auth_imports = ["pandas", "numpy", "datetime", "json", "re", "math", "os", "requests", "csv", "urllib"]
92
  self.max_steps = 5
93
  self.agent = CodeAgent(
94
- model=InferenceClientModel(model_id=self.model_id, provider=self.provider, token=os.getenv("HF_TOKEN")),
 
 
 
 
 
95
  tools=self.tools,
96
  additional_authorized_imports=self.auth_imports,
97
  max_steps=self.max_steps,
 
87
  def __init__(self):
88
  self.model_id = "Qwen/Qwen3-235B-A22B-Thinking-2507"
89
  self.provider = "auto"
90
+ self.timeout = 360
91
  self.tools = [read_file_as_text, reverse_string, web_search, visit_webpage, final_answer]
92
  self.auth_imports = ["pandas", "numpy", "datetime", "json", "re", "math", "os", "requests", "csv", "urllib"]
93
  self.max_steps = 5
94
  self.agent = CodeAgent(
95
+ model=InferenceClientModel(
96
+ model_id=self.model_id,
97
+ provider=self.provider,
98
+ timeout=self.timeout,
99
+ token=os.getenv("HF_TOKEN"),
100
+ ),
101
  tools=self.tools,
102
  additional_authorized_imports=self.auth_imports,
103
  max_steps=self.max_steps,