Skip to main content

Python SDK for the OpenAI Codex agent

Project description

Codex SDK for Python

Embed the Codex agent in your workflows and apps.

The Python SDK wraps the bundled codex binary. It spawns the CLI and exchanges JSONL events over stdin/stdout.

TypeScript版との互換性

このPython SDKは TypeScript版 Codex SDK を1対1でポーティングしており、全ての機能・データ構造・動作が同一です。

  • 全8種のイベント型(ThreadStartedEvent, TurnCompletedEvent など)
  • 全8種のアイテム型(AgentMessageItem, CommandExecutionItem など)
  • 全4種のEnum(ApprovalMode, SandboxMode, ModelReasoningEffort, WebSearchMode)
  • CLI引数の構築ロジック、環境変数の処理、TOML設定シリアライズ

詳細な比較は docs/typescript-python-comparison.md を参照してください。

Installation

pip install codex-sdk-py

Requires Python 3.10+.

Quickstart

import asyncio
from codex_sdk import Codex

async def main():
    codex = Codex()
    thread = codex.start_thread()
    turn = await thread.run("Diagnose the test failure and propose a fix")

    print(turn.final_response)
    print(turn.items)

asyncio.run(main())

Call run() repeatedly on the same Thread instance to continue that conversation.

next_turn = await thread.run("Implement the fix")

Streaming responses

run() buffers events until the turn finishes. To react to intermediate progress—tool calls, streaming responses, and file change notifications—use run_streamed() instead, which returns an async generator of structured events.

async def stream_example():
    codex = Codex()
    thread = codex.start_thread()

    streamed = await thread.run_streamed("Diagnose the test failure and propose a fix")

    async for event in streamed.events:
        match event.get("type"):
            case "item.completed":
                print("item", event.get("item"))
            case "turn.completed":
                print("usage", event.get("usage"))

Structured output

The Codex agent can produce a JSON response that conforms to a specified schema. The schema can be provided for each turn as a plain JSON object.

schema = {
    "type": "object",
    "properties": {
        "summary": {"type": "string"},
        "status": {"type": "string", "enum": ["ok", "action_required"]},
    },
    "required": ["summary", "status"],
    "additionalProperties": False,
}

turn = await thread.run("Summarize repository status", {"output_schema": schema})
print(turn.final_response)

You can also create a JSON schema from a Pydantic model using model_json_schema().

from pydantic import BaseModel
from typing import Literal

class StatusResponse(BaseModel):
    summary: str
    status: Literal["ok", "action_required"]

turn = await thread.run(
    "Summarize repository status",
    {"output_schema": StatusResponse.model_json_schema()}
)
print(turn.final_response)

Attaching images

Provide structured input entries when you need to include images alongside text. Text entries are concatenated into the final prompt while image entries are passed to the Codex CLI via --image.

turn = await thread.run([
    {"type": "text", "text": "Describe these screenshots"},
    {"type": "local_image", "path": "./ui.png"},
    {"type": "local_image", "path": "./diagram.jpg"},
])

Resuming an existing thread

Threads are persisted in ~/.codex/sessions. If you lose the in-memory Thread object, reconstruct it with resume_thread() and keep going.

import os

saved_thread_id = os.environ["CODEX_THREAD_ID"]
thread = codex.resume_thread(saved_thread_id)
await thread.run("Implement the fix")

Working directory controls

Codex runs in the current working directory by default. To avoid unrecoverable errors, Codex requires the working directory to be a Git repository. You can skip the Git repository check by passing the skip_git_repo_check option when creating a thread.

thread = codex.start_thread({
    "working_directory": "/path/to/project",
    "skip_git_repo_check": True,
})

Sandbox modes

Control how the agent interacts with your filesystem using sandbox_mode.

from codex_sdk import Codex, SandboxMode

thread = codex.start_thread({
    "sandbox_mode": SandboxMode.WORKSPACE_WRITE,
})

Available modes:

  • SandboxMode.READ_ONLY - Agent can only read files
  • SandboxMode.WORKSPACE_WRITE - Agent can read/write in the workspace
  • SandboxMode.DANGER_FULL_ACCESS - Full filesystem access (use with caution)

Approval policies

Control when the agent requires approval for actions.

from codex_sdk import Codex, ApprovalMode

thread = codex.start_thread({
    "approval_policy": ApprovalMode.ON_FAILURE,
})

Available modes:

  • ApprovalMode.NEVER - Never require approval
  • ApprovalMode.ON_REQUEST - Approve on explicit request
  • ApprovalMode.ON_FAILURE - Approve after failures
  • ApprovalMode.UNTRUSTED - Always require approval

Cancelling a turn

Use asyncio.Event to cancel an ongoing turn (equivalent to AbortSignal in TypeScript).

import asyncio

async def cancellable_example():
    codex = Codex()
    thread = codex.start_thread()

    cancel_event = asyncio.Event()

    async def cancel_after_delay():
        await asyncio.sleep(5)
        cancel_event.set()

    # Start cancellation timer
    asyncio.create_task(cancel_after_delay())

    try:
        turn = await thread.run(
            "Long running task",
            {"cancel_event": cancel_event}
        )
    except asyncio.CancelledError:
        print("Turn was cancelled")

Controlling the Codex CLI environment

By default, the Codex CLI inherits the Python process environment. Provide the optional env parameter when instantiating the Codex client to fully control which variables the CLI receives—useful for sandboxed hosts.

codex = Codex({
    "env": {
        "PATH": "/usr/local/bin",
    },
})

The SDK still injects its required variables (such as OPENAI_BASE_URL and CODEX_API_KEY) on top of the environment you provide.

Passing --config overrides

Use the config option to provide additional Codex CLI configuration overrides. The SDK accepts a dict, flattens it into dotted paths, and serializes values as TOML literals before passing them as repeated --config key=value flags.

