benismaw commited on
Commit
72ee8fe
·
verified ·
1 Parent(s): 3ccdf55

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -23,7 +23,9 @@ class BasicAgent:
23
  print("BasicAgent initialized.")
24
  self.model = LiteLLMModel(
25
  model_id="groq/llama-3.1-8b-instant", # TPM plus généreux que le 70b
26
- api_key=os.getenv("GROQ_API_KEY")
 
 
27
  )
28
  self.agent = CodeAgent(
29
  tools=[DuckDuckGoSearchTool(max_results=2)], # très limité
 
23
  print("BasicAgent initialized.")
24
  self.model = LiteLLMModel(
25
  model_id="groq/llama-3.1-8b-instant", # TPM plus généreux que le 70b
26
+ api_key=os.getenv("GROQ_API_KEY"),
27
+ num_retries=1,
28
+ timeout=30
29
  )
30
  self.agent = CodeAgent(
31
  tools=[DuckDuckGoSearchTool(max_results=2)], # très limité