mchabrol commited on
Commit
d6f7c78
·
verified ·
1 Parent(s): 3126e6d

change model

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -52,10 +52,9 @@ class DummyModel:
52
  self.last_input_token_count = 0
53
  self.last_output_token_count = 0
54
 
55
- def __call__(self, prompt: str):
56
- # Simule une sortie d'agent bien structurée
57
- self.last_input_token_count = len(prompt.split()) # juste pour simuler
58
- self.last_output_token_count = 12 # nombre fictif
59
 
60
  return {
61
  "text": """Thought: I now know the final answer.
 
52
  self.last_input_token_count = 0
53
  self.last_output_token_count = 0
54
 
55
+ def __call__(self, prompt: str, **kwargs):
56
+ self.last_input_token_count = len(prompt.split())
57
+ self.last_output_token_count = 10 # juste une valeur arbitraire
 
58
 
59
  return {
60
  "text": """Thought: I now know the final answer.