@github/copilot-sdk 1.0.2 → 1.0.4

package/README.md

@@ -618,14 +618,17 @@ const session = await client.createSession({
 });
 ```
 
-Available section IDs: `identity`, `tone`, `tool_efficiency`, `environment_context`, `code_change_rules`, `guidelines`, `safety`, `tool_instructions`, `custom_instructions`, `runtime_instructions`, `last_instructions`. Use the `SYSTEM_MESSAGE_SECTIONS` constant for descriptions of each section.
+Available section IDs: `preamble`, `identity`, `tone`, `tool_efficiency`, `environment_context`, `code_change_rules`, `guidelines`, `safety`, `tool_instructions`, `custom_instructions`, `runtime_instructions`, `last_instructions`. Use the `SYSTEM_MESSAGE_SECTIONS` constant for descriptions of each section.
 
-Each section override supports four actions:
+`identity` and `tool_instructions` are section _groups_ that target a collection of related sub-sections as a unit. Use `preamble` to target just the identity preamble without affecting its sibling sub-sections.
+
+Each section override supports five actions:
 
 - **`replace`** — Replace the section content entirely
 - **`remove`** — Remove the section from the prompt
 - **`append`** — Add content after the existing section
 - **`prepend`** — Add content before the existing section
+- **`preserve`** — No-op that opts an individually-addressable section out of a group-level `remove`
 
 Unknown section IDs are handled gracefully: content from `replace`/`append`/`prepend` overrides is appended to additional instructions, and `remove` overrides are silently ignored.
 
@@ -684,6 +687,7 @@ information across turns. Provide a `memory` configuration on session create or
 when omitted, the runtime default applies. In the default `"copilot-cli"` client mode the
 SDK leaves `memory` unset so the runtime applies its own default, while `"empty"` mode
 defaults `memory` to disabled unless you set it explicitly.
+For more background, see [About GitHub Copilot Memory](https://docs.github.com/en/copilot/concepts/agents/copilot-memory).
 
 ```typescript
 // Enable memory for a session

package/dist/cjs/client.js

@@ -33,6 +33,7 @@ var import_rpc = require("./generated/rpc.js");
 var import_sdkProtocolVersion = require("./sdkProtocolVersion.js");
 var import_session = require("./session.js");
 var import_sessionFsProvider = require("./sessionFsProvider.js");
+var import_copilotRequestHandler = require("./copilotRequestHandler.js");
 var import_telemetry = require("./telemetry.js");
 var import_toolSet = require("./toolSet.js");
 var import_types = require("./types.js");
@@ -93,6 +94,32 @@ function toJsonSchema(parameters) {
   }
   return parameters;
 }
