GilbertoEwaldFilho commited on
Commit
8a6e4ed
·
verified ·
1 Parent(s): 5f44925

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -72,15 +72,16 @@ class BasicAgent:
72
  def __init__(self):
73
  print("Initializing smolagents BasicAgent...")
74
 
75
- # Modelo remoto via Inference API (utiliza HF_TOKEN configurado no Space)
76
- self.model = InferenceClientModel()
 
 
77
 
78
  # CodeAgent sem ferramentas (agente simples)
79
  self.agent = CodeAgent(
80
  model=self.model,
81
- tools=[], # agente simples: sem tools
82
- max_steps=1, # sem tools, 1 passo é suficiente
83
- system_prompt=SYSTEM_PROMPT,
84
  )
85
 
86
  def __call__(self, question: str) -> str:
 
72
  def __init__(self):
73
  print("Initializing smolagents BasicAgent...")
74
 
75
+ # Passamos o system_prompt para o modelo, não para o CodeAgent
76
+ self.model = InferenceClientModel(
77
+ system_prompt=SYSTEM_PROMPT
78
+ )
79
 
80
  # CodeAgent sem ferramentas (agente simples)
81
  self.agent = CodeAgent(
82
  model=self.model,
83
+ tools=[], # agente simples: sem tools
84
+ max_steps=1 # sem tools, 1 passo é suficiente
 
85
  )
86
 
87
  def __call__(self, question: str) -> str: