jupyter-agent 2025.6.102__py3-none-any.whl → 2025.6.103__py3-none-any.whl

jupyter_agent/bot_magics.py

@@ -61,10 +61,33 @@ class BotMagics(Magics, Configurable):
 
         return options
 
+    def ensure_notebook_path(self):
+        if self.notebook_path:
+            return self.notebook_path
+        result = self.shell and self.shell.run_cell("globals().get('__vsc_ipynb_file__')")
+        if result and result.success and result.result:
+            self.notebook_path = result.result
+            return self.notebook_path
+        try:
+            self.notebook_path = str(ipynbname.path())
+            return self.notebook_path
+        except Exception as e:
+            _F(f"Failed to get notebook path: {e}")
+            return None
+
     @cell_magic
     def bot(self, line, cell):
         """Jupyter cell magic: %%bot"""
         try:
+            if not self.ensure_notebook_path():
+                _O(
+                    Markdown(
+                        "The notebook path is **empty**, we can't do anything.\n\n"
+                        "Please set the notebook path in the configuration, and **RERUN** the cell again.\n\n"
+                        'For example: `%config BotMagics.notebook_path = globals()["__vsc_ipynb_file__"]`'
+                    )
+                )
+                return
             AgentCellContext.SUPPORT_SAVE_META = self.support_save_meta
             reset_output(stage="Logging", logging_level=self.logging_level)
             _I("Cell magic %%bot executing ...")
@@ -85,8 +108,6 @@ class BotMagics(Magics, Configurable):
             options = self.parse_args(line)
             _D(f"Cell magic called with options: {options}")
             set_logging_level(options.logging_level)