+const DEFAULT_PROVIDER_NAME = "default";
+function extractBearerTokenProviders(provider, providers) {
+  const callbacks = /* @__PURE__ */ new Map();
+  let wireProvider = provider;
+  if (provider?.getBearerToken) {
+    const { getBearerToken, ...rest } = provider;
+    callbacks.set(DEFAULT_PROVIDER_NAME, getBearerToken);
+    wireProvider = {
+      ...rest,
+      hasBearerTokenProvider: true
+    };
+  }
+  let wireProviders = providers;
+  if (providers?.some((p) => p.getBearerToken)) {
+    wireProviders = providers.map((p) => {
+      if (!p.getBearerToken) return p;
+      const { getBearerToken, ...rest } = p;
+      callbacks.set(p.name, getBearerToken);
+      return {
+        ...rest,
+        hasBearerTokenProvider: true
+      };
+    });
+  }
+  return { wireProvider, wireProviders, callbacks };
+}
 function toWireMcpServers(mcpServers) {
   if (!mcpServers) return void 0;
   return Object.fromEntries(
@@ -170,28 +197,57 @@ function getNodeExecPath() {
   }
   return process.execPath;
 }
+function getCliPlatformPackageNames() {
+  const arch = process.arch;
+  const variants = process.platform === "linux" ? ["linux", "linuxmusl"] : [process.platform];
+  return variants.map((variant) => `@github/copilot-${variant}-${arch}`);
+}
 function getBundledCliPath() {
+  const packageNames = getCliPlatformPackageNames();
   if (typeof import_meta.resolve === "function") {
-    const sdkUrl = import_meta.resolve("@github/copilot/sdk");
-    const sdkPath = (0, import_node_url.fileURLToPath)(sdkUrl);
-    return (0, import_node_path.join)((0, import_node_path.dirname)((0, import_node_path.dirname)(sdkPath)), "index.js");
+    for (const packageName of packageNames) {
+      try {
+        const sdkUrl = import_meta.resolve(`${packageName}/sdk`);
+        const sdkPath = (0, import_node_url.fileURLToPath)(sdkUrl);
+        return (0, import_node_path.join)((0, import_node_path.dirname)((0, import_node_path.dirname)(sdkPath)), "index.js");
+      } catch {
+      }
+    }
+    throw new Error(
+      `Could not resolve a @github/copilot platform package (tried ${packageNames.join(", ")}). Ensure @github/copilot is installed, or pass cliPath/cliUrl to CopilotClient.`
+    );
   }
   const req = (0, import_node_module.createRequire)(__filename);
   const searchPaths = req.resolve.paths("@github/copilot") ?? [];
   for (const base of searchPaths) {
-    const candidate = (0, import_node_path.join)(base, "@github", "copilot", "index.js");
-    if ((0, import_node_fs.existsSync)(candidate)) {
-      return candidate;
+    for (const packageName of packageNames) {
+      const candidate = (0, import_node_path.join)(base, ...packageName.split("/"), "index.js");
+      if ((0, import_node_fs.existsSync)(candidate)) {
+        return candidate;
+      }
     }
   }
   throw new Error(
-    `Could not find @github/copilot package. Searched ${searchPaths.length} paths. Ensure it is installed, or pass cliPath/cliUrl to CopilotClient.`
+    `Could not find a @github/copilot platform package (tried ${packageNames.join(", ")}). Searched ${searchPaths.length} paths. Ensure @github/copilot is installed, or pass cliPath/cliUrl to CopilotClient.`
   );
 }
+class TeardownResilientStreamMessageWriter extends import_node.StreamMessageWriter {
+  suppressWriteErrors = false;
+  async write(msg) {
+    try {
+      await super.write(msg);
+    } catch (error) {
+      if (!this.suppressWriteErrors) {
+        throw error;
+      }
+    }
+  }
+}
 class CopilotClient {
   cliStartTimeout = null;
   cliProcess = null;
   connection = null;
+  messageWriter = null;
   socket = null;
   runtimePort = null;
   actualHost = "localhost";
@@ -223,6 +279,8 @@ class CopilotClient {
   negotiatedProtocolVersion = null;
   /** Connection-level session filesystem config, set via constructor option. */
   sessionFsConfig = null;
+  requestHandler = null;
+  llmInferenceHandlers = {};
   /**
    * Typed server-scoped RPC methods.
    * @throws Error if the client is not connected
@@ -253,6 +311,13 @@ class CopilotClient {
     const level = this.options.logLevel?.toLowerCase();
     if (level === "debug" || level === "all") {
       process.stderr.write(`[copilot-sdk] ${message}. Elapsed=${Date.now() - startMs}ms
+`);
+    }
+  }
+  logDebug(message) {
+    const level = this.options.logLevel?.toLowerCase();
+    if (level === "debug" || level === "all") {
+      process.stderr.write(`[copilot-sdk] ${message}
 `);
     }
   }
@@ -315,6 +380,8 @@ class CopilotClient {
     this.onListModels = options.onListModels;
     this.onGetTraceContext = options.onGetTraceContext;
     this.sessionFsConfig = options.sessionFs ?? null;
+    this.requestHandler = options.requestHandler ?? null;
+    this.setupLlmInference();
     const effectiveEnv = options.env ?? process.env;
     this.resolvedEnv = effectiveEnv;
     this.resolvedCliPath = conn.kind === "stdio" || conn.kind === "tcp" ? conn.path ?? effectiveEnv.COPILOT_CLI_PATH ?? getBundledCliPath() : void 0;
@@ -394,6 +461,20 @@ class CopilotClient {
     }
     session.clientSessionApis.sessionFs = (0, import_sessionFsProvider.createSessionFsAdapter)(provider);
   }
+  setupLlmInference() {
+    if (!this.requestHandler) {
+      return;
+    }
+    this.llmInferenceHandlers = {
+      llmInference: (0, import_copilotRequestHandler.createCopilotRequestAdapter)(this.requestHandler, () => {
+        if (!this.connection) {
+          return void 0;
+        }
+        this._rpc ??= (0, import_rpc.createServerRpc)(this.connection);
+        return this._rpc;
+      })
+    };
+  }
   /**
    * Starts the CLI server and establishes a connection.
    *
@@ -431,6 +512,9 @@ class CopilotClient {
           capabilities: this.sessionFsConfig.capabilities
         });
       }
+      if (this.requestHandler) {
+        await this.connection.sendRequest("llmInference.setProvider", {});
+      }
       this.state = "connected";
     } catch (error) {
       this.state = "error";
@@ -463,7 +547,8 @@ class CopilotClient {
    */
   async stop() {
     const errors = [];
-    for (const session of this.sessions.values()) {
+    const activeSessions = [...this.sessions.values()];
+    for (const session of activeSessions) {
       const sessionId = session.sessionId;
       let lastError = null;
       for (let attempt = 1; attempt <= 3; attempt++) {
@@ -487,8 +572,10 @@ class CopilotClient {
         );
       }
     }
+    for (const session of activeSessions) {
+      session._markDisconnected();
+    }
     this.sessions.clear();
-    let runtimeShutdownCompleted = false;
     if (this.connection && this.cliProcess && !this.isExternalServer) {
       const runtimeShutdownStart = Date.now();
       const shutdownPromise = this.rpc.runtime.shutdown();
@@ -499,7 +586,6 @@ class CopilotClient {
           RUNTIME_SHUTDOWN_TIMEOUT_MS,
           `runtime.shutdown timed out after ${RUNTIME_SHUTDOWN_TIMEOUT_MS}ms`
         );
-        runtimeShutdownCompleted = true;
         this.logDebugTiming(
           "CopilotClient.stop runtime shutdown complete",
           runtimeShutdownStart
@@ -516,6 +602,9 @@ class CopilotClient {
         );
       }
     }
+    if (this.messageWriter) {
+      this.messageWriter.suppressWriteErrors = true;
+    }
     if (this.connection) {
       try {
         this.connection.dispose();
@@ -527,6 +616,7 @@ class CopilotClient {
         );
       }
       this.connection = null;
+      this.messageWriter = null;
       this._rpc = null;
       this._internalRpc = null;
     }
@@ -554,16 +644,13 @@ class CopilotClient {
       this.cliProcess = null;
       try {
         if (child.exitCode == null && child.signalCode == null) {
-          const exitedGracefully = runtimeShutdownCompleted ? await waitForChildExit(child, RUNTIME_SHUTDOWN_TIMEOUT_MS) : false;
-          if (!exitedGracefully) {
-            child.kill();
-            if (!await waitForChildExit(child, RUNTIME_SHUTDOWN_TIMEOUT_MS)) {
-              errors.push(
-                new Error(
-                  `Timed out waiting for CLI process to exit after kill: ${RUNTIME_SHUTDOWN_TIMEOUT_MS}ms`
-                )
-              );
-            }
+          child.kill();
+          if (!await waitForChildExit(child, RUNTIME_SHUTDOWN_TIMEOUT_MS)) {
+            errors.push(
+              new Error(
+                `Timed out waiting for CLI process to exit after kill: ${RUNTIME_SHUTDOWN_TIMEOUT_MS}ms`
+              )
+            );
           }
         }
       } catch (error) {
@@ -626,13 +713,20 @@ class CopilotClient {
    */
   async forceStop() {
     this.forceStopping = true;
+    for (const session of this.sessions.values()) {
+      session._markDisconnected();
+    }
     this.sessions.clear();
+    if (this.messageWriter) {
+      this.messageWriter.suppressWriteErrors = true;
+    }
     if (this.connection) {
       try {
         this.connection.dispose();
       } catch {
       }
       this.connection = null;
+      this.messageWriter = null;
       this._rpc = null;
       this._internalRpc = null;
     }
@@ -818,6 +912,11 @@ class CopilotClient {
     const callerSessionId = config.sessionId;
     const useServerGeneratedId = config.cloud != null && callerSessionId == null;
     const localSessionId = useServerGeneratedId ? void 0 : callerSessionId ?? (0, import_node_crypto.randomUUID)();
+    const {
+      wireProvider: bearerWireProvider,
+      wireProviders: bearerWireProviders,
+      callbacks: bearerTokenCallbacks
+    } = extractBearerTokenProviders(config.provider, config.providers);
     const { wirePayload: wireSystemMessage, transformCallbacks } = extractTransformCallbacks(
       config.systemMessage
     );
@@ -831,6 +930,9 @@ class CopilotClient {
       s.registerTools(config.tools);
       s.registerCanvases(config.canvases);
       s.registerCommands(config.commands);
+      if (bearerTokenCallbacks.size > 0) {
+        s.registerBearerTokenProviders(bearerTokenCallbacks);
+      }
       s.registerPermissionHandler(config.onPermissionRequest);
       if (config.onUserInputRequest) {
         s.registerUserInputHandler(config.onUserInputRequest);
@@ -894,7 +996,10 @@ class CopilotClient {
         availableTools: toolFilterOptions.availableTools,
         excludedTools: toolFilterOptions.excludedTools,
         toolFilterPrecedence: toolFilterOptions.toolFilterPrecedence,
-        provider: config.provider,
+        provider: bearerWireProvider,
+        capi: config.capi,
+        providers: bearerWireProviders,
+        models: config.models,
         enableSessionTelemetry: config.enableSessionTelemetry,
         modelCapabilities: config.modelCapabilities,
         largeOutput: toWireLargeOutput(config.largeOutput),
@@ -932,7 +1037,8 @@ class CopilotClient {
         memory: config.memory,
         gitHubToken: config.gitHubToken,
         remoteSession: config.remoteSession,
-        cloud: config.cloud
+        cloud: config.cloud,
+        expAssignments: config.expAssignments
       });
       const {
         sessionId: returnedSessionId,
@@ -999,6 +1105,14 @@ class CopilotClient {
     session.registerTools(config.tools);
     session.registerCanvases(config.canvases);
     session.registerCommands(config.commands);
+    const {
+      wireProvider: bearerWireProvider,
+      wireProviders: bearerWireProviders,
+      callbacks: bearerTokenCallbacks
+    } = extractBearerTokenProviders(config.provider, config.providers);
+    if (bearerTokenCallbacks.size > 0) {
+      session.registerBearerTokenProviders(bearerTokenCallbacks);
+    }
     session.registerPermissionHandler(config.onPermissionRequest);
     if (config.onUserInputRequest) {
       session.registerUserInputHandler(config.onUserInputRequest);
@@ -1060,7 +1174,10 @@ class CopilotClient {
           name: cmd.name,
           description: cmd.description
         })),
-        provider: config.provider,
+        provider: bearerWireProvider,
+        capi: config.capi,
+        providers: bearerWireProviders,
+        models: config.models,
         modelCapabilities: config.modelCapabilities,
         largeOutput: toWireLargeOutput(config.largeOutput),
         requestPermission: config.onPermissionRequest !== import_types.defaultJoinSessionPermissionHandler,
@@ -1099,7 +1216,8 @@ class CopilotClient {
         continuePendingWork: config.continuePendingWork,
         gitHubToken: config.gitHubToken,
         remoteSession: config.remoteSession,
-        openCanvases: config.openCanvases
+        openCanvases: config.openCanvases,
+        expAssignments: config.expAssignments
       });
       const { workspacePath, capabilities, openCanvases } = response;
       session["_workspacePath"] = workspacePath;
@@ -1641,13 +1759,21 @@ stderr: ${stderrOutput}`
       throw new Error("CLI process not started");
     }
     this.cliProcess.stdin?.on("error", (err) => {
-      if (!this.forceStopping) {
-        throw err;
+      if (this.forceStopping) {
+        return;
+      }
+      this.state = "error";
+      const reason = err instanceof Error ? err.stack ?? err.message : String(err);
+      this.logDebug(`stdin pipe error: ${reason}`);
+      try {
+        this.connection?.dispose();
+      } catch {
       }
     });
+    this.messageWriter = new TeardownResilientStreamMessageWriter(this.cliProcess.stdin);
     this.connection = (0, import_node.createMessageConnection)(
       new import_node.StreamMessageReader(this.cliProcess.stdout),
-      new import_node.StreamMessageWriter(this.cliProcess.stdin)
+      this.messageWriter
     );
     this.attachConnectionHandlers();
     this.connection.listen();
@@ -1659,9 +1785,10 @@ stderr: ${stderrOutput}`
     if (this.cliProcess) {
       throw new Error("CLI child process was unexpectedly started in parent process mode");
     }
+    this.messageWriter = new TeardownResilientStreamMessageWriter(process.stdout);
     this.connection = (0, import_node.createMessageConnection)(
       new import_node.StreamMessageReader(process.stdin),
-      new import_node.StreamMessageWriter(process.stdout)
+      this.messageWriter
     );
     this.attachConnectionHandlers();
     this.connection.listen();
@@ -1681,9 +1808,10 @@ stderr: ${stderrOutput}`
       }, 1e4);
       this.socket.connect(this.runtimePort, this.actualHost, () => {
         clearTimeout(connectionTimeout);
+        this.messageWriter = new TeardownResilientStreamMessageWriter(this.socket);
         this.connection = (0, import_node.createMessageConnection)(
           new import_node.StreamMessageReader(this.socket),
-          new import_node.StreamMessageWriter(this.socket)
+          this.messageWriter
         );
         this.attachConnectionHandlers();
         this.connection.listen();
@@ -1731,6 +1859,7 @@ stderr: ${stderrOutput}`
       if (!session) throw new Error(`No session found for sessionId: ${sessionId}`);
       return session.clientSessionApis;
     });
+    (0, import_rpc.registerClientGlobalApiHandlers)(this.connection, this.llmInferenceHandlers);
     this.connection.onClose(() => {
       this.state = "disconnected";
     });