python-codex 0.2.7__py3-none-any.whl → 0.3.0__py3-none-any.whl

pycodex/interactive_session.py

@@ -1,415 +0,0 @@
-import asyncio
-import json
-from dataclasses import asdict
-
-from .protocol import AgentEvent
-from .runtime import CliSubmissionQueue
-from .runtime_services import create_agent_runtime_environment
-from .utils import CliSessionView, uuid7_string
-from .utils.compactor import compact_agent
-from .utils.session_persist import (
-    SessionRolloutRecorder,
-    conversation_history_to_turns,
-    list_resumable_sessions,
-    load_resumed_session,
-    resolve_codex_home,
-)
-import typing
-
-
-EXIT_COMMANDS = {"/exit", "/quit"}
-HELP_COMMAND = "/help"
-HISTORY_COMMAND = "/history"
-TITLE_COMMAND = "/title"
-MODEL_COMMAND = "/model"
-QUEUE_COMMAND = "/queue"
-RESUME_COMMAND = "/resume"
-COMPACT_COMMAND = "/compact"
-FORK_COMMAND = "/fork"
-LINK_COMMAND = "/link"
-UNLINK_COMMAND = "/unlink"
-EXTRA_COMMANDS_LINE = (
-    "Extra commands: /help, /history, /title, /model, /resume, /compact, /fork, /link, /unlink"
-)
-
-
-def format_turn_output(result, json_mode: "bool") -> "str":
-    if json_mode:
-        return json.dumps(asdict(result), ensure_ascii=False, indent=2)
-    return result.output_text or ""
-
-
-async def prompt_request_user_input(
-    view,
-    payload: "typing.Dict[str, object]",
-) -> "typing.Union[typing.Dict[str, object], None]":
-    view.finish_stream()
-    view.write_line("[request_user_input] waiting for user response")
-    answers: "typing.Dict[str, typing.Dict[str, typing.List[str]]]" = {}
-    for question in payload.get("questions", []):
-        if not isinstance(question, dict):
-            continue
-        header = str(question.get("header", "")).strip()
-        question_text = str(question.get("question", "")).strip()
-        question_id = str(question.get("id", "")).strip()
-        if header:
-            view.write_line(f"[{header}] {question_text}")
-        else:
-            view.write_line(question_text)
-
-        options = question.get("options") or []
-        if isinstance(options, list):
-            for index, option in enumerate(options, start=1):
-                if not isinstance(option, dict):
-                    continue
-                label = str(option.get("label", "")).strip()
-                description = str(option.get("description", "")).strip()
-                view.write_line(f"  {index}. {label} - {description}")
-        view.write_line("  0. Other")
-
-        try:
-            raw_answer = await view.get_prompt("answer> ")
-        except EOFError:
-            return None
-        answer_text = raw_answer.strip()
-        if not answer_text:
-            return None
-
-        selected_answer = answer_text
-        if answer_text.isdigit() and isinstance(options, list):
-            choice = int(answer_text)
-            if 1 <= choice <= len(options):
-                option = options[choice - 1]
-                if isinstance(option, dict):
-                    selected_answer = (
-                        str(option.get("label", "")).strip() or answer_text
-                    )
-            elif choice == 0:
-                try:
-                    raw_answer = await view.get_prompt("other> ")
-                except EOFError:
-                    return None
-                selected_answer = raw_answer.strip()
-                if not selected_answer:
-                    return None
-
-        answers[question_id] = {"answers": [selected_answer]}
-
-    return {"answers": answers}
-
-
-async def prompt_request_permissions(
-    view,
-    payload: "typing.Dict[str, object]",
-) -> "typing.Union[typing.Dict[str, object], None]":
-    view.finish_stream()
-    view.write_line("[request_permissions] user approval required")
-    reason = payload.get("reason")
-    if reason:
-        view.write_line(f"Reason: {reason}")
-    view.write_line("Requested permissions:")
-    view.write_line(
-        json.dumps(payload.get("permissions", {}), ensure_ascii=False, indent=2)
-    )
-    view.write_line("Choose: [n] deny / [t] grant for turn / [s] grant for session")
-    try:
-        raw_answer = await view.get_prompt("permissions> ")
-    except EOFError:
-        return None
-
-    answer = raw_answer.strip().lower()
-    if answer in {"t", "turn", "y", "yes"}:
-        return {
-            "permissions": payload.get("permissions", {}),
-            "scope": "turn",
-        }
-    if answer in {"s", "session"}:
-        return {
-            "permissions": payload.get("permissions", {}),
-            "scope": "session",
-        }
-    return {
-        "permissions": {},
-        "scope": "turn",
-    }
-
-
-async def run_interactive_session(
-    queue: "CliSubmissionQueue",
-    json_mode: "bool",
-    config_path: "typing.Union[str, None]" = None,
-    view=None,
-    view_factory=None,
-    show_banner: "bool" = True,
-) -> "int":
-    worker = asyncio.create_task(queue.run_forever())
-    context_window_tokens = queue._agent._context_manager.resolve_model_context_window()
-    if view is None:
-        factory = view_factory or CliSessionView
-        view = factory()
-    view.set_context_window_tokens(context_window_tokens)
-    model_client = queue._agent._model_client
-    codex_home = resolve_codex_home(config_path)
-    queue.set_event_handler(view.handle_event)
-    pending_turn_tasks: "typing.Set[asyncio.Task[None]]" = set()
-    runtime_environment = queue._agent.runtime_environment
-    if runtime_environment is None:
-        runtime_environment = create_agent_runtime_environment()
-        queue._agent.runtime_environment = runtime_environment
-    runtime_environment.request_user_input_manager.set_handler(
-        lambda payload: prompt_request_user_input(view, payload)
-    )
-    runtime_environment.request_permissions_manager.set_handler(
-        lambda payload: prompt_request_permissions(view, payload)
-    )
-    if show_banner:
-        view.write_line("pycodex interactive mode. Type /exit to quit.")
-        view.write_line(EXTRA_COMMANDS_LINE)
-    feishu_link = None
-    try:
-
-        def has_pending_turn_tasks() -> "bool":
-            pending_turn_tasks.difference_update(
-                task for task in tuple(pending_turn_tasks) if task.done()
-            )
-            return bool(pending_turn_tasks)
-
-        async def run_manual_compact() -> "None":
-            current_agent = queue._agent
-            if not current_agent.history:
-                view.write_line("Nothing to compact.")
-                return
-
-            compact_turn_id = uuid7_string()
-
-            def handle_compact_stream_event(event) -> "None":
-                if event.kind not in {"token_count", "stream_error"}:
-                    return
-                view.handle_event(
-                    AgentEvent(
-                        kind=event.kind,
-                        turn_id=compact_turn_id,
-                        payload=dict(event.payload),
-                    )
-                )
-
-            view.write_line("Compacting conversation history...")
-            compact_result = await compact_agent(
-                current_agent,
-                handle_compact_stream_event,
-                True,
-            )
-            if compact_result is None:
-                view.write_line("Nothing to compact.")
-                return
-            view.load_session_history(
-                getattr(view, "_title", None),
-                conversation_history_to_turns(compact_result.history),
-            )
-            view.write_line(compact_result.display_text())
-
-        async def wait_for_turn_result(future) -> "None":
-            try:
-                result = await future
-            except Exception as exc:  # pragma: no cover - defensive surface
-                if str(exc) == "submission interrupted":
-                    return
-                view.show_error(str(exc))
-                return
-
-            if json_mode:
-                view.write_line(format_turn_output(result, True))
-
-        while True:
-            try:
-                raw_line = await view.poll_prompt()
-            except EOFError:
-                break
-            if raw_line is None:
-                await asyncio.sleep(0.05)
-                continue
-
-            prompt_text = raw_line.strip()
-            if not prompt_text:
-                continue
-            if prompt_text in EXIT_COMMANDS:
-                break
-            if prompt_text == HELP_COMMAND:
-                view.write_line(EXTRA_COMMANDS_LINE)
-                continue
-            if prompt_text == HISTORY_COMMAND:
-                view.show_history()
-                continue
-            if prompt_text == TITLE_COMMAND:
-                view.show_title()
-                continue
-            if prompt_text.startswith(f"{TITLE_COMMAND} "):
-                title = prompt_text[len(TITLE_COMMAND) :].strip()
-                if not title:
-                    view.write_line("Usage: /title <title>")
-                    continue
-                set_session_title = getattr(view, "set_session_title", None)
-                if callable(set_session_title):
-                    set_session_title(title)
-                else:
-                    view.write_line(f"Session: {title}")
-                continue
-            if prompt_text == RESUME_COMMAND:
-                sessions = list_resumable_sessions(codex_home)
-                if not sessions:
-                    view.write_line("No resumable sessions found.")
-                    continue
-                view.write_line("Available sessions:")
-                for index, session in enumerate(sessions, start=1):
-                    view.write_line(f"[{index}] {session['preview']}")
-                continue
-            if prompt_text.startswith(f"{RESUME_COMMAND} "):
-                if has_pending_turn_tasks():
-                    view.write_line(
-                        "Cannot resume while work is running or queued."
-                    )
-                    continue
-                resume_target = prompt_text[len(RESUME_COMMAND) :].strip()
-                try:
-                    resumed = load_resumed_session(codex_home, resume_target)
-                    queue._agent.replace_history(resumed["history"])
-                    if hasattr(model_client, "_session_id"):
-                        model_client._session_id = str(resumed["session_id"])
-                    queue._agent.set_rollout_recorder(
-                        SessionRolloutRecorder.resume(resumed["rollout_path"])
-                    )
-                    view.load_session_history(
-                        str(resumed["title"]),
-                        tuple(resumed["turns"]),
-                    )
-                    show_resumed_session = getattr(view, "show_resumed_session", None)
-                    if callable(show_resumed_session):
-                        show_resumed_session(str(resumed["title"]))
-                    else:
-                        view.write_line(f"Resumed session: {resumed['title']}")
-                        view.show_history()
-                except Exception as exc:  # pragma: no cover - defensive surface
-                    view.show_error(str(exc))
-                continue
-            if prompt_text == COMPACT_COMMAND:
-                if has_pending_turn_tasks():
-                    view.write_line(
-                        "Cannot compact while work is running or queued."
-                    )
-                    continue
-                try:
-                    await run_manual_compact()
-                except Exception as exc:  # pragma: no cover - defensive surface
-                    view.show_error(str(exc))
-                continue
-            if prompt_text == FORK_COMMAND:
-                if has_pending_turn_tasks():
-                    view.write_line(
-                        "Cannot fork while work is running or queued."
-                    )
-                    continue
-                if not hasattr(model_client, "_session_id"):
-                    view.write_line("Current model does not support session IDs.")
-                    continue
-                new_session_id = uuid7_string()
-                model_client._session_id = new_session_id
-                view.write_line(f"Forked session: {new_session_id}")
-                continue
-            if prompt_text.startswith(f"{LINK_COMMAND} "):
-                link_target = prompt_text[len(LINK_COMMAND) :].strip()
-                if not link_target:
-                    view.write_line("Usage: /link <feishu-email|open_id|chat_id>")
-                    continue
-                if feishu_link:
-                    view.write_line("A Feishu card is already linked. Use /unlink first.")
-                    continue
-                try:
-                    from .feishu_link import PycodexRuntimeLink
-
-                    view.write_line(f"Linking Feishu card to current session: {link_target}")
-                    link = await PycodexRuntimeLink(
-                        queue,
-                        link_target,
-                    ).start_async()
-                    feishu_link = link
-                    view.write_line(
-                        "Linked Feishu card: session_key={0} message_id={1}".format(
-                            link.session_key,
-                            link.message_id or "-",
-                        )
-                    )
-                except Exception as exc:  # pragma: no cover - defensive surface
-                    view.show_error(str(exc))
-                continue
-            if prompt_text == UNLINK_COMMAND:
-                if not feishu_link:
-                    view.write_line("No Feishu card is linked.")
-                    continue
-                feishu_link.detach()
-                feishu_link = None
-                view.write_line("Unlinked Feishu card.")
-                continue
-            if prompt_text.startswith(f"{QUEUE_COMMAND} "):
-                queued_text = prompt_text[len(QUEUE_COMMAND) :].strip()
-                if not queued_text:
-                    view.write_line("Usage: /queue <message>")
-                    continue
-                try:
-                    submission_id, future = await queue.enqueue_user_turn(
-                        queued_text, queue="enqueue"
-                    )
-                    view.show_steer_queued(submission_id, queued_text)
-                    turn_task = asyncio.create_task(wait_for_turn_result(future))
-                    pending_turn_tasks.add(turn_task)
-                except Exception as exc:  # pragma: no cover - defensive surface
-                    view.show_error(str(exc))
-                continue
-            if prompt_text == MODEL_COMMAND:
-                view.write_line(
-                    f"Current model: {getattr(model_client, 'model', None) or 'unavailable'}"
-                )
-                models = await model_client.list_models()
-                view.write_line(f"Available models: {', '.join(models)}")
-                continue
-            if prompt_text.startswith(f"{MODEL_COMMAND} "):
-                if has_pending_turn_tasks():
-                    view.write_line(
-                        "Cannot change model while work is running or queued in steer mode."
-                    )
-                    continue
-                model_name = prompt_text[len(MODEL_COMMAND) :].strip()
-                if not model_name:
-                    view.write_line("Usage: /model <model>")
-                    continue
-
-                model_client.model = model_name
-                view.write_line(f"Switched model to {model_name}.")
-                continue
-
-            try:
-                steered = has_pending_turn_tasks()
-                submission_id, future = await queue.enqueue_user_turn(
-                    prompt_text,
-                    queue="steer",
-                )
-                if steered:
-                    view.schedule_steer_inserted(submission_id, prompt_text)
-                turn_task = asyncio.create_task(wait_for_turn_result(future))
-                pending_turn_tasks.add(turn_task)
-                continue
-            except Exception as exc:  # pragma: no cover - defensive surface
-                view.show_error(str(exc))
-                continue
-    finally:
-        if feishu_link:
-            feishu_link.detach()
-            feishu_link.stop()
-        runtime_environment.request_user_input_manager.set_handler(None)
-        runtime_environment.request_permissions_manager.set_handler(None)
-        await queue.shutdown()
-        await worker
-        if pending_turn_tasks:
-            await asyncio.gather(*pending_turn_tasks, return_exceptions=True)
-        view.close()
-
-    return 0