-            self.notebook_path = self.notebook_path or ipynbname.path()
-            _D(f"Cell magic called with notebook path: {self.notebook_path}")
             nb_context = NotebookContext(line, cell, notebook_path=self.notebook_path)
             agent_factory = AgentFactory(
                 nb_context,

{jupyter_agent-2025.6.102.dist-info → jupyter_agent-2025.6.103.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jupyter-agent
-Version: 2025.6.102
+Version: 2025.6.103
 Summary: 调用LLM实现Jupyter代码的自动生成、执行、调试等功能
 Author: viewstar000
 License: MIT
@@ -75,6 +75,8 @@ pip install /path/to/jupyter-agent/dist/jupyter_agent-xxxx-py3-none-any.whl
 
 ### 全局配置
 
+基础配置
+
 ```python
 # 加载扩展的Magic命令
 %load_ext jupyter_agent.bot_magics
@@ -84,12 +86,26 @@ pip install /path/to/jupyter-agent/dist/jupyter_agent-xxxx-py3-none-any.whl
 %config BotMagics.default_api_key = 'API_KEY'
 %config BotMagics.default_model_name = 'qwen3-30b-a3b' 
 %config BotMagics.coding_model_name = 'devstral-small-2505-mlx'
+```
+
+扩展配置
 
-# 设置当前Notebook的路径，由于vscode中运行里无法自动获取到该路径，需要手工指定
+```python
+# 设置当前Notebook的路径，当无法自动获取时需要手工指定，以Vscode中的Notebook为例
 %config BotMagics.notebook_path = globals()["__vsc_ipynb_file__"]
 
 # 设置是否保存任务数据到Metadata，只有Vscode中安装了jupyter-agent-extension后才支持
 %config BotMagics.support_save_meta = True
+
+# 设置日志级别，可选值为DEBUG、INFO、WARN、ERROR、FATAL，默认为INFO
+%config BotMagics.logging_level = 'DEBUG'
+
+# 设置是否显示思考过程，默认为True
+%config BotMagics.display_think = True
+
+# 设置是否显示发送给出LLM的消息和LLM的回答，默认为False
+%config BotMagics.display_message = True
+%config BotMagics.display_response = True
 ```
 
 ### 全局任务规划
@@ -118,7 +134,7 @@ pip install /path/to/jupyter-agent/dist/jupyter_agent-xxxx-py3-none-any.whl
 
 ![docs/image-task-empty.png](https://raw.githubusercontent.com/viewstar000/jupyter-agent/refs/heads/main/docs/image-task-empty.png)
 
-> 注：由于cell magic命令无法直接定位当前cell，需要通过cell的内容进行匹配，因此首次执行%%bot命令时，需要在cell中额外添加一些随机字符
+> **注：**由于cell magic命令无法直接定位当前cell，需要通过cell的内容进行匹配，因此首次执行%%bot命令时，需要在cell中额外添加一些随机字符
 
 接下来工具会调用相应的agent自动生成并执行相应步骤的代码，如下图：
 
@@ -130,6 +146,8 @@ pip install /path/to/jupyter-agent/dist/jupyter_agent-xxxx-py3-none-any.whl
 
 ![docs/image-task-confirm.png](https://raw.githubusercontent.com/viewstar000/jupyter-agent/refs/heads/main/docs/image-task-confirm.png)
 
+> **注：**在执行`%%bot`命令前，必须确保当前Notebook已保存，否则Agent无法读取到完整的Notebook上下文。建议开启Notebook编辑器自动保存功能。
+
 更详细用法可参考[示例Notebook](https://github.com/viewstar000/jupyter-agent/blob/main/examples/data_loader.ipynb)
 
 ## 贡献
@@ -195,6 +213,10 @@ pip install /path/to/jupyter-agent/dist/jupyter_agent-xxxx-py3-none-any.whl
 
 After installing `jupyter-agent` and `jupyter-agent-extension`, you can use `%%bot` magic command to work on task planning, code generation and execution.
 
+### Configuration
+
+Basic Configuration:
+
 First create or open a notebook in Vscode, create a new cell, enter and execute the following commands:
 
 ```python
@@ -206,12 +228,26 @@ First create or open a notebook in Vscode, create a new cell, enter and execute
 %config BotMagics.default_api_key = 'API_KEY'
 %config BotMagics.default_model_name = 'qwen3-30b-a3b' 
 %config BotMagics.coding_model_name = 'devstral-small-2505-mlx'
+```
 
-# Set the path of the current Notebook, which cannot be automatically obtained in vscode when running, you need to manually specify
+Advanced Configuration:
+
+```python
+# Set the current notebook path, when it is not automatically obtained, it needs to be manually specified, for example, in Vscode Notebook
 %config BotMagics.notebook_path = globals()["__vsc_ipynb_file__"]
 
 # Set whether to save task data to Metadata, only Vscode installed with jupyter-agent-extension supports
 %config BotMagics.support_save_meta = True
+
+# Set the log level, available values are DEBUG、INFO、WARN、ERROR、FATAL, default is INFO
+%config BotMagics.logging_level = 'DEBUG'
+
+# Set whether to display thinking process, default is True
+%config BotMagics.display_think = True
+
+# Set whether to display messages sent to LLM and LLM responses, default is False
+%config BotMagics.display_message = True
+%config BotMagics.display_response = True
 ```
 
 Now, you can use the `%%bot` command to work on task rules and code generation.
@@ -249,6 +285,8 @@ After generating code for a subtask, the tool will call the corresponding agent
 
 ![docs/image-task-confirm.png](https://raw.githubusercontent.com/viewstar000/jupyter-agent/refs/heads/main/docs/image-task-confirm.png)
 
+> **Note:** Before using the `%%bot` command, you must ensure that the current notebook has been saved, otherwise the agent will not be able to read the full context of the notebook. Suggested to enable the notebook editor's automatic save function.
+
 For more details, please refer to [example notebook](https://github.com/viewstar000/jupyter-agent/blob/main/examples/data_loader.ipynb)
 
 ## Contributing

{jupyter_agent-2025.6.102.dist-info → jupyter_agent-2025.6.103.dist-info}/RECORD

@@ -1,7 +1,7 @@
 jupyter_agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 jupyter_agent/bot_chat.py,sha256=RkaT-Cpkp3G24F1AtbcgqjKo-LBxwMNAYHVitaPZNOg,8834
 jupyter_agent/bot_contexts.py,sha256=eFezr7ImokMz33jJJMlLLQspQBYnuE6Bc2bE1wLbXWU,19123
-jupyter_agent/bot_magics.py,sha256=PQib9GoefCAkh2Hy1InfdoiFo-KE-9wJ4qKEj_k9zO4,6570
+jupyter_agent/bot_magics.py,sha256=whyGu6Jru7ZAIWXjwoq_CfNxqvlDawGTREUHX9XjHFY,7434
 jupyter_agent/bot_outputs.py,sha256=7_cbOBaGdoYMR6qu43nZEsB9KY6wp4zA0Dh7Jq7LMTg,14065
 jupyter_agent/utils.py,sha256=jbvDtVK6MfGaf5ZLoam_sq9R5TMriEG4HbMF0bHHDes,4387
 jupyter_agent/bot_agents/__init__.py,sha256=IdlBlvfaDUfp3qhkNuwUVV_CdplafZsgLezLVkZCREw,1323
@@ -26,8 +26,8 @@ jupyter_agent/bot_flows/master_planner.py,sha256=2MvAMikMsBIarTL4QNYTjAs70bxo7VA
 jupyter_agent/bot_flows/task_executor_v1.py,sha256=WzTfoOTjBpk1emvpiL3yeiudKdDf6EpVdJIugxxbqM4,2975
 jupyter_agent/bot_flows/task_executor_v2.py,sha256=IyfxhzA4TljNd6iqqUYxxzB63r9lxfe1Zbu177hytRs,2949
 jupyter_agent/bot_flows/task_executor_v3.py,sha256=NwJy7iAo2U3rl5iQYpACWVAkIA5DRMDdx0JadheRMMQ,3198
-jupyter_agent-2025.6.102.dist-info/licenses/LICENSE,sha256=nWMmSIg7OepTIDX_OPP0-T9ImeCBBoog7eJxm5awtcM,1068
-jupyter_agent-2025.6.102.dist-info/METADATA,sha256=ghJ-4ompPy8WtYEam-O_jHLQZQF55wEgeKXrwBwXoIQ,8647
-jupyter_agent-2025.6.102.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-jupyter_agent-2025.6.102.dist-info/top_level.txt,sha256=c3USTBZ7DZGuvLKlEW-QfGIx0tzn98iCEn3bpdYnDtE,14
-jupyter_agent-2025.6.102.dist-info/RECORD,,
+jupyter_agent-2025.6.103.dist-info/licenses/LICENSE,sha256=nWMmSIg7OepTIDX_OPP0-T9ImeCBBoog7eJxm5awtcM,1068
+jupyter_agent-2025.6.103.dist-info/METADATA,sha256=ZtUJtanX1-RXpH1mMA6JH9fWCWK4YMaFU1h9e2nP_tE,9998
+jupyter_agent-2025.6.103.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+jupyter_agent-2025.6.103.dist-info/top_level.txt,sha256=c3USTBZ7DZGuvLKlEW-QfGIx0tzn98iCEn3bpdYnDtE,14
+jupyter_agent-2025.6.103.dist-info/RECORD,,