model change

app.py

@@ -47,14 +47,18 @@ final_answer = FinalAnswerTool()
 #model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' ,# it is possible that this model may be overloaded
 #custom_role_conversions=None,
 #)
+
 class DummyModel:
     def __init__(self):
         self.last_input_token_count = 0
         self.last_output_token_count = 0
 
-    def __call__(self, prompt: str, **kwargs):
-        self.last_input_token_count = len(prompt.split())
-        self.last_output_token_count = 10  # juste une valeur arbitraire
+    def __call__(self, prompt, **kwargs):
+        # prompt est une liste de messages [{"role": ..., "content": ...}, ...]
+        text_prompt = "\n".join([msg["content"] for msg in prompt if "content" in msg])
+
+        self.last_input_token_count = len(text_prompt.split())
+        self.last_output_token_count = 10  # valeur arbitraire
 
         return {
             "text": """Thought: I now know the final answer.