pycodex/prompts/collaboration_default.md

@@ -1,11 +0,0 @@
-# Collaboration Mode: Default
-
-You are now in Default mode. Any previous instructions for other modes (e.g. Plan mode) are no longer active.
-
-Your active mode changes only when new developer instructions with a different `<collaboration_mode>...</collaboration_mode>` change it; user requests or tool descriptions do not change mode by themselves. Known mode names are Default and Plan.
-
-## request_user_input availability
-
-The `request_user_input` tool is unavailable in Default mode. If you call it while in Default mode, it will return an error.
-
-In Default mode, strongly prefer making reasonable assumptions and executing the user's request rather than stopping to ask questions. If you absolutely must ask a question because the answer cannot be discovered from local context and a reasonable assumption would be risky, ask the user directly with a concise plain-text question. Never write a multiple choice question as a textual assistant message.

pycodex/prompts/collaboration_plan.md

@@ -1,128 +0,0 @@
-# Plan Mode (Conversational)
-
-You work in 3 phases, and you should *chat your way* to a great plan before finalizing it. A great plan is very detailed—intent- and implementation-wise—so that it can be handed to another engineer or agent to be implemented right away. It must be **decision complete**, where the implementer does not need to make any decisions.
-
-## Mode rules (strict)
-
-You are in **Plan Mode** until a developer message explicitly ends it.
-
-Plan Mode is not changed by user intent, tone, or imperative language. If a user asks for execution while still in Plan Mode, treat it as a request to **plan the execution**, not perform it.
-
-## Plan Mode vs update_plan tool
-
-Plan Mode is a collaboration mode that can involve requesting user input and eventually issuing a `<proposed_plan>` block.
-
-Separately, `update_plan` is a checklist/progress/TODOs tool; it does not enter or exit Plan Mode. Do not confuse it with Plan mode or try to use it while in Plan mode. If you try to use `update_plan` in Plan mode, it will return an error.
-
-## Execution vs. mutation in Plan Mode
-
-You may explore and execute **non-mutating** actions that improve the plan. You must not perform **mutating** actions.
-
-### Allowed (non-mutating, plan-improving)
-
-Actions that gather truth, reduce ambiguity, or validate feasibility without changing repo-tracked state. Examples:
-
-* Reading or searching files, configs, schemas, types, manifests, and docs
-* Static analysis, inspection, and repo exploration
-* Dry-run style commands when they do not edit repo-tracked files
-* Tests, builds, or checks that may write to caches or build artifacts (for example, `target/`, `.cache/`, or snapshots) so long as they do not edit repo-tracked files
-
-### Not allowed (mutating, plan-executing)
-
-Actions that implement the plan or change repo-tracked state. Examples:
-
-* Editing or writing files
-* Running formatters or linters that rewrite files
-* Applying patches, migrations, or codegen that updates repo-tracked files
-* Side-effectful commands whose purpose is to carry out the plan rather than refine it
-
-When in doubt: if the action would reasonably be described as "doing the work" rather than "planning the work," do not do it.
-
-## PHASE 1 — Ground in the environment (explore first, ask second)
-
-Begin by grounding yourself in the actual environment. Eliminate unknowns in the prompt by discovering facts, not by asking the user. Resolve all questions that can be answered through exploration or inspection. Identify missing or ambiguous details only if they cannot be derived from the environment. Silent exploration between turns is allowed and encouraged.
-
-Before asking the user any question, perform at least one targeted non-mutating exploration pass (for example: search relevant files, inspect likely entrypoints/configs, confirm current implementation shape), unless no local environment/repo is available.
-
-Exception: you may ask clarifying questions about the user's prompt before exploring, ONLY if there are obvious ambiguities or contradictions in the prompt itself. However, if ambiguity might be resolved by exploring, always prefer exploring first.
-
-Do not ask questions that can be answered from the repo or system (for example, "where is this struct?" or "which UI component should we use?" when exploration can make it clear). Only ask once you have exhausted reasonable non-mutating exploration.
-
-## PHASE 2 — Intent chat (what they actually want)
-
-* Keep asking until you can clearly state: goal + success criteria, audience, in/out of scope, constraints, current state, and the key preferences/tradeoffs.
-* Bias toward questions over guessing: if any high-impact ambiguity remains, do NOT plan yet—ask.
-
-## PHASE 3 — Implementation chat (what/how we’ll build)
-
-* Once intent is stable, keep asking until the spec is decision complete: approach, interfaces (APIs/schemas/I/O), data flow, edge cases/failure modes, testing + acceptance criteria, rollout/monitoring, and any migrations/compat constraints.
-
-## Asking questions
-
-Critical rules:
-
-* Strongly prefer using the `request_user_input` tool to ask any questions.
-* Offer only meaningful multiple‑choice options; don’t include filler choices that are obviously wrong or irrelevant.
-* In rare cases where an unavoidable, important question can’t be expressed with reasonable multiple‑choice options (due to extreme ambiguity), you may ask it directly without the tool.
-
-You SHOULD ask many questions, but each question must:
-
-* materially change the spec/plan, OR
-* confirm/lock an assumption, OR
-* choose between meaningful tradeoffs.
-* not be answerable by non-mutating commands.
-
-Use the `request_user_input` tool only for decisions that materially change the plan, for confirming important assumptions, or for information that cannot be discovered via non-mutating exploration.
-
-## Two kinds of unknowns (treat differently)
-
-1. **Discoverable facts** (repo/system truth): explore first.
-
-   * Before asking, run targeted searches and check likely sources of truth (configs/manifests/entrypoints/schemas/types/constants).
-   * Ask only if: multiple plausible candidates; nothing found but you need a missing identifier/context; or ambiguity is actually product intent.
-   * If asking, present concrete candidates (paths/service names) + recommend one.
-   * Never ask questions you can answer from your environment (e.g., “where is this struct”).
-
-2. **Preferences/tradeoffs** (not discoverable): ask early.
-
-   * These are intent or implementation preferences that cannot be derived from exploration.
-   * Provide 2–4 mutually exclusive options + a recommended default.
-   * If unanswered, proceed with the recommended option and record it as an assumption in the final plan.
-
-## Finalization rule
-
-Only output the final plan when it is decision complete and leaves no decisions to the implementer.
-
-When you present the official plan, wrap it in a `<proposed_plan>` block so the client can render it specially:
-
-1) The opening tag must be on its own line.
-2) Start the plan content on the next line (no text on the same line as the tag).
-3) The closing tag must be on its own line.
-4) Use Markdown inside the block.
-5) Keep the tags exactly as `<proposed_plan>` and `</proposed_plan>` (do not translate or rename them), even if the plan content is in another language.
-
-Example:
-
-<proposed_plan>
-plan content
-</proposed_plan>
-
-plan content should be human and agent digestible. The final plan must be plan-only, concise by default, and include:
-
-* A clear title
-* A brief summary section
-* Important changes or additions to public APIs/interfaces/types
-* Test cases and scenarios
-* Explicit assumptions and defaults chosen where needed
-
-When possible, prefer a compact structure with 3-5 short sections, usually: Summary, Key Changes or Implementation Changes, Test Plan, and Assumptions. Do not include a separate Scope section unless scope boundaries are genuinely important to avoid mistakes.
-
-Prefer grouped implementation bullets by subsystem or behavior over file-by-file inventories. Mention files only when needed to disambiguate a non-obvious change, and avoid naming more than 3 paths unless extra specificity is necessary to prevent mistakes. Prefer behavior-level descriptions over symbol-by-symbol removal lists. For v1 feature-addition plans, do not invent detailed schema, validation, precedence, fallback, or wire-shape policy unless the request establishes it or it is needed to prevent a concrete implementation mistake; prefer the intended capability and minimum interface/behavior changes.
-
-Keep bullets short and avoid explanatory sub-bullets unless they are needed to prevent ambiguity. Prefer the minimum detail needed for implementation safety, not exhaustive coverage. Within each section, compress related changes into a few high-signal bullets and omit branch-by-branch logic, repeated invariants, and long lists of unaffected behavior unless they are necessary to prevent a likely implementation mistake. Avoid repeated repo facts and irrelevant edge-case or rollout detail. For straightforward refactors, keep the plan to a compact summary, key edits, tests, and assumptions. If the user asks for more detail, then expand.
-
-Do not ask "should I proceed?" in the final output. The user can easily switch out of Plan mode and request implementation if you have included a `<proposed_plan>` block in your response. Alternatively, they can decide to stay in Plan mode and continue refining the plan.
-
-Only produce at most one `<proposed_plan>` block per turn, and only when you are presenting a complete spec.
-
-If the user stays in Plan mode and asks for revisions after a prior `<proposed_plan>`, any new `<proposed_plan>` must be a complete replacement.