From f57b5455c83abfc110db88f728e0330bfba5f9f9 Mon Sep 17 00:00:00 2001 From: Brett Cannon Date: Wed, 18 Mar 2026 13:30:46 -0700 Subject: [PATCH 1/2] Declare various modules private - `telemetry` - `jsonrpc` - `sdk_protocol_version` --- nodejs/scripts/update-protocol-version.ts | 6 +++--- python/copilot/{jsonrpc.py => _jsonrpc.py} | 0 .../{sdk_protocol_version.py => _sdk_protocol_version.py} | 0 python/copilot/{telemetry.py => _telemetry.py} | 0 python/copilot/client.py | 6 +++--- python/copilot/generated/rpc.py | 6 +++--- python/copilot/session.py | 4 ++-- python/test_jsonrpc.py | 2 +- python/test_telemetry.py | 2 +- scripts/codegen/python.ts | 6 +++--- 10 files changed, 16 insertions(+), 16 deletions(-) rename python/copilot/{jsonrpc.py => _jsonrpc.py} (100%) rename python/copilot/{sdk_protocol_version.py => _sdk_protocol_version.py} (100%) rename python/copilot/{telemetry.py => _telemetry.py} (100%) diff --git a/nodejs/scripts/update-protocol-version.ts b/nodejs/scripts/update-protocol-version.ts index 46f6189e8..a18a560c7 100644 --- a/nodejs/scripts/update-protocol-version.ts +++ b/nodejs/scripts/update-protocol-version.ts @@ -8,7 +8,7 @@ * Reads from sdk-protocol-version.json and generates: * - nodejs/src/sdkProtocolVersion.ts * - go/sdk_protocol_version.go - * - python/copilot/sdk_protocol_version.py + * - python/copilot/_sdk_protocol_version.py * - dotnet/src/SdkProtocolVersion.cs * * Run this script whenever the protocol version changes. @@ -89,8 +89,8 @@ def get_sdk_protocol_version() -> int: """ return SDK_PROTOCOL_VERSION `; -fs.writeFileSync(path.join(rootDir, "python", "copilot", "sdk_protocol_version.py"), pythonCode); -console.log(" ✓ python/copilot/sdk_protocol_version.py"); +fs.writeFileSync(path.join(rootDir, "python", "copilot", "_sdk_protocol_version.py"), pythonCode); +console.log(" ✓ python/copilot/_sdk_protocol_version.py"); // Generate C# const csharpCode = `// Code generated by update-protocol-version.ts. DO NOT EDIT. diff --git a/python/copilot/jsonrpc.py b/python/copilot/_jsonrpc.py similarity index 100% rename from python/copilot/jsonrpc.py rename to python/copilot/_jsonrpc.py diff --git a/python/copilot/sdk_protocol_version.py b/python/copilot/_sdk_protocol_version.py similarity index 100% rename from python/copilot/sdk_protocol_version.py rename to python/copilot/_sdk_protocol_version.py diff --git a/python/copilot/telemetry.py b/python/copilot/_telemetry.py similarity index 100% rename from python/copilot/telemetry.py rename to python/copilot/_telemetry.py diff --git a/python/copilot/client.py b/python/copilot/client.py index 0d8074fe0..81c1459f2 100644 --- a/python/copilot/client.py +++ b/python/copilot/client.py @@ -25,12 +25,12 @@ from pathlib import Path from typing import Any, cast, overload +from ._jsonrpc import JsonRpcClient, ProcessExitedError +from ._sdk_protocol_version import get_sdk_protocol_version +from ._telemetry import get_trace_context, trace_context from .generated.rpc import ServerRpc from .generated.session_events import PermissionRequest, session_event_from_dict -from .jsonrpc import JsonRpcClient, ProcessExitedError -from .sdk_protocol_version import get_sdk_protocol_version from .session import CopilotSession -from .telemetry import get_trace_context, trace_context from .types import ( ConnectionState, CustomAgentConfig, diff --git a/python/copilot/generated/rpc.py b/python/copilot/generated/rpc.py index 29b7463df..a33c34c05 100644 --- a/python/copilot/generated/rpc.py +++ b/python/copilot/generated/rpc.py @@ -6,7 +6,7 @@ from typing import TYPE_CHECKING if TYPE_CHECKING: - from ..jsonrpc import JsonRpcClient + from .._jsonrpc import JsonRpcClient from dataclasses import dataclass @@ -510,9 +510,9 @@ def to_dict(self) -> dict: class Mode(Enum): """The current agent mode. - + The agent mode after switching. - + The mode to switch to. Valid values: "interactive", "plan", "autopilot". """ AUTOPILOT = "autopilot" diff --git a/python/copilot/session.py b/python/copilot/session.py index e4a17f2f9..90d156c4c 100644 --- a/python/copilot/session.py +++ b/python/copilot/session.py @@ -11,6 +11,8 @@ from collections.abc import Callable from typing import Any, Literal, cast +from ._jsonrpc import JsonRpcError, ProcessExitedError +from ._telemetry import get_trace_context, trace_context from .generated.rpc import ( Kind, Level, @@ -23,8 +25,6 @@ SessionToolsHandlePendingToolCallParams, ) from .generated.session_events import SessionEvent, SessionEventType, session_event_from_dict -from .jsonrpc import JsonRpcError, ProcessExitedError -from .telemetry import get_trace_context, trace_context from .types import ( Attachment, PermissionRequest, diff --git a/python/test_jsonrpc.py b/python/test_jsonrpc.py index 7c3c8dab2..c0ab2c6f4 100644 --- a/python/test_jsonrpc.py +++ b/python/test_jsonrpc.py @@ -13,7 +13,7 @@ import pytest -from copilot.jsonrpc import JsonRpcClient +from copilot._jsonrpc import JsonRpcClient class MockProcess: diff --git a/python/test_telemetry.py b/python/test_telemetry.py index 2b4649011..aec38f816 100644 --- a/python/test_telemetry.py +++ b/python/test_telemetry.py @@ -4,7 +4,7 @@ from unittest.mock import patch -from copilot.telemetry import get_trace_context, trace_context +from copilot._telemetry import get_trace_context, trace_context from copilot.types import SubprocessConfig, TelemetryConfig diff --git a/scripts/codegen/python.ts b/scripts/codegen/python.ts index 65563d741..3c70fb575 100644 --- a/scripts/codegen/python.ts +++ b/scripts/codegen/python.ts @@ -10,11 +10,11 @@ import fs from "fs/promises"; import type { JSONSchema7 } from "json-schema"; import { FetchingJSONSchemaStore, InputData, JSONSchemaInput, quicktype } from "quicktype-core"; import { - getSessionEventsSchemaPath, getApiSchemaPath, + getSessionEventsSchemaPath, + isRpcMethod, postProcessSchema, writeGeneratedFile, - isRpcMethod, type ApiSchema, type RpcMethod, } from "./utils.js"; @@ -224,7 +224,7 @@ Generated from: api.schema.json from typing import TYPE_CHECKING if TYPE_CHECKING: - from ..jsonrpc import JsonRpcClient + from .._jsonrpc import JsonRpcClient `); lines.push(typesCode); From 079261605163f8164abd3902c1ff3462ffbaee87 Mon Sep 17 00:00:00 2001 From: Brett Cannon Date: Wed, 18 Mar 2026 13:35:03 -0700 Subject: [PATCH 2/2] Add back in trailing whitespace --- python/copilot/generated/rpc.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/copilot/generated/rpc.py b/python/copilot/generated/rpc.py index a33c34c05..aebae4e5f 100644 --- a/python/copilot/generated/rpc.py +++ b/python/copilot/generated/rpc.py @@ -510,9 +510,9 @@ def to_dict(self) -> dict: class Mode(Enum): """The current agent mode. - + The agent mode after switching. - + The mode to switch to. Valid values: "interactive", "plan", "autopilot". """ AUTOPILOT = "autopilot"