Akjava commited on
Commit
0c69327
·
verified ·
1 Parent(s): e4ae0b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -183,7 +183,7 @@ model = LiteLLMModel(
183
  "deepseek-r1-distill-llama-70b",
184
  api_base="https://api.groq.com/openai/v1",
185
  custom_role_conversions=custom_role_conversions,
186
- max_tokens=int(6000*0.65),
187
  #num_ctx=819,
188
 
189
  api_key=os.getenv("OPENAI_API_KEY")
@@ -213,7 +213,7 @@ def create_agent():
213
  return CodeAgent(
214
  model=model,
215
  tools=[visualizer] + WEB_TOOLS,
216
- max_steps=20,
217
  verbosity_level=1,
218
  additional_authorized_imports=AUTHORIZED_IMPORTS,
219
  planning_interval=60*1000,
 
183
  "deepseek-r1-distill-llama-70b",
184
  api_base="https://api.groq.com/openai/v1",
185
  custom_role_conversions=custom_role_conversions,
186
+ max_tokens=int(6000*0.6),
187
  #num_ctx=819,
188
 
189
  api_key=os.getenv("OPENAI_API_KEY")
 
213
  return CodeAgent(
214
  model=model,
215
  tools=[visualizer] + WEB_TOOLS,
216
+ max_steps=10,
217
  verbosity_level=1,
218
  additional_authorized_imports=AUTHORIZED_IMPORTS,
219
  planning_interval=60*1000,