mirror of
https://github.com/pchuan98/codex.git
synced 2026-07-01 00:31:56 +08:00
Ahmed Ibrahim
5a0f913426
## Why Goal continuation turns are emitted by the existing runtime as separate physical turns. The Python SDK needs private thread-scoped routing before it can present those notifications as one logical operation, without changing ordinary turn routing or the app-server protocol. ## What - add private goal operation state and thread-scoped notification routing - add internal wrappers for the existing `thread/goal/clear` and `thread/goal/set` RPCs - include existing goal notifications in the SDK notification union - preserve ordinary turn-ID routing unchanged - add focused routing coverage This PR does not expose a public goal API. It is the first PR in the Python goal operations stack. ## Test plan - online CI, including the Python SDK suite - focused typed-notification routing coverage
104 lines
3.0 KiB
Python
104 lines
3.0 KiB
Python
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from typing import TypeAlias
|
|
|
|
from pydantic import BaseModel
|
|
|
|
from .generated.v2_all import (
|
|
AccountLoginCompletedNotification,
|
|
AccountRateLimitsUpdatedNotification,
|
|
AccountUpdatedNotification,
|
|
AgentMessageDeltaNotification,
|
|
AppListUpdatedNotification,
|
|
CommandExecutionOutputDeltaNotification,
|
|
ConfigWarningNotification,
|
|
ContextCompactedNotification,
|
|
DeprecationNoticeNotification,
|
|
ErrorNotification,
|
|
FileChangeOutputDeltaNotification,
|
|
ItemCompletedNotification,
|
|
ItemStartedNotification,
|
|
McpServerOauthLoginCompletedNotification,
|
|
McpToolCallProgressNotification,
|
|
PlanDeltaNotification,
|
|
RawResponseItemCompletedNotification,
|
|
ReasoningSummaryPartAddedNotification,
|
|
ReasoningSummaryTextDeltaNotification,
|
|
ReasoningTextDeltaNotification,
|
|
TerminalInteractionNotification,
|
|
ThreadGoalClearedNotification,
|
|
ThreadGoalUpdatedNotification,
|
|
ThreadNameUpdatedNotification,
|
|
ThreadStartedNotification,
|
|
ThreadTokenUsageUpdatedNotification,
|
|
TurnCompletedNotification,
|
|
TurnDiffUpdatedNotification,
|
|
TurnPlanUpdatedNotification,
|
|
TurnStartedNotification,
|
|
WindowsWorldWritableWarningNotification,
|
|
)
|
|
|
|
JsonScalar: TypeAlias = str | int | float | bool | None
|
|
JsonValue: TypeAlias = JsonScalar | dict[str, "JsonValue"] | list["JsonValue"]
|
|
JsonObject: TypeAlias = dict[str, JsonValue]
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class UnknownNotification:
|
|
params: JsonObject
|
|
|
|
|
|
NotificationPayload: TypeAlias = (
|
|
AccountLoginCompletedNotification
|
|
| AccountRateLimitsUpdatedNotification
|
|
| AccountUpdatedNotification
|
|
| AgentMessageDeltaNotification
|
|
| AppListUpdatedNotification
|
|
| CommandExecutionOutputDeltaNotification
|
|
| ConfigWarningNotification
|
|
| ContextCompactedNotification
|
|
| DeprecationNoticeNotification
|
|
| ErrorNotification
|
|
| FileChangeOutputDeltaNotification
|
|
| ItemCompletedNotification
|
|
| ItemStartedNotification
|
|
| McpServerOauthLoginCompletedNotification
|
|
| McpToolCallProgressNotification
|
|
| PlanDeltaNotification
|
|
| RawResponseItemCompletedNotification
|
|
| ReasoningSummaryPartAddedNotification
|
|
| ReasoningSummaryTextDeltaNotification
|
|
| ReasoningTextDeltaNotification
|
|
| TerminalInteractionNotification
|
|
| ThreadNameUpdatedNotification
|
|
| ThreadGoalClearedNotification
|
|
| ThreadGoalUpdatedNotification
|
|
| ThreadStartedNotification
|
|
| ThreadTokenUsageUpdatedNotification
|
|
| TurnCompletedNotification
|
|
| TurnDiffUpdatedNotification
|
|
| TurnPlanUpdatedNotification
|
|
| TurnStartedNotification
|
|
| WindowsWorldWritableWarningNotification
|
|
| UnknownNotification
|
|
)
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class Notification:
|
|
method: str
|
|
payload: NotificationPayload
|
|
|
|
|
|
class ServerInfo(BaseModel):
|
|
name: str | None = None
|
|
version: str | None = None
|
|
|
|
|
|
class InitializeResponse(BaseModel):
|
|
serverInfo: ServerInfo | None = None
|
|
userAgent: str | None = None
|
|
platformFamily: str | None = None
|
|
platformOs: str | None = None
|