mchabrol commited on
Commit
be1c76f
·
verified ·
1 Parent(s): 785d9af
Files changed (1) hide show
  1. app.py +9 -4
app.py CHANGED
@@ -54,15 +54,20 @@ class DummyModel:
54
  self.last_output_token_count = 0
55
 
56
  def __call__(self, prompt, **kwargs):
57
- # prompt est une liste de messages [{"role": ..., "content": ...}, ...]
58
- text_prompt = "\n".join([msg["content"] for msg in prompt if "content" in msg])
 
 
 
 
 
59
 
60
  self.last_input_token_count = len(text_prompt.split())
61
- self.last_output_token_count = 10 # valeur arbitraire
62
 
63
  return {
64
  "text": """Thought: I now know the final answer.
65
- Final Answer: 6.4"""
66
  }
67
  model = DummyModel()
68
 
 
54
  self.last_output_token_count = 0
55
 
56
  def __call__(self, prompt, **kwargs):
57
+ def safe_str(msg):
58
+ if isinstance(msg["content"], list):
59
+ return "\n".join(map(str, msg["content"]))
60
+ return str(msg["content"])
61
+
62
+ # Recompose le texte à partir de la liste des messages
63
+ text_prompt = "\n".join([safe_str(msg) for msg in prompt if "content" in msg])
64
 
65
  self.last_input_token_count = len(text_prompt.split())
66
+ self.last_output_token_count = 10
67
 
68
  return {
69
  "text": """Thought: I now know the final answer.
70
+ Final Answer: 6.2"""
71
  }
72
  model = DummyModel()
73