mohammadreza pakzadian commited on
Commit
14d28bc
·
1 Parent(s): ca403c2

remove additional_authorized_imports

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -25,7 +25,8 @@ class BasicAgent:
25
  model = HfApiModel(
26
  max_tokens=2096,
27
  temperature=0.5,
28
- model_id='Qwen/Qwen2.5-Coder-7B-Instruct', custom_role_conversions=None,
 
29
  )
30
  self.agent = CodeAgent(
31
  model=model,
@@ -33,7 +34,7 @@ class BasicAgent:
33
  max_steps=5,
34
  verbosity_level=1,
35
  add_base_tools=True,
36
- additional_authorized_imports=['random', 'stat', 'time', 'collections', 'itertools', 'queue', 're', 'datetime', 'statistics', 'math', 'unicodedata', 'csv', 'pandas']
37
  )
38
  def __call__(self, question: str) -> str:
39
  print(f"Agent received question (first 50 chars): {question}")
 
25
  model = HfApiModel(
26
  max_tokens=2096,
27
  temperature=0.5,
28
+ model_id='Qwen/Qwen2.5-Coder-7B-Instruct',
29
+ custom_role_conversions=None,
30
  )
31
  self.agent = CodeAgent(
32
  model=model,
 
34
  max_steps=5,
35
  verbosity_level=1,
36
  add_base_tools=True,
37
+ # additional_authorized_imports=['random', 'stat', 'time', 'collections', 'itertools', 'queue', 're', 'datetime', 'statistics', 'math', 'unicodedata', 'csv', 'pandas']
38
  )
39
  def __call__(self, question: str) -> str:
40
  print(f"Agent received question (first 50 chars): {question}")