autogen_ext.code_executors.jupyter

class JupyterCodeExecutor(kernel_name: str = 'python3', timeout: int = 60, output_dir: Path | str | None = None)

基类：CodeExecutor, Component[JupyterCodeExecutorConfig]

一个使用[nbclient](jupyter/nbclient)有状态执行代码的代码执行器类。

危险

这将在本地机器上执行代码。如果用于执行LLM生成的代码，应谨慎使用。

直接使用示例：

import asyncio
from autogen_core import CancellationToken
from autogen_core.code_executor import CodeBlock
from autogen_ext.code_executors.jupyter import JupyterCodeExecutor


async def main() -> None:
    async with JupyterCodeExecutor() as executor:
        cancel_token = CancellationToken()
        code_blocks = [CodeBlock(code="print('hello world!')", language="python")]
        code_result = await executor.execute_code_blocks(code_blocks, cancel_token)
        print(code_result)


asyncio.run(main())

与:class:`~autogen_ext.tools.code_execution.PythonCodeExecutionTool`一起使用的示例：

import asyncio
from autogen_agentchat.agents import AssistantAgent
from autogen_ext.code_executors.jupyter import JupyterCodeExecutor
from autogen_ext.models.openai import OpenAIChatCompletionClient
from autogen_ext.tools.code_execution import PythonCodeExecutionTool


async def main() -> None:
    async with JupyterCodeExecutor() as executor:
        tool = PythonCodeExecutionTool(executor)
        model_client = OpenAIChatCompletionClient(model="gpt-4o")
        agent = AssistantAgent("assistant", model_client=model_client, tools=[tool])
        result = await agent.run(task="What is the 10th Fibonacci number? Use Python to calculate it.")
        print(result)


asyncio.run(main())

在:class:`~autogen_agentchat.agents._code_executor_agent.CodeExecutorAgent`内部使用的示例：

import asyncio
from autogen_agentchat.agents import CodeExecutorAgent
from autogen_agentchat.messages import TextMessage
from autogen_ext.code_executors.jupyter import JupyterCodeExecutor
from autogen_core import CancellationToken


async def main() -> None:
    async with JupyterCodeExecutor() as executor:
        code_executor_agent = CodeExecutorAgent("code_executor", code_executor=executor)
        task = TextMessage(
            content='''Here is some code
    ```python
    print('Hello world')
    ```
    ''',
            source="user",
        )
        response = await code_executor_agent.on_messages([task], CancellationToken())
        print(response.chat_message)


asyncio.run(main())

参数:

• kernel_name (str) -- 使用的内核名称，默认为"python3"。

• timeout (int) -- 代码执行超时时间，默认为60。

• output_dir (Path) -- 输出文件保存目录，默认为临时目录。

备注

使用当前目录(".")作为输出目录已被弃用。使用它将引发弃用警告。

component_config_schema

JupyterCodeExecutorConfig 的别名

component_provider_override: ClassVar[str | None] = 'autogen_ext.code_executors.jupyter.JupyterCodeExecutor'

覆盖组件的provider字符串。这应该用于防止内部模块名称成为模块名称的一部分。

async execute_code_blocks(code_blocks: list[CodeBlock], cancellation_token: CancellationToken) → JupyterCodeResult

执行代码块并返回结果。

参数:

code_blocks (list[CodeBlock]) -- 要执行的代码块列表。

Returns:

JupyterCodeResult -- 代码执行的结果。

property output_dir: Path

async restart() → None

重启代码执行器。

async start() → None

(实验性) 启动代码执行器。

通过创建新的Notebook并配置指定的Jupyter内核来初始化Jupyter Notebook执行环境。 将执行器标记为已启动状态，允许执行代码。 在执行任何代码块之前应先调用此方法。

async stop() → None

（实验性）停止代码执行器。

通过退出内核上下文并清理相关资源来终止 Jupyter Notebook 的执行。

class JupyterCodeResult(exit_code: int, output: str, output_files: list[Path])

基类：CodeResult

Jupyter代码执行器的代码结果类。

output_files: list[Path]