Spaces:
Sleeping
Sleeping
dd
Browse files
app.py
CHANGED
|
@@ -54,15 +54,20 @@ class DummyModel:
|
|
| 54 |
self.last_output_token_count = 0
|
| 55 |
|
| 56 |
def __call__(self, prompt, **kwargs):
|
| 57 |
-
|
| 58 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 59 |
|
| 60 |
self.last_input_token_count = len(text_prompt.split())
|
| 61 |
-
self.last_output_token_count = 10
|
| 62 |
|
| 63 |
return {
|
| 64 |
"text": """Thought: I now know the final answer.
|
| 65 |
-
Final Answer: 6.
|
| 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 |
|