feat(opencode): add OpenAI WebSocket transport with incremental follow-up handling

packages/opencode/src/flag/flag.ts

@@ -52,6 +52,9 @@ export namespace Flag {
   export const OPENCODE_DISABLE_FILETIME_CHECK = truthy("OPENCODE_DISABLE_FILETIME_CHECK")
   export const OPENCODE_EXPERIMENTAL_PLAN_MODE = OPENCODE_EXPERIMENTAL || truthy("OPENCODE_EXPERIMENTAL_PLAN_MODE")
   export const OPENCODE_EXPERIMENTAL_MARKDOWN = truthy("OPENCODE_EXPERIMENTAL_MARKDOWN")
+  export const OPENCODE_EXPERIMENTAL_WS_TRANSPORT = truthy("OPENCODE_EXPERIMENTAL_WS_TRANSPORT")
+  export const OPENCODE_EXPERIMENTAL_WS_INCREMENTAL =
+    OPENCODE_EXPERIMENTAL_WS_TRANSPORT || truthy("OPENCODE_EXPERIMENTAL_WS_INCREMENTAL")
   export const OPENCODE_MODELS_URL = process.env["OPENCODE_MODELS_URL"]
   export const OPENCODE_MODELS_PATH = process.env["OPENCODE_MODELS_PATH"]
 

packages/opencode/src/plugin/codex.ts

@@ -4,6 +4,8 @@ import { Installation } from "../installation"
 import { Auth, OAUTH_DUMMY_KEY } from "../auth"
 import os from "os"
 import { ProviderTransform } from "@/provider/transform"
+import { Flag } from "@/flag/flag"
+import { OpenAIWebSocket } from "@/provider/transport/openai-websocket"
 
 const log = Log.create({ service: "plugin.codex" })
 
@@ -487,6 +489,17 @@ export async function CodexAuthPlugin(input: PluginInput): Promise<Hooks> {
                 ? new URL(CODEX_API_ENDPOINT)
                 : parsed
 
+            if (Flag.OPENCODE_EXPERIMENTAL_WS_TRANSPORT) {
+              const response = await OpenAIWebSocket.stream({
+                request: url,
+                init: {
+                  ...init,
+                  headers,
+                },
+              })
+              if (response) return response
+            }
+
             return fetch(url, {
               ...init,
               headers,

packages/opencode/src/provider/provider.ts

@@ -44,6 +44,7 @@ import { fromNodeProviderChain } from "@aws-sdk/credential-providers"
 import { GoogleAuth } from "google-auth-library"
 import { ProviderTransform } from "./transform"
 import { Installation } from "../installation"
+import { OpenAIWebSocket } from "./transport/openai-websocket"
 
 export namespace Provider {
   const log = Log.create({ service: "provider" })
@@ -1103,6 +1104,14 @@ export namespace Provider {
           }
         }
 
+        if (Flag.OPENCODE_EXPERIMENTAL_WS_TRANSPORT && model.providerID === "openai" && !customFetch) {
+          const response = await OpenAIWebSocket.stream({
+            request: input,
+            init: opts,
+          })
+          if (response) return response
+        }
+
         return fetchFn(input, {
           ...opts,
           // @ts-ignore see here: https://github.com/oven-sh/bun/issues/16682
@@ -1147,6 +1156,10 @@ export namespace Provider {
     return state().then((s) => s.providers[providerID])
   }
 
+  export function closeSessionTransport(sessionID: string) {
+    OpenAIWebSocket.closeSession(sessionID)
+  }
+
   export async function getModel(providerID: string, modelID: string) {
     const s = await state()
     const provider = s.providers[providerID]