codex = Codex({
    "config": {
        "show_raw_agent_reasoning": True,
        "sandbox_workspace_write": {"network_access": True},
    },
})

Thread options still take precedence for overlapping settings because they are emitted after these global overrides.

API Reference

Classes

Codex

Main entry point for the SDK.

codex = Codex(options: CodexOptions | None = None)

Methods:

  • start_thread(options: ThreadOptions | None = None) -> Thread - Start a new conversation
  • resume_thread(thread_id: str, options: ThreadOptions | None = None) -> Thread - Resume an existing conversation

Thread

Represents a conversation with the agent.

Properties:

  • id: str | None - Thread ID (populated after first turn)

Methods:

  • async run(input: Input, turn_options: TurnOptions | None = None) -> Turn - Execute a turn and return results
  • async run_streamed(input: Input, turn_options: TurnOptions | None = None) -> StreamedTurn - Execute a turn with streaming events

Turn

Result of a completed turn.

@dataclass
class Turn:
    items: list[ThreadItem]      # All items produced during the turn
    final_response: str          # The agent's final response text
    usage: Usage | None          # Token usage statistics

StreamedTurn

Result of a streamed turn.

@dataclass
class StreamedTurn:
    events: AsyncGenerator[ThreadEvent, None]  # Async generator of events

Types

Input Types

  • Input = str | list[UserInput] - User input (string or structured)
  • TextUserInput - Text input: {"type": "text", "text": "..."}
  • ImageUserInput - Image input: {"type": "local_image", "path": "..."}

Event Types

  • ThreadStartedEvent - Thread started
  • TurnStartedEvent - Turn started
  • TurnCompletedEvent - Turn completed (includes usage)
  • TurnFailedEvent - Turn failed (includes error)
  • ItemStartedEvent - Item started
  • ItemUpdatedEvent - Item updated
  • ItemCompletedEvent - Item completed
  • ThreadErrorEvent - Unrecoverable error

Item Types

  • AgentMessageItem - Agent's response text
  • ReasoningItem - Agent's reasoning summary
  • CommandExecutionItem - Shell command execution
  • FileChangeItem - File modifications
  • McpToolCallItem - MCP tool invocation
  • WebSearchItem - Web search query
  • TodoListItem - Agent's task list
  • ErrorItem - Non-fatal error

Enums

class SandboxMode(StrEnum):
    READ_ONLY = "read-only"
    WORKSPACE_WRITE = "workspace-write"
    DANGER_FULL_ACCESS = "danger-full-access"

class ApprovalMode(StrEnum):
    NEVER = "never"
    ON_REQUEST = "on-request"
    ON_FAILURE = "on-failure"
    UNTRUSTED = "untrusted"

class ModelReasoningEffort(StrEnum):
    MINIMAL = "minimal"
    LOW = "low"
    MEDIUM = "medium"
    HIGH = "high"
    XHIGH = "xhigh"

class WebSearchMode(StrEnum):
    DISABLED = "disabled"
    CACHED = "cached"
    LIVE = "live"

Development

Setup

# Clone the repository
git clone https://github.com/nogataka/codex-sdk-py.git
cd codex-sdk-py

# Install development dependencies
pip install -e ".[dev]"

Running tests

pytest tests/ -v

Linting and formatting

# Lint
ruff check codex_sdk/

# Format
ruff format codex_sdk/

# Type check
mypy codex_sdk/

Release Process

Releases are automated via GitHub Actions.

Creating a release

  1. Update the version in codex_sdk/__init__.py:

    __version__ = "0.1.0"
    
  2. Commit the version change:

    git add codex_sdk/__init__.py
    git commit -m "Bump version to 0.1.0"
    
  3. Create and push a tag:

    git tag v0.1.0
    git push origin main --tags
    
  4. GitHub Actions will automatically:

    • Build the package
    • Publish to PyPI
    • Create a GitHub Release

Required secrets

Set the following secrets in your GitHub repository:

  • PYPI_TOKEN: Your PyPI API token for publishing packages

To create a PyPI token:

  1. Go to https://pypi.org/manage/account/token/
  2. Create a new token with "Upload packages" scope
  3. Add it to your GitHub repository secrets

License

Apache-2.0

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

codex_sdk_py-0.0.5.tar.gz (21.3 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

codex_sdk_py-0.0.5-py3-none-any.whl (19.2 kB view details)

Uploaded Python 3

File details

Details for the file codex_sdk_py-0.0.5.tar.gz.

File metadata

  • Download URL: codex_sdk_py-0.0.5.tar.gz
  • Upload date:
  • Size: 21.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for codex_sdk_py-0.0.5.tar.gz
Algorithm Hash digest
SHA256 22581f19c151b649c6c88d002e2d5e77b5f88b3d5621e8a3153505ccc66e05e7
MD5 2ee6436f0dd6bb0fda8ce3b255485642
BLAKE2b-256 06d29eee421e136cb767d0ba3a61ba490223ab7edc51d7c7e94270cb844c7677

See more details on using hashes here.

File details

Details for the file codex_sdk_py-0.0.5-py3-none-any.whl.

File metadata

  • Download URL: codex_sdk_py-0.0.5-py3-none-any.whl
  • Upload date:
  • Size: 19.2 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for codex_sdk_py-0.0.5-py3-none-any.whl
Algorithm Hash digest
SHA256 a1e7b3056d87117e3fe03dbc43aaa7b16717111982d5f1cb4df58f74d62387f9
MD5 144aeb8822537c251fcd050bedcf0588
BLAKE2b-256 3b94a6d852fe6ed1f979de16b5ed97e18ea46f36b48f723b99e1f8d9e232f362

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page