Spaces:

data-agents
/

jupyter-agent

Running

lvwerra HF staff commited on 21 days ago

Commit

fad3b25

verified ·

1 Parent(s): 185eab3

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -286,8 +286,8 @@ def run_interactive_notebook(client, model, tokenizer, messages, sbx, max_new_to
             else:
                 notebook_data["cells"][-1]["source"] = assistant_response
             if i%16 == 0:
-                yield update_notebook_display(notebook_data), messages
-        yield update_notebook_display(notebook_data), messages
         # Handle code execution

             else:
                 notebook_data["cells"][-1]["source"] = assistant_response
             if i%16 == 0:
+                yield update_notebook_display(notebook_data), notebook_data, messages
+        yield update_notebook_display(notebook_data), notebook_data, messages
         # Handle code execution