Spaces:
Sleeping
Sleeping
Update Gradio_UI.py
Browse files- Gradio_UI.py +1 -1
Gradio_UI.py
CHANGED
|
@@ -144,7 +144,7 @@ def stream_to_gradio(
|
|
| 144 |
if hasattr(agent.model, "last_input_token_count"):
|
| 145 |
if agent.model.last_input_token_count is not None:
|
| 146 |
total_output_tokens += agent.model.last_output_token_count
|
| 147 |
-
|
| 148 |
step_log.input_token_count = agent.model.last_input_token_count
|
| 149 |
step_log.output_token_count = agent.model.last_output_token_count
|
| 150 |
|
|
|
|
| 144 |
if hasattr(agent.model, "last_input_token_count"):
|
| 145 |
if agent.model.last_input_token_count is not None:
|
| 146 |
total_output_tokens += agent.model.last_output_token_count
|
| 147 |
+
if isinstance(step_log, ActionStep):
|
| 148 |
step_log.input_token_count = agent.model.last_input_token_count
|
| 149 |
step_log.output_token_count = agent.model.last_output_token_count
|
| 150 |
|