codex-python 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

codex/__init__.py

@@ -27,4 +27,4 @@ __all__ = [
 ]
 
 # Managed by Hatch via pyproject.toml [tool.hatch.version]
-__version__ = "0.1.1"
+__version__ = "0.1.2"

codex/api.py

@@ -51,12 +51,15 @@ def run_exec(
     prompt: str,
     *,
     model: str | None = None,
+    oss: bool = False,
     full_auto: bool = False,
     cd: str | None = None,
+    skip_git_repo_check: bool = False,
     timeout: float | None = None,
     env: Mapping[str, str] | None = None,
     executable: str = "codex",
     extra_args: Iterable[str] | None = None,
+    json: bool = False,
 ) -> str:
     """
     Run `codex exec` with the given prompt and return stdout as text.
@@ -72,12 +75,19 @@ def run_exec(
         cmd.extend(["--cd", cd])
     if model:
         cmd.extend(["-m", model])
+    if oss:
+        cmd.append("--oss")
     if full_auto:
         cmd.append("--full-auto")
+    if skip_git_repo_check:
+        cmd.append("--skip-git-repo-check")
     if extra_args:
         cmd.extend(list(extra_args))
 
-    cmd.extend(["exec", prompt])
+    cmd.append("exec")
+    if json:
+        cmd.append("--json")
+    cmd.append(prompt)
 
     completed = subprocess.run(
         cmd,
@@ -123,8 +133,10 @@ class CodexClient:
         prompt: str,
         *,
         model: str | None = None,
+        oss: bool | None = None,
         full_auto: bool | None = None,
         cd: str | None = None,
+        skip_git_repo_check: bool | None = None,
         timeout: float | None = None,
         env: Mapping[str, str] | None = None,
         extra_args: Iterable[str] | None = None,
@@ -136,6 +148,8 @@ class CodexClient:
         eff_model = model if model is not None else self.model
         eff_full_auto = full_auto if full_auto is not None else self.full_auto
         eff_cd = cd if cd is not None else self.cd
+        eff_oss = bool(oss) if oss is not None else False
+        eff_skip_git = bool(skip_git_repo_check) if skip_git_repo_check is not None else False
 
         # Merge environment overlays; run_exec will merge with os.environ
         merged_env: Mapping[str, str] | None
@@ -156,8 +170,10 @@ class CodexClient:
         return run_exec(
             prompt,
             model=eff_model,
+            oss=eff_oss,
             full_auto=eff_full_auto,
             cd=eff_cd,
+            skip_git_repo_check=eff_skip_git,
             timeout=timeout,
             env=merged_env,
             executable=self.executable,

codex/protocol/runtime.py

@@ -0,0 +1,80 @@
+from __future__ import annotations
+
+import json
+import os
+import subprocess
+from collections.abc import Iterator
+from typing import Any
+
+from pydantic import BaseModel
+
+
+class Event(BaseModel):
+    """Protocol event envelope emitted by `codex exec --json`.
+
+    Note: `msg` is kept as a raw mapping to preserve all fields from
+    intersection types. If you need strong typing, try validating
+    against `codex.protocol.types.EventMsg` manually.
+    """
+
+    id: str
+    msg: dict[str, Any]
+
+
+def stream_exec_events(
+    prompt: str,
+    *,
+    executable: str = "codex",
+    model: str | None = None,
+    oss: bool = False,
+    full_auto: bool = False,
+    cd: str | None = None,
+    skip_git_repo_check: bool = False,
+    env: dict[str, str] | None = None,
+) -> Iterator[Event]:
+    """Spawn `codex exec --json` and yield Event objects from NDJSON stdout.
+
+    Non-event lines (config summary, prompt echo) are ignored.
+    """
+    cmd: list[str] = [executable]
+    if cd:
+        cmd += ["--cd", cd]
+    if model:
+        cmd += ["-m", model]
+    if oss:
+        cmd.append("--oss")
+    if full_auto:
+        cmd.append("--full-auto")
+    if skip_git_repo_check:
+        cmd.append("--skip-git-repo-check")
+    cmd += ["exec", "--json", prompt]
+
+    with subprocess.Popen(
+        cmd,
+        stdout=subprocess.PIPE,
+        stderr=subprocess.PIPE,
+        text=True,
+        env={**os.environ, **(env or {})},
+    ) as proc:
+        assert proc.stdout is not None
+        for line in proc.stdout:
+            line = line.strip()
+            if not line:
+                continue
+            try:
+                obj = json.loads(line)
+            except json.JSONDecodeError:
+                continue
+
+            # Filter out non-event helper lines
+            if not isinstance(obj, dict):
+                continue
+            if "id" in obj and "msg" in obj:
+                # Attempt to validate into our Pydantic Event model
+                yield Event.model_validate(obj)
+
+        # Drain stderr for diagnostics if the process failed
+        ret = proc.wait()
+        if ret != 0 and proc.stderr is not None:
+            err = proc.stderr.read()
+            raise RuntimeError(f"codex exec failed with {ret}: {err}")