rovecode 0.3.2
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +662 -0
- package/README.md +737 -0
- package/THIRD_PARTY_NOTICES.md +268 -0
- package/bin/rovecode.ts +21 -0
- package/package.json +56 -0
- package/src/acp/server.ts +374 -0
- package/src/cli/auth-login.ts +122 -0
- package/src/cli/connect.ts +244 -0
- package/src/cli/context-cmd.ts +199 -0
- package/src/cli/dispatch.ts +82 -0
- package/src/cli/doctor.ts +362 -0
- package/src/cli/export.ts +276 -0
- package/src/cli/help.ts +293 -0
- package/src/cli/is-tui-invocation.ts +8 -0
- package/src/cli/main.ts +583 -0
- package/src/cli/market-cmd.ts +658 -0
- package/src/cli/mcp-login.ts +141 -0
- package/src/cli/mcp-market-cmd.ts +302 -0
- package/src/cli/output.ts +382 -0
- package/src/cli/repl.ts +250 -0
- package/src/cli/repomap-root.ts +14 -0
- package/src/cli/resume.ts +57 -0
- package/src/cli/run-flags.ts +43 -0
- package/src/cli/run-limits.ts +78 -0
- package/src/cli/runtime.ts +931 -0
- package/src/cli/session-arg.ts +30 -0
- package/src/cli/sessions-cmd.ts +145 -0
- package/src/cli/setup.ts +153 -0
- package/src/cli/skills-cmd.ts +194 -0
- package/src/cli/start-chat.ts +65 -0
- package/src/cli/trust-cmd.ts +52 -0
- package/src/coding/bash.ts +148 -0
- package/src/coding/checkpoints.ts +327 -0
- package/src/coding/diff.ts +138 -0
- package/src/coding/files.ts +341 -0
- package/src/coding/hashline.ts +274 -0
- package/src/coding/lsp-gate.ts +254 -0
- package/src/coding/lsp-servers.ts +147 -0
- package/src/coding/lsp.ts +283 -0
- package/src/coding/repomap-cache.ts +99 -0
- package/src/coding/repomap-files.ts +192 -0
- package/src/coding/repomap.ts +481 -0
- package/src/core/agents.ts +255 -0
- package/src/core/compaction.ts +259 -0
- package/src/core/config.ts +289 -0
- package/src/core/context-report.ts +228 -0
- package/src/core/context.ts +60 -0
- package/src/core/count-remote.ts +107 -0
- package/src/core/execpolicy-rules.ts +196 -0
- package/src/core/execpolicy.ts +385 -0
- package/src/core/executor.ts +454 -0
- package/src/core/guardrails.ts +400 -0
- package/src/core/hooks.ts +411 -0
- package/src/core/images.ts +230 -0
- package/src/core/intro.ts +266 -0
- package/src/core/loop.ts +567 -0
- package/src/core/modes.ts +372 -0
- package/src/core/orchestrator.ts +245 -0
- package/src/core/proc-group.ts +48 -0
- package/src/core/project-trust.ts +98 -0
- package/src/core/reflection.ts +165 -0
- package/src/core/sandbox-config.ts +186 -0
- package/src/core/session-id.ts +24 -0
- package/src/core/session-images.ts +73 -0
- package/src/core/session-ops.ts +183 -0
- package/src/core/session-text.ts +29 -0
- package/src/core/session.ts +469 -0
- package/src/core/settings.ts +170 -0
- package/src/core/tasks.ts +646 -0
- package/src/core/token-scale.ts +108 -0
- package/src/core/tools.ts +309 -0
- package/src/core/trust.ts +104 -0
- package/src/core/types.ts +330 -0
- package/src/core/update-check.ts +171 -0
- package/src/core/usage.ts +204 -0
- package/src/core/validate.ts +121 -0
- package/src/core/verify-gate.ts +159 -0
- package/src/core/verify.ts +236 -0
- package/src/core/voice.ts +158 -0
- package/src/core/win-job.ts +183 -0
- package/src/core/workspace.ts +184 -0
- package/src/design/audit.ts +797 -0
- package/src/design/direction.ts +190 -0
- package/src/design/rules.ts +157 -0
- package/src/eval/bench.ts +150 -0
- package/src/eval/gauntlet-runner.ts +215 -0
- package/src/eval/gauntlet-support.ts +84 -0
- package/src/eval/gauntlet-wave3.ts +269 -0
- package/src/eval/gauntlet-wave4.ts +217 -0
- package/src/eval/gauntlet.ts +253 -0
- package/src/index.ts +17 -0
- package/src/lanes/agy.ts +95 -0
- package/src/lanes/approval.ts +24 -0
- package/src/lanes/claude.ts +129 -0
- package/src/lanes/codex.ts +127 -0
- package/src/lanes/events.ts +130 -0
- package/src/lanes/job.ts +142 -0
- package/src/lanes/opencode.ts +122 -0
- package/src/lanes/process.ts +184 -0
- package/src/lanes/progress.ts +183 -0
- package/src/lanes/registry.ts +178 -0
- package/src/lanes/runner.ts +124 -0
- package/src/lanes/types.ts +112 -0
- package/src/market/catalogs/mcp-docs.json +111 -0
- package/src/market/catalogs/plugins.json +111 -0
- package/src/market/catalogs/skills.json +478 -0
- package/src/market/clone.ts +72 -0
- package/src/market/context-cost.ts +121 -0
- package/src/market/digest.ts +106 -0
- package/src/market/index.ts +22 -0
- package/src/market/install.ts +578 -0
- package/src/market/manifest.ts +187 -0
- package/src/market/prereq.ts +145 -0
- package/src/market/registry.ts +363 -0
- package/src/market/resolve.ts +111 -0
- package/src/market/types.ts +236 -0
- package/src/market/validate.ts +227 -0
- package/src/mcp/client.ts +449 -0
- package/src/mcp/config.ts +252 -0
- package/src/mcp/local-package.ts +211 -0
- package/src/mcp/market-catalog.ts +84 -0
- package/src/mcp/market-install.ts +289 -0
- package/src/mcp/market.ts +362 -0
- package/src/mcp/oauth.ts +251 -0
- package/src/mcp/prompts-resources.ts +249 -0
- package/src/mcp/shared.ts +149 -0
- package/src/mcp/status.ts +67 -0
- package/src/mcp/tools.ts +275 -0
- package/src/mcp/transport.ts +122 -0
- package/src/mcp/trust.ts +25 -0
- package/src/memory/blocks.ts +278 -0
- package/src/memory/recall.ts +355 -0
- package/src/memory/scope.ts +182 -0
- package/src/memory/store.ts +105 -0
- package/src/memory/tools.ts +99 -0
- package/src/plugins/cli.ts +119 -0
- package/src/plugins/discover.ts +108 -0
- package/src/plugins/index.ts +50 -0
- package/src/plugins/install.ts +184 -0
- package/src/plugins/load.ts +124 -0
- package/src/plugins/manifest.ts +92 -0
- package/src/plugins/state.ts +83 -0
- package/src/providers/auth.ts +408 -0
- package/src/providers/cache.ts +223 -0
- package/src/providers/catalog-local.ts +160 -0
- package/src/providers/catalog.ts +421 -0
- package/src/providers/middleware-context.ts +86 -0
- package/src/providers/middleware.ts +373 -0
- package/src/providers/model-list.ts +23 -0
- package/src/providers/models-index.json +1 -0
- package/src/providers/oauth/common.ts +105 -0
- package/src/providers/oauth/device-code.ts +107 -0
- package/src/providers/oauth/github-copilot.ts +146 -0
- package/src/providers/oauth/loopback.ts +158 -0
- package/src/providers/oauth/openai.ts +163 -0
- package/src/providers/oauth/openrouter.ts +89 -0
- package/src/providers/oauth/pkce.ts +45 -0
- package/src/providers/oauth/registry.ts +39 -0
- package/src/providers/oauth/seam.ts +89 -0
- package/src/providers/profile-glm53.ts +111 -0
- package/src/providers/profile-sonnet5-persona.ts +65 -0
- package/src/providers/profile-sonnet5-voice.ts +23 -0
- package/src/providers/profiles.ts +156 -0
- package/src/providers/provider-config.ts +311 -0
- package/src/providers/registry.ts +333 -0
- package/src/providers/responses.ts +209 -0
- package/src/providers/retry.ts +234 -0
- package/src/providers/router.ts +294 -0
- package/src/providers/sse.ts +26 -0
- package/src/providers/stream-errors.ts +117 -0
- package/src/providers/stream.ts +566 -0
- package/src/providers/thinking.ts +189 -0
- package/src/providers/wire-messages.ts +129 -0
- package/src/providers/wire-responses.ts +79 -0
- package/src/providers/wire-select.ts +53 -0
- package/src/server/http.ts +291 -0
- package/src/server/openapi.ts +246 -0
- package/src/sextant/card-hits.ts +102 -0
- package/src/sextant/card-keys.ts +55 -0
- package/src/sextant/context-source.ts +157 -0
- package/src/sextant/crew-cards.ts +350 -0
- package/src/sextant/draw-agents.ts +273 -0
- package/src/sextant/draw-code.ts +388 -0
- package/src/sextant/draw-context.ts +222 -0
- package/src/sextant/draw-frame.ts +164 -0
- package/src/sextant/draw-market.ts +573 -0
- package/src/sextant/draw-messages.ts +386 -0
- package/src/sextant/draw-pet.ts +230 -0
- package/src/sextant/draw-plan.ts +187 -0
- package/src/sextant/draw-tabs.ts +85 -0
- package/src/sextant/draw-util.ts +65 -0
- package/src/sextant/draw-wizard.ts +378 -0
- package/src/sextant/engine.ts +230 -0
- package/src/sextant/frame-hits.ts +25 -0
- package/src/sextant/frame.ts +101 -0
- package/src/sextant/git-status.ts +197 -0
- package/src/sextant/grid.ts +59 -0
- package/src/sextant/input.ts +119 -0
- package/src/sextant/keys.ts +521 -0
- package/src/sextant/layout.ts +86 -0
- package/src/sextant/local-commands.ts +169 -0
- package/src/sextant/market-source.ts +287 -0
- package/src/sextant/mentions.ts +200 -0
- package/src/sextant/message-hits.ts +26 -0
- package/src/sextant/model.ts +387 -0
- package/src/sextant/overlays.ts +456 -0
- package/src/sextant/panel-hits.ts +38 -0
- package/src/sextant/pet.ts +399 -0
- package/src/sextant/screen.ts +324 -0
- package/src/sextant/scroll-hits.ts +66 -0
- package/src/sextant/scrollbar.ts +82 -0
- package/src/sextant/sextant-bridge.ts +174 -0
- package/src/sextant/sextant-cards.ts +142 -0
- package/src/sextant/sextant-diff-base.ts +63 -0
- package/src/sextant/sextant-files.ts +154 -0
- package/src/sextant/sextant-frame-loop.ts +335 -0
- package/src/sextant/sextant-renderer.ts +574 -0
- package/src/sextant/sextant-repo.ts +140 -0
- package/src/sextant/theme.ts +66 -0
- package/src/sextant/tool-rows.ts +189 -0
- package/src/sextant/types.ts +493 -0
- package/src/skills/index.ts +387 -0
- package/src/skills/pack.ts +220 -0
- package/src/skills/spec.ts +162 -0
- package/src/skills/tools.ts +69 -0
- package/src/skills/versioned.ts +227 -0
- package/src/telemetry/otel-export.ts +122 -0
- package/src/telemetry/otel-lanes.ts +89 -0
- package/src/telemetry/otel-logs.ts +131 -0
- package/src/telemetry/otel-metrics.ts +136 -0
- package/src/telemetry/otel.ts +397 -0
- package/src/telemetry/otlp.ts +76 -0
- package/src/tools/ask-user.ts +156 -0
- package/src/tools/bash-bg.ts +94 -0
- package/src/tools/bash-jobs.ts +237 -0
- package/src/tools/design.ts +151 -0
- package/src/tools/evalcell.ts +338 -0
- package/src/tools/html-text.ts +139 -0
- package/src/tools/provider.ts +149 -0
- package/src/tools/task.ts +250 -0
- package/src/tools/todo.ts +320 -0
- package/src/tools/webfetch.ts +332 -0
- package/src/tools/websearch.ts +359 -0
- package/src/tui/agents-cmd.ts +41 -0
- package/src/tui/app.ts +749 -0
- package/src/tui/attach.ts +127 -0
- package/src/tui/boot-notes.ts +41 -0
- package/src/tui/builtin-prompts.ts +59 -0
- package/src/tui/checkpoints-cmd.ts +70 -0
- package/src/tui/clipboard-image.ts +81 -0
- package/src/tui/clipboard.ts +78 -0
- package/src/tui/commands.ts +283 -0
- package/src/tui/config-view.ts +53 -0
- package/src/tui/context-cmds.ts +282 -0
- package/src/tui/cost.ts +108 -0
- package/src/tui/crash-guard.ts +173 -0
- package/src/tui/focus-terminal.ts +34 -0
- package/src/tui/git-cmds.ts +273 -0
- package/src/tui/git-plain.ts +58 -0
- package/src/tui/info-cmd.ts +150 -0
- package/src/tui/input-plain.ts +76 -0
- package/src/tui/mcp-cmd.ts +128 -0
- package/src/tui/memory-note.ts +77 -0
- package/src/tui/modes-cmd.ts +45 -0
- package/src/tui/notify-seq.ts +100 -0
- package/src/tui/notify.ts +318 -0
- package/src/tui/overlays.ts +97 -0
- package/src/tui/pi-renderer.ts +428 -0
- package/src/tui/providers-cmd.ts +377 -0
- package/src/tui/reasoning-view.ts +56 -0
- package/src/tui/renderer.ts +128 -0
- package/src/tui/replay-marker.ts +29 -0
- package/src/tui/session-cmd.ts +148 -0
- package/src/tui/session-manage.ts +95 -0
- package/src/tui/sextant-attach.ts +102 -0
- package/src/tui/sextant-io.ts +202 -0
- package/src/tui/sextant-smoke.ts +110 -0
- package/src/tui/shell-cmd.ts +158 -0
- package/src/tui/smoke.ts +72 -0
- package/src/tui/staged-terminal.ts +50 -0
- package/src/tui/startup.ts +12 -0
- package/src/tui/theme.ts +59 -0
- package/src/tui/todo-label.ts +7 -0
- package/src/tui/trust-card.ts +107 -0
- package/src/tui/tui-commands.ts +87 -0
- package/tsconfig.json +30 -0
- package/vendor/pi-tui/LICENSE +21 -0
- package/vendor/pi-tui/PATCHES.md +12 -0
- package/vendor/pi-tui/PROVENANCE.md +12 -0
- package/vendor/pi-tui/README.upstream.md +854 -0
- package/vendor/pi-tui/native/win32/prebuilds/win32-arm64/win32-console-mode.node +0 -0
- package/vendor/pi-tui/native/win32/prebuilds/win32-x64/win32-console-mode.node +0 -0
- package/vendor/pi-tui/src/alt-screen-search.ts +158 -0
- package/vendor/pi-tui/src/autocomplete.ts +827 -0
- package/vendor/pi-tui/src/components/alt-screen-flash.ts +52 -0
- package/vendor/pi-tui/src/components/box.ts +138 -0
- package/vendor/pi-tui/src/components/cancellable-loader.ts +41 -0
- package/vendor/pi-tui/src/components/editor.ts +2364 -0
- package/vendor/pi-tui/src/components/h-stack.ts +45 -0
- package/vendor/pi-tui/src/components/image.ts +128 -0
- package/vendor/pi-tui/src/components/input.ts +448 -0
- package/vendor/pi-tui/src/components/loader.ts +93 -0
- package/vendor/pi-tui/src/components/markdown.ts +1016 -0
- package/vendor/pi-tui/src/components/scroll-view.ts +217 -0
- package/vendor/pi-tui/src/components/select-list.ts +230 -0
- package/vendor/pi-tui/src/components/settings-list.ts +277 -0
- package/vendor/pi-tui/src/components/spacer.ts +29 -0
- package/vendor/pi-tui/src/components/stack.ts +155 -0
- package/vendor/pi-tui/src/components/text.ts +108 -0
- package/vendor/pi-tui/src/components/truncated-text.ts +66 -0
- package/vendor/pi-tui/src/components/v-stack.ts +34 -0
- package/vendor/pi-tui/src/editor-component.ts +75 -0
- package/vendor/pi-tui/src/fuzzy.ts +138 -0
- package/vendor/pi-tui/src/index.ts +149 -0
- package/vendor/pi-tui/src/keybindings.ts +321 -0
- package/vendor/pi-tui/src/keys.ts +1402 -0
- package/vendor/pi-tui/src/kill-ring.ts +47 -0
- package/vendor/pi-tui/src/latex.ts +1381 -0
- package/vendor/pi-tui/src/layout-node.ts +52 -0
- package/vendor/pi-tui/src/layout.ts +411 -0
- package/vendor/pi-tui/src/native-modifiers.ts +60 -0
- package/vendor/pi-tui/src/native-module-path.ts +32 -0
- package/vendor/pi-tui/src/stdin-buffer.ts +445 -0
- package/vendor/pi-tui/src/terminal-colors.ts +74 -0
- package/vendor/pi-tui/src/terminal-image.ts +701 -0
- package/vendor/pi-tui/src/terminal.ts +554 -0
- package/vendor/pi-tui/src/tui-alt-screen.ts +1379 -0
- package/vendor/pi-tui/src/tui-main-screen.ts +655 -0
- package/vendor/pi-tui/src/tui.ts +1264 -0
- package/vendor/pi-tui/src/undo-stack.ts +29 -0
- package/vendor/pi-tui/src/utils.ts +1327 -0
- package/vendor/pi-tui/src/word-navigation.ts +118 -0
- package/vendor/pi-tui/test/test-themes.ts +39 -0
- package/vendor/pi-tui/test/virtual-terminal.ts +219 -0
|
@@ -0,0 +1,333 @@
|
|
|
1
|
+
/** Live provider registry — the ADAPTER half of the provider layer, over provider-config.ts (data).
|
|
2
|
+
*
|
|
3
|
+
* One registry per runtime. stream() returns a DISPATCHING StreamFn: every call resolves
|
|
4
|
+
* `model.provider` against the live snapshot (hot reload — provider-config.ts), so a provider added
|
|
5
|
+
* from another terminal, by `rovecode provider add`, by `/provider add` in the TUI or by the agent's
|
|
6
|
+
* provider_edit tool serves the very next model call. No restart, no runtime rebuild. Concrete
|
|
7
|
+
* adapters (stream.ts) are cached per provider id and rebuilt when baseUrl, protocol, key or
|
|
8
|
+
* headers change — a rotated key takes effect on the next call too.
|
|
9
|
+
*
|
|
10
|
+
* Seam contract (ADR-003): the dispatcher never throws. An unknown provider or a missing key is ONE
|
|
11
|
+
* error turn whose text starts with `config: ` — router.ts classifyStreamError treats that prefix
|
|
12
|
+
* as non-retryable, so neither the same-model retry (port #23) nor a fallback chain (port #14)
|
|
13
|
+
* spins on a configuration mistake; the text tells the human exactly how to fix it.
|
|
14
|
+
*
|
|
15
|
+
* Because the dispatcher honours model.provider per call, cross-provider fallback chains
|
|
16
|
+
* (ROVECODE_MODEL_<ROLE>=a/x,b/y) now really route each candidate to its own endpoint.
|
|
17
|
+
*
|
|
18
|
+
* Secrets: this module reads keys to build adapters and to redact them from probe errors
|
|
19
|
+
* (scrubSecret). Nothing here prints or returns a key; formatProviderList shows key SOURCES. */
|
|
20
|
+
|
|
21
|
+
import type { AssistantTurn, Message, ModelRef, StreamEvent, StreamFn, StreamOptions } from "../core/types.ts";
|
|
22
|
+
import { fetchModels, providerStream, providerStreaming, wantsStreaming, type ProviderConfig as WireProviderConfig } from "./stream.ts";
|
|
23
|
+
import {
|
|
24
|
+
ProviderConfig, isConfigured, parseSelector, pickDefault, providersPathFor, readProvidersFile, validateSpec, writeProvidersFile,
|
|
25
|
+
type FileScope, type ProviderEntry, type ProviderSnapshot, type ProviderSpec, type ResolvedProvider,
|
|
26
|
+
} from "./provider-config.ts";
|
|
27
|
+
import { next } from "../core/voice.ts";
|
|
28
|
+
|
|
29
|
+
export type AdapterFactory = (p: ResolvedProvider, opts: { sse: boolean }) => StreamFn;
|
|
30
|
+
|
|
31
|
+
/** stream.ts ProviderConfig for a resolved provider (a keyless provider sends an empty bearer). */
|
|
32
|
+
export function toWireConfig(p: ResolvedProvider): WireProviderConfig {
|
|
33
|
+
return {
|
|
34
|
+
id: p.id, baseUrl: p.baseUrl, apiKey: p.apiKey ?? "", protocol: p.protocol,
|
|
35
|
+
...(p.defaultModel !== undefined ? { defaultModel: p.defaultModel } : {}),
|
|
36
|
+
...(p.headers !== undefined ? { headers: p.headers } : {}),
|
|
37
|
+
};
|
|
38
|
+
}
|
|
39
|
+
|
|
40
|
+
export const defaultAdapterFactory: AdapterFactory = (p, { sse }) => {
|
|
41
|
+
const wire = toWireConfig(p);
|
|
42
|
+
// both protocols stream (openai SSE chunks, anthropic Messages events); `sse` is off only when
|
|
43
|
+
// ROVECODE_STREAM asked for the one-shot JSON adapters
|
|
44
|
+
return sse ? providerStreaming(wire) : providerStream(wire);
|
|
45
|
+
};
|
|
46
|
+
|
|
47
|
+
/** Error-text prefix the router/retry classifier treats as non-retryable (router.ts). */
|
|
48
|
+
export const CONFIG_ERROR_PREFIX = "config: ";
|
|
49
|
+
|
|
50
|
+
export function configErrorTurn(text: string): AssistantTurn {
|
|
51
|
+
return { parts: [], stopReason: "error", usage: { input: 0, output: 0 }, error: CONFIG_ERROR_PREFIX + text };
|
|
52
|
+
}
|
|
53
|
+
|
|
54
|
+
/** Replace a secret wherever it leaked into text (some proxies echo request headers in error bodies). */
|
|
55
|
+
export function scrubSecret(text: string, secret: string | null | undefined): string {
|
|
56
|
+
return secret !== null && secret !== undefined && secret.length > 0 ? text.split(secret).join("…") : text;
|
|
57
|
+
}
|
|
58
|
+
|
|
59
|
+
export interface RegistryOptions {
|
|
60
|
+
env?: NodeJS.ProcessEnv;
|
|
61
|
+
/** test seam: build adapters without touching the network */
|
|
62
|
+
adapterFactory?: AdapterFactory;
|
|
63
|
+
/** provider-config.ts reload throttle (ms); tests pass 0 */
|
|
64
|
+
throttleMs?: number;
|
|
65
|
+
}
|
|
66
|
+
|
|
67
|
+
export class ProviderRegistry {
|
|
68
|
+
readonly cwd: string;
|
|
69
|
+
private readonly env: NodeJS.ProcessEnv;
|
|
70
|
+
private readonly cfg: ProviderConfig;
|
|
71
|
+
private readonly factory: AdapterFactory;
|
|
72
|
+
private readonly adapters = new Map<string, { sig: string; fn: StreamFn }>();
|
|
73
|
+
|
|
74
|
+
constructor(cwd: string, opts: RegistryOptions = {}) {
|
|
75
|
+
this.cwd = cwd;
|
|
76
|
+
this.env = opts.env ?? process.env;
|
|
77
|
+
this.factory = opts.adapterFactory ?? defaultAdapterFactory;
|
|
78
|
+
this.cfg = new ProviderConfig(cwd, this.env, opts.throttleMs);
|
|
79
|
+
}
|
|
80
|
+
|
|
81
|
+
snapshot(): ProviderSnapshot { return this.cfg.snapshot(); }
|
|
82
|
+
list(): ResolvedProvider[] { return this.snapshot().providers; }
|
|
83
|
+
ids(): string[] { return this.list().map((p) => p.id); }
|
|
84
|
+
get(id: string): ResolvedProvider | undefined { return this.list().find((p) => p.id === id); }
|
|
85
|
+
/** at least one provider has a key (or needs none) */
|
|
86
|
+
configured(): boolean { return this.list().some(isConfigured); }
|
|
87
|
+
warnings(): string[] { return this.snapshot().warnings; }
|
|
88
|
+
/** fires after every rebuild: external file edits (detected lazily) and in-process add/remove/setDefault */
|
|
89
|
+
onChange(cb: (snap: ProviderSnapshot) => void): () => void { return this.cfg.onChange(cb); }
|
|
90
|
+
/** re-read now (in-process writers call it right after writing) and notify listeners */
|
|
91
|
+
refresh(): ProviderSnapshot { this.cfg.invalidate(); return this.cfg.snapshot(); }
|
|
92
|
+
|
|
93
|
+
/** default provider + model with env ROVECODE_MODEL layered on top; null when nothing is configured */
|
|
94
|
+
defaultRef(): ModelRef | null {
|
|
95
|
+
const d = pickDefault(this.snapshot());
|
|
96
|
+
if (d === null) return null;
|
|
97
|
+
return { provider: d.provider.id, model: this.env["ROVECODE_MODEL"] ?? d.model ?? "" };
|
|
98
|
+
}
|
|
99
|
+
|
|
100
|
+
/** stream.ts-shaped config of the default provider (Runtime.provider compatibility) */
|
|
101
|
+
defaultConfig(): WireProviderConfig | null {
|
|
102
|
+
const d = pickDefault(this.snapshot());
|
|
103
|
+
if (d === null) return null;
|
|
104
|
+
return { ...toWireConfig(d.provider), ...(d.model !== undefined ? { defaultModel: d.model } : {}) };
|
|
105
|
+
}
|
|
106
|
+
|
|
107
|
+
defaultSelector(): string | undefined { return this.snapshot().defaultSelector; }
|
|
108
|
+
|
|
109
|
+
/** "provider/model" or a bare model on `currentProvider`. A leading segment counts as a provider
|
|
110
|
+
* only when it names one, so "zai-org/glm-5.3" stays a model id on the current provider. The
|
|
111
|
+
* current provider is not validated (it may be "mock" or an injected test stream): a bare model
|
|
112
|
+
* always lands there, exactly as `/model <id>` always did. */
|
|
113
|
+
resolveSelector(selector: string, currentProvider: string): ModelRef | { error: string } {
|
|
114
|
+
const s = selector.trim();
|
|
115
|
+
if (s.length === 0) return { error: "empty model selector" };
|
|
116
|
+
const i = s.indexOf("/");
|
|
117
|
+
if (i > 0) {
|
|
118
|
+
const head = s.slice(0, i);
|
|
119
|
+
const tail = s.slice(i + 1).trim();
|
|
120
|
+
if (this.get(head) !== undefined) return tail.length > 0 ? { provider: head, model: tail } : { error: `"${s}" names no model — use ${head}/<model>` };
|
|
121
|
+
}
|
|
122
|
+
return { provider: currentProvider, model: s };
|
|
123
|
+
}
|
|
124
|
+
|
|
125
|
+
add(spec: ProviderSpec, scope: FileScope = "user"): ResolvedProvider | { error: string } {
|
|
126
|
+
const { id, ...raw } = spec;
|
|
127
|
+
const v = validateSpec(id, raw);
|
|
128
|
+
if ("error" in v) return v;
|
|
129
|
+
if (this.get(id)?.scope === "env") return { error: `"custom" is reserved for the ROVECODE_BASE_URL/ROVECODE_API_KEY pair — pick another id` };
|
|
130
|
+
const path = providersPathFor(scope, this.cwd);
|
|
131
|
+
const { data } = readProvidersFile(path);
|
|
132
|
+
const { id: _id, ...rest } = v.spec;
|
|
133
|
+
data.providers = { ...(data.providers ?? {}), [id]: rest };
|
|
134
|
+
writeProvidersFile(path, data);
|
|
135
|
+
this.refresh();
|
|
136
|
+
return this.get(id)!;
|
|
137
|
+
}
|
|
138
|
+
|
|
139
|
+
remove(id: string): { removed: FileScope[] } | { error: string } {
|
|
140
|
+
const p = this.get(id);
|
|
141
|
+
if (p === undefined) return { error: `unknown provider "${id}" — known: ${this.ids().join(" ")}` };
|
|
142
|
+
if (p.scope === "builtin") return { error: `"${id}" is built in; drop its key with \`rovecode auth remove ${id}\` instead` };
|
|
143
|
+
if (p.scope === "env") return { error: `"custom" comes from ROVECODE_BASE_URL/ROVECODE_API_KEY — unset those env vars` };
|
|
144
|
+
const removed: FileScope[] = [];
|
|
145
|
+
for (const scope of ["user", "project"] as const) {
|
|
146
|
+
const path = providersPathFor(scope, this.cwd);
|
|
147
|
+
const { data } = readProvidersFile(path);
|
|
148
|
+
if (data.providers !== undefined && id in data.providers) {
|
|
149
|
+
delete data.providers[id];
|
|
150
|
+
if (Object.keys(data.providers).length === 0) delete data.providers;
|
|
151
|
+
writeProvidersFile(path, data);
|
|
152
|
+
removed.push(scope);
|
|
153
|
+
}
|
|
154
|
+
}
|
|
155
|
+
this.refresh();
|
|
156
|
+
return { removed };
|
|
157
|
+
}
|
|
158
|
+
|
|
159
|
+
setDefault(selector: string, scope: FileScope = "user"): ModelRef | { error: string } {
|
|
160
|
+
const sel = parseSelector(selector);
|
|
161
|
+
const p = this.get(sel.provider);
|
|
162
|
+
if (p === undefined) return { error: `unknown provider "${sel.provider}" — known: ${this.ids().join(" ")}` };
|
|
163
|
+
const model = sel.model ?? p.defaultModel;
|
|
164
|
+
if (model === undefined) return { error: `"${selector}" names no model and ${p.id} has no default — use ${p.id}/<model>` };
|
|
165
|
+
const path = providersPathFor(scope, this.cwd);
|
|
166
|
+
const { data } = readProvidersFile(path);
|
|
167
|
+
data.default = `${p.id}/${model}`;
|
|
168
|
+
writeProvidersFile(path, data);
|
|
169
|
+
this.refresh();
|
|
170
|
+
return { provider: p.id, model };
|
|
171
|
+
}
|
|
172
|
+
|
|
173
|
+
/** pin a provider's ACTIVE model list (the connect wizard's model step): writes `models` on the
|
|
174
|
+
* provider's row in the scope's file — creating the row for a builtin, which is how `add` works
|
|
175
|
+
* too. An EMPTY list removes the field: the provider goes back to asking its /models endpoint
|
|
176
|
+
* every time (the "use them all" answer — nothing frozen). Returns the written list. */
|
|
177
|
+
setModels(id: string, models: readonly string[], scope: FileScope = "user"): { models: string[] } | { error: string } {
|
|
178
|
+
const p = this.get(id);
|
|
179
|
+
if (p === undefined) return { error: `unknown provider "${id}" — known: ${this.ids().join(" ")}` };
|
|
180
|
+
if (p.scope === "env") return { error: `"custom" comes from ROVECODE_BASE_URL/ROVECODE_API_KEY — a model list cannot be pinned on it` };
|
|
181
|
+
for (const m of models) if (m.trim().length === 0) return { error: "model ids must be non-empty strings" };
|
|
182
|
+
const path = providersPathFor(scope, this.cwd);
|
|
183
|
+
const { data } = readProvidersFile(path);
|
|
184
|
+
const rows = data.providers ?? {};
|
|
185
|
+
const existing = rows[id];
|
|
186
|
+
// a builtin with no file row yet carries its baseUrl in the spec — the new row must too
|
|
187
|
+
const row: ProviderEntry = existing ?? { baseUrl: p.baseUrl };
|
|
188
|
+
const deduped = [...new Set(models.map((m) => m.trim()))];
|
|
189
|
+
if (deduped.length === 0) {
|
|
190
|
+
delete row.models;
|
|
191
|
+
} else {
|
|
192
|
+
row.models = deduped;
|
|
193
|
+
// a pinned list that does not contain the current default silently breaks the default —
|
|
194
|
+
// keep the field only when it still names a usable model
|
|
195
|
+
if (row.defaultModel !== undefined && !deduped.includes(row.defaultModel)) delete row.defaultModel;
|
|
196
|
+
}
|
|
197
|
+
rows[id] = row;
|
|
198
|
+
data.providers = rows;
|
|
199
|
+
writeProvidersFile(path, data);
|
|
200
|
+
this.refresh();
|
|
201
|
+
return { models: deduped };
|
|
202
|
+
}
|
|
203
|
+
|
|
204
|
+
keyHint(p: ResolvedProvider): string {
|
|
205
|
+
return `no API key for provider "${p.id}" — I can't call it without one. ${next(`rovecode auth set ${p.id}`)} (masked prompt) · or /provider key ${p.id} <key> in the TUI · or set ${p.keyEnv}`;
|
|
206
|
+
}
|
|
207
|
+
|
|
208
|
+
async models(id: string): Promise<{ ok: true; models: string[]; source: "file" | "endpoint" } | { ok: false; error: string }> {
|
|
209
|
+
const p = this.get(id);
|
|
210
|
+
if (p === undefined) return { ok: false, error: `unknown provider "${id}" — known: ${this.ids().join(" ")}` };
|
|
211
|
+
if (p.models !== undefined && p.models.length > 0) return { ok: true, models: p.models, source: "file" };
|
|
212
|
+
if (!isConfigured(p)) return { ok: false, error: this.keyHint(p) };
|
|
213
|
+
const list = await fetchModels(toWireConfig(p), true);
|
|
214
|
+
return { ok: true, models: list.map((m) => m.id), source: "endpoint" };
|
|
215
|
+
}
|
|
216
|
+
|
|
217
|
+
/** One tiny real call ("ping", ≤8 output tokens): proves url + key + model together. */
|
|
218
|
+
async probe(id: string, model?: string): Promise<{ ok: boolean; model: string; detail: string }> {
|
|
219
|
+
const p = this.get(id);
|
|
220
|
+
if (p === undefined) return { ok: false, model: model ?? "", detail: `unknown provider "${id}" — known: ${this.ids().join(" ")}` };
|
|
221
|
+
if (!isConfigured(p)) return { ok: false, model: model ?? "", detail: this.keyHint(p) };
|
|
222
|
+
const m = model ?? p.defaultModel;
|
|
223
|
+
if (m === undefined) return { ok: false, model: "", detail: `${p.id} has no default model — pass one: test ${p.id} <model>` };
|
|
224
|
+
const probeMsg: Message = { id: "probe", role: "user", parts: [{ kind: "text", text: "ping" }], parentId: null, createdAt: Date.now() };
|
|
225
|
+
const t0 = Date.now();
|
|
226
|
+
let turn: AssistantTurn | undefined;
|
|
227
|
+
try {
|
|
228
|
+
for await (const ev of this.adapterFor(p)({ provider: p.id, model: m, maxTokens: 8 }, [probeMsg], { tools: [] })) if (ev.type === "turn") turn = ev.turn;
|
|
229
|
+
} catch (e) {
|
|
230
|
+
return { ok: false, model: m, detail: scrubSecret(e instanceof Error ? e.message : String(e), p.apiKey) };
|
|
231
|
+
}
|
|
232
|
+
if (turn === undefined) return { ok: false, model: m, detail: "stream produced no turn" };
|
|
233
|
+
if (turn.stopReason === "error") return { ok: false, model: m, detail: scrubSecret(turn.error ?? "stream error", p.apiKey) };
|
|
234
|
+
return { ok: true, model: m, detail: `ok in ${Date.now() - t0} ms (${turn.usage.input} in / ${turn.usage.output} out tokens)` };
|
|
235
|
+
}
|
|
236
|
+
|
|
237
|
+
/** The dispatching StreamFn (see the header). */
|
|
238
|
+
stream(): StreamFn {
|
|
239
|
+
const self = this;
|
|
240
|
+
return async function* (model: ModelRef, messages: Message[], options?: StreamOptions): AsyncGenerator<StreamEvent> {
|
|
241
|
+
const p = self.get(model.provider);
|
|
242
|
+
if (p === undefined) {
|
|
243
|
+
yield { type: "turn", turn: configErrorTurn(`provider "${model.provider}" is not configured — I can't route to it. ${next(`rovecode provider add ${model.provider} <baseUrl> [--protocol openai|anthropic]`)} (or /provider add in the TUI); known: ${self.ids().join(" ")}`) };
|
|
244
|
+
return;
|
|
245
|
+
}
|
|
246
|
+
if (!isConfigured(p)) { yield { type: "turn", turn: configErrorTurn(self.keyHint(p)) }; return; }
|
|
247
|
+
yield* self.adapterFor(p)(model, messages, options);
|
|
248
|
+
};
|
|
249
|
+
}
|
|
250
|
+
|
|
251
|
+
private adapterFor(p: ResolvedProvider): StreamFn {
|
|
252
|
+
const sse = wantsStreaming({ ROVECODE_STREAM: this.env["ROVECODE_STREAM"] }); // default ON
|
|
253
|
+
const sig = [p.baseUrl, p.protocol, p.apiKey ?? "", JSON.stringify(p.headers ?? {}), sse ? "sse" : "json"].join("\u0000");
|
|
254
|
+
const hit = this.adapters.get(p.id);
|
|
255
|
+
if (hit !== undefined && hit.sig === sig) return hit.fn;
|
|
256
|
+
const fn = this.factory(p, { sse });
|
|
257
|
+
this.adapters.set(p.id, { sig, fn });
|
|
258
|
+
return fn;
|
|
259
|
+
}
|
|
260
|
+
}
|
|
261
|
+
|
|
262
|
+
// ---------- shared CLI / TUI helpers ----------
|
|
263
|
+
|
|
264
|
+
export const ADD_USAGE = "add <id> <baseUrl> [--protocol openai|anthropic] [--key-env NAME] [--model <id>] [--no-key] [--project]";
|
|
265
|
+
|
|
266
|
+
export interface AddArgs { spec: ProviderSpec; scope: FileScope; promptKey: boolean }
|
|
267
|
+
|
|
268
|
+
/** Parse `add` words (CLI argv tail or TUI slash args) into a validated spec + scope. `--key` (CLI:
|
|
269
|
+
* prompt for the secret afterwards) is recorded, never a value. Pure; never throws. */
|
|
270
|
+
export function parseAddArgs(words: readonly string[]): AddArgs | { error: string } {
|
|
271
|
+
const pos: string[] = [];
|
|
272
|
+
const raw: Record<string, unknown> = {};
|
|
273
|
+
let scope: FileScope = "user";
|
|
274
|
+
let promptKey = false;
|
|
275
|
+
for (let i = 0; i < words.length; i++) {
|
|
276
|
+
const w = words[i]!;
|
|
277
|
+
const value = (): string | { error: string } => {
|
|
278
|
+
const v = words[i + 1];
|
|
279
|
+
if (v === undefined || v.startsWith("-")) return { error: `${w} needs a value — ${ADD_USAGE}` };
|
|
280
|
+
i++;
|
|
281
|
+
return v;
|
|
282
|
+
};
|
|
283
|
+
switch (w) {
|
|
284
|
+
case "--protocol": { const v = value(); if (typeof v !== "string") return v; raw["protocol"] = v; break; }
|
|
285
|
+
case "--key-env": { const v = value(); if (typeof v !== "string") return v; raw["keyEnv"] = v; break; }
|
|
286
|
+
case "--model": { const v = value(); if (typeof v !== "string") return v; raw["defaultModel"] = v; break; }
|
|
287
|
+
case "--scope": {
|
|
288
|
+
const v = value(); if (typeof v !== "string") return v;
|
|
289
|
+
if (v !== "user" && v !== "project") return { error: "--scope must be user or project" };
|
|
290
|
+
scope = v; break;
|
|
291
|
+
}
|
|
292
|
+
case "--project": scope = "project"; break;
|
|
293
|
+
case "--user": scope = "user"; break;
|
|
294
|
+
case "--no-key": raw["noKey"] = true; break;
|
|
295
|
+
case "--key": promptKey = true; break;
|
|
296
|
+
default:
|
|
297
|
+
if (w.startsWith("-")) return { error: `unknown flag ${w} — ${ADD_USAGE}` };
|
|
298
|
+
pos.push(w);
|
|
299
|
+
}
|
|
300
|
+
}
|
|
301
|
+
const [id, baseUrl] = pos;
|
|
302
|
+
if (id === undefined || baseUrl === undefined) return { error: `usage: ${ADD_USAGE}` };
|
|
303
|
+
raw["baseUrl"] = baseUrl;
|
|
304
|
+
const v = validateSpec(id, raw);
|
|
305
|
+
if ("error" in v) return v;
|
|
306
|
+
return { spec: v.spec, scope, promptKey };
|
|
307
|
+
}
|
|
308
|
+
|
|
309
|
+
/** One row per provider — key SOURCE only, never the value. */
|
|
310
|
+
export function formatProviderLine(p: ResolvedProvider): string {
|
|
311
|
+
const key = p.keySource === "stored" ? "key: stored"
|
|
312
|
+
: p.keySource === "env" ? `key: env ${p.keyEnv}`
|
|
313
|
+
: p.keySource === "inline" ? "key: ROVECODE_API_KEY"
|
|
314
|
+
: p.noKey === true ? "key: not needed"
|
|
315
|
+
: `key: NONE (rovecode auth set ${p.id} · or set ${p.keyEnv})`;
|
|
316
|
+
return `${p.id.padEnd(12)} ${p.protocol.padEnd(9)} ${p.baseUrl.padEnd(40)} ${key}${p.defaultModel !== undefined ? ` model: ${p.defaultModel}` : ""} [${p.scope}]`;
|
|
317
|
+
}
|
|
318
|
+
|
|
319
|
+
/** Listing for `rovecode provider list`, `/provider list` and the provider_list tool. Built-in providers
|
|
320
|
+
* without a key are folded into a count unless `all` — sixteen dead rows hide the live ones. */
|
|
321
|
+
export function formatProviderList(reg: ProviderRegistry, opts: { all?: boolean } = {}): string {
|
|
322
|
+
const d = reg.defaultRef();
|
|
323
|
+
const all = reg.list();
|
|
324
|
+
const rows = all.filter((p) => opts.all === true || isConfigured(p) || p.scope !== "builtin");
|
|
325
|
+
const lines = [d !== null
|
|
326
|
+
? `default → ${d.provider}/${d.model || "(no model — pass one with /model or ROVECODE_MODEL)"}${reg.defaultSelector() !== undefined ? ` (providers.json default: ${reg.defaultSelector()})` : ""}`
|
|
327
|
+
: `default → none yet. ${next("rovecode setup")} (or rovecode provider add <id> <baseUrl>, then rovecode auth set <id>)`];
|
|
328
|
+
lines.push(...rows.map(formatProviderLine));
|
|
329
|
+
const hidden = all.length - rows.length;
|
|
330
|
+
if (hidden > 0) lines.push(`(+${hidden} built-in providers without a key — \`list --all\` shows them)`);
|
|
331
|
+
lines.push(...reg.warnings().map((w) => `warning: ${w}`));
|
|
332
|
+
return lines.join("\n");
|
|
333
|
+
}
|
|
@@ -0,0 +1,209 @@
|
|
|
1
|
+
/** OpenAI Responses API wire (aion port #75, brought over 2026-09-07) — the second OpenAI adapter beside stream.ts's
|
|
2
|
+
* /chat/completions pair. ADR-003 seam: ONE StreamFn, errors cross as stopReason "error" / "aborted", never a throw.
|
|
3
|
+
* Hand-rolled fetch + SSE exactly like stream.ts (no `openai` npm dependency). providers/wire-select.ts decides per
|
|
4
|
+
* call which of the two wires a request takes; this module never imports stream.ts at runtime (it shares sse.ts,
|
|
5
|
+
* stream-errors.ts, usage.ts, thinking.ts, catalog.ts and wire-responses.ts with it).
|
|
6
|
+
*
|
|
7
|
+
* Request: POST <baseUrl>/responses `{ model, instructions?, input, tools?, stream: true, store: false,
|
|
8
|
+
* max_output_tokens?, reasoning? }`. `max_output_tokens` is max(cap, 16) — cap = model.maxTokens, the stream.ts rule
|
|
9
|
+
* — and is OMITTED on the ChatGPT Codex backend (a URL containing chatgpt.com/backend-api; pi's codex path never
|
|
10
|
+
* sends it), which also gets the `OpenAI-Beta: responses=experimental` + `originator: rovecode` headers.
|
|
11
|
+
* `reasoning: { effort, summary: "auto" }` is the Responses spelling of rovecode's ONE thinking dial
|
|
12
|
+
* (providers/thinking.ts thinkingPlan — the same dialect table stream.ts spreads as `reasoning_effort`; auto sends
|
|
13
|
+
* nothing, a model the catalog or the dialect marks without a reasoning mode sends nothing).
|
|
14
|
+
* NOT brought over: `include: ["reasoning.encrypted_content"]` and the reasoning item as a message part — rovecode
|
|
15
|
+
* has no reasoning MessagePart to replay it from (wire-responses.ts header), so asking the provider for encrypted
|
|
16
|
+
* content we would then discard is cost without product. Reasoning TEXT still streams live as reasoning_delta.
|
|
17
|
+
* Pattern sources, MIT — earendil-works/pi packages/ai/src/api: openai-responses.ts:290-302 (store:false,
|
|
18
|
+
* stream:true, max_output_tokens), :32-33 (the 16-token floor the API enforces), :323-332 (reasoning effort +
|
|
19
|
+
* summary "auto"); openai-codex-responses.ts:549-556 (`instructions`), :45 + :633-639 (chatgpt.com/backend-api →
|
|
20
|
+
* codex/responses), :1607-1609 + :1622 (chatgpt-account-id, originator, `OpenAI-Beta: responses=experimental`,
|
|
21
|
+
* accept text/event-stream); openai-responses-shared.ts:440,:463 (output slots keyed by output_index), :597-668
|
|
22
|
+
* (the deltas), :680-739 (output_item.done finalises a slot: message text authoritative, function_call arguments
|
|
23
|
+
* parsed), :757-759 (a stream without a terminal event is an error), :762-792 (stop mapping), :592-594 (a tool call
|
|
24
|
+
* overrides a completed stop), :742-755 (`<code>: <message>` for response.failed / error events).
|
|
25
|
+
* Pattern source, Apache-2.0 at pattern level (no code copied) — openai/codex: codex-rs/codex-api/src/sse/
|
|
26
|
+
* responses.rs:357-518 (the event set a client consumes; :843-845 fixtures pair `event:` and `data:` lines),
|
|
27
|
+
* codex-rs/model-provider-info/src/lib.rs:40 (CHATGPT_CODEX_BASE_URL), codex-rs/core/src/client.rs:688 (an
|
|
28
|
+
* `originator` header on every request — the value is `rovecode` here; codex's own OpenAI-Beta value is
|
|
29
|
+
* websocket-only, so `responses=experimental` is pi's). NOTICE entry: THIRD_PARTY_NOTICES.md, `## openai/codex`.
|
|
30
|
+
* NOT ported: `previous_response_id`, `prompt_cache_key` / `prompt_cache_retention`, `text.verbosity`,
|
|
31
|
+
* `service_tier`, `parallel_tool_calls`, `tool_choice`, strict / grammar / custom tools, developer-role items,
|
|
32
|
+
* the websocket and zstd transports, pi's retry loop (rovecode's withRetry + router classify the error turn as
|
|
33
|
+
* for chat completions) and codex's error taxonomy. Selection is static — no runtime 404 → chat re-issue.
|
|
34
|
+
*
|
|
35
|
+
* Events (`data:` payloads through sse.ts; `event:` lines are ignored — `type` names the event):
|
|
36
|
+
* `response.output_text.delta` / `response.refusal.delta` → text_delta (and the abort-salvage buffer);
|
|
37
|
+
* `response.reasoning_summary_text.delta` / `response.reasoning_text.delta` → reasoning_delta — NEVER into the
|
|
38
|
+
* text buffer, never a text part (`…summary_part.done` → one "\n\n" reasoning_delta);
|
|
39
|
+
* `response.function_call_arguments.delta` accumulates per output_index (a tool_call_delta once the slot knows
|
|
40
|
+
* its call id and name); `response.output_item.done` finalises the slot (a reasoning item finalises to nothing);
|
|
41
|
+
* `response.completed` → tool_use when a tool_call part exists, else end_turn; `response.incomplete` → length for
|
|
42
|
+
* max_output_tokens, else error naming the reason; `response.failed` / `error` → error "<code>: <message>"; no
|
|
43
|
+
* terminal event → error. Usage: `response.usage` through core/usage.ts normalizeUsage (input_tokens −
|
|
44
|
+
* input_tokens_details.cached_tokens, cacheRead; reasoning_tokens are not carried — TokenUsage has no field for
|
|
45
|
+
* them and normalizeUsage does not expose them). Non-2xx → httpErrorTurn (status + Retry-After for withRetry / the
|
|
46
|
+
* router); a thrown fetch or abort → failedTurn (an aborted turn keeps the streamed text, never a tool_call
|
|
47
|
+
* fragment). No header, token or account id ever reaches an error text or an event (oauth/common.ts hygiene rule). */
|
|
48
|
+
|
|
49
|
+
import type { AssistantTurn, Message, MessagePart, ModelRef, StopReason, StreamEvent, StreamFn, StreamOptions, TokenUsage } from "../core/types.ts";
|
|
50
|
+
import { normalizeUsage } from "../core/usage.ts";
|
|
51
|
+
import { supportsImages } from "./catalog.ts";
|
|
52
|
+
import { sseLines } from "./sse.ts";
|
|
53
|
+
import { failedTurn, fetchFirstByte, httpErrorTurn } from "./stream-errors.ts";
|
|
54
|
+
import { thinkingPlan } from "./thinking.ts";
|
|
55
|
+
import { toResponsesInput, toResponsesTools } from "./wire-responses.ts";
|
|
56
|
+
|
|
57
|
+
/** the same opts as openaiCompatStreaming — oauth/seam.ts MakeStream is unchanged */
|
|
58
|
+
export interface ResponsesAdapterOptions { baseUrl: string; apiKey: string; headers?: Record<string, string> }
|
|
59
|
+
|
|
60
|
+
/** the API rejects max_output_tokens below 16 (pi openai-responses.ts:32-33) */
|
|
61
|
+
export const RESPONSES_MIN_OUTPUT_TOKENS = 16;
|
|
62
|
+
/** the ChatGPT Codex backend: no max_output_tokens, plus the two headers below (pi openai-codex-responses.ts) */
|
|
63
|
+
export const CODEX_BACKEND = /chatgpt\.com\/backend-api/;
|
|
64
|
+
export const CODEX_HEADERS: Readonly<Record<string, string>> = { "OpenAI-Beta": "responses=experimental", originator: "rovecode" };
|
|
65
|
+
export const NO_TERMINAL_EVENT = "OpenAI Responses stream ended before a terminal response event";
|
|
66
|
+
|
|
67
|
+
export interface ResponsesRequest { url: string; headers: Record<string, string>; body: Record<string, unknown> }
|
|
68
|
+
|
|
69
|
+
/** the Responses spelling of the thinking dial: the word the OpenAI-compatible dialect would put in `reasoning_effort`
|
|
70
|
+
* (or OpenRouter's `reasoning.effort`) becomes `reasoning: { effort, summary: "auto" }`; nothing when the dial sends
|
|
71
|
+
* nothing (auto, a model without a reasoning mode, the catalog's `reasoning: false` stamp) */
|
|
72
|
+
export function responsesReasoning(model: ModelRef): { reasoning?: { effort: string; summary: "auto" } } {
|
|
73
|
+
const fields = thinkingPlan(model, "openai").fields;
|
|
74
|
+
const nested = fields.reasoning as { effort?: unknown } | undefined;
|
|
75
|
+
const word = typeof fields.reasoning_effort === "string" ? fields.reasoning_effort : typeof nested?.effort === "string" ? nested.effort : undefined;
|
|
76
|
+
return word === undefined ? {} : { reasoning: { effort: word, summary: "auto" } };
|
|
77
|
+
}
|
|
78
|
+
|
|
79
|
+
/** URL, headers and body for one call (header rules); exported so the request shape is testable without fetch */
|
|
80
|
+
export function responsesRequest(opts: ResponsesAdapterOptions, model: ModelRef, messages: Message[], options?: StreamOptions): ResponsesRequest {
|
|
81
|
+
const url = opts.baseUrl.replace(/\/$/, "") + "/responses";
|
|
82
|
+
const codex = CODEX_BACKEND.test(url);
|
|
83
|
+
const { instructions, input } = toResponsesInput(messages, { vision: supportsImages(model) !== false, model }); // port #34 vision gate (stream.ts wireOptions rule)
|
|
84
|
+
const cap = model.maxTokens;
|
|
85
|
+
const body: Record<string, unknown> = {
|
|
86
|
+
model: model.model,
|
|
87
|
+
...(instructions !== undefined ? { instructions } : {}),
|
|
88
|
+
input,
|
|
89
|
+
...(options?.tools?.length ? { tools: toResponsesTools(options.tools) } : {}),
|
|
90
|
+
stream: true,
|
|
91
|
+
store: false,
|
|
92
|
+
...(cap && !codex ? { max_output_tokens: Math.max(cap, RESPONSES_MIN_OUTPUT_TOKENS) } : {}),
|
|
93
|
+
...responsesReasoning(model),
|
|
94
|
+
};
|
|
95
|
+
const headers: Record<string, string> = { "content-type": "application/json", accept: "text/event-stream", ...(codex ? CODEX_HEADERS : {}), ...opts.headers, authorization: `Bearer ${opts.apiKey}` };
|
|
96
|
+
return { url, headers, body };
|
|
97
|
+
}
|
|
98
|
+
|
|
99
|
+
// ---------- SSE state ----------
|
|
100
|
+
|
|
101
|
+
type Slot = { kind: "message"; text: string } | { kind: "function_call"; callId: string; name: string; args: string } | { kind: "reasoning" };
|
|
102
|
+
interface Item { type?: string; id?: string; call_id?: string; name?: string; arguments?: string; content?: { type?: string; text?: string; refusal?: string }[]; summary?: { text?: string }[] }
|
|
103
|
+
interface Event {
|
|
104
|
+
type?: string; output_index?: number; delta?: string; arguments?: string; item?: Item; code?: string; message?: string;
|
|
105
|
+
response?: { usage?: unknown; incomplete_details?: { reason?: string } | null; error?: { code?: string; message?: string } | null };
|
|
106
|
+
}
|
|
107
|
+
|
|
108
|
+
function newSlot(item: Item | undefined): Slot | undefined {
|
|
109
|
+
if (item?.type === "message") return { kind: "message", text: "" };
|
|
110
|
+
if (item?.type === "function_call") return { kind: "function_call", callId: item.call_id ?? "", name: item.name ?? "", args: item.arguments ?? "" };
|
|
111
|
+
if (item?.type === "reasoning") return { kind: "reasoning" };
|
|
112
|
+
return undefined;
|
|
113
|
+
}
|
|
114
|
+
|
|
115
|
+
/** the part an `output_item.done` item finalises to (pi shared.ts:680-739); undefined for an empty message and for a
|
|
116
|
+
* reasoning item (no reasoning MessagePart here — header) */
|
|
117
|
+
function finishedPart(item: Item, slot: Slot | undefined, index: number): MessagePart | undefined {
|
|
118
|
+
if (item.type === "message") {
|
|
119
|
+
const text = item.content ? item.content.map((c) => (c.type === "refusal" ? c.refusal ?? "" : c.text ?? "")).join("") : slot?.kind === "message" ? slot.text : "";
|
|
120
|
+
return text ? { kind: "text", text } : undefined;
|
|
121
|
+
}
|
|
122
|
+
if (item.type === "function_call") {
|
|
123
|
+
const fc = slot?.kind === "function_call" ? slot : undefined;
|
|
124
|
+
const raw = item.arguments ?? fc?.args ?? "";
|
|
125
|
+
let args: unknown = {};
|
|
126
|
+
try { args = JSON.parse(raw || "{}"); } catch { args = { _raw: raw }; }
|
|
127
|
+
return { kind: "tool_call", id: item.call_id || fc?.callId || item.id || `fc${index}`, tool: item.name || fc?.name || "unknown", args };
|
|
128
|
+
}
|
|
129
|
+
return undefined;
|
|
130
|
+
}
|
|
131
|
+
|
|
132
|
+
/** the same normalization the stream.ts adapters apply (cached share subtracted, zero fields dropped) */
|
|
133
|
+
function turnUsage(raw: unknown): TokenUsage {
|
|
134
|
+
const u = normalizeUsage(raw);
|
|
135
|
+
return { input: u.input, output: u.output, cacheRead: u.cacheRead || undefined, cacheWrite: u.cacheWrite || undefined };
|
|
136
|
+
}
|
|
137
|
+
|
|
138
|
+
// ---------- the adapter ----------
|
|
139
|
+
|
|
140
|
+
export function openaiResponsesStream(opts: ResponsesAdapterOptions): StreamFn {
|
|
141
|
+
return async function* (model: ModelRef, messages: Message[], options?: StreamOptions): AsyncGenerator<StreamEvent> {
|
|
142
|
+
let turn: AssistantTurn;
|
|
143
|
+
let buffer = ""; // text deltas only — the abort salvage (failedTurn); reasoning deltas never land here
|
|
144
|
+
try {
|
|
145
|
+
const req = responsesRequest(opts, model, messages, options);
|
|
146
|
+
const res = await fetchFirstByte(req.url, { method: "POST", headers: req.headers, body: JSON.stringify(req.body), signal: options?.signal });
|
|
147
|
+
if (!res.ok || !res.body) {
|
|
148
|
+
yield { type: "turn", turn: await httpErrorTurn(res) }; // status + Retry-After recorded for withRetry (stream-errors.ts)
|
|
149
|
+
return;
|
|
150
|
+
}
|
|
151
|
+
const slots = new Map<number, Slot>();
|
|
152
|
+
const parts = new Map<number, MessagePart>();
|
|
153
|
+
let usage: TokenUsage = { input: 0, output: 0 };
|
|
154
|
+
let terminal: { stop: StopReason; error?: string } | undefined;
|
|
155
|
+
for await (const line of sseLines(res.body)) {
|
|
156
|
+
const ev = JSON.parse(line) as Event;
|
|
157
|
+
const idx = ev.output_index ?? 0;
|
|
158
|
+
switch (ev.type) {
|
|
159
|
+
case "response.output_item.added": { const s = newSlot(ev.item); if (s) slots.set(idx, s); break; }
|
|
160
|
+
case "response.output_text.delta":
|
|
161
|
+
case "response.refusal.delta": {
|
|
162
|
+
if (!ev.delta) break;
|
|
163
|
+
const s: Slot = slots.get(idx) ?? { kind: "message", text: "" };
|
|
164
|
+
if (s.kind === "message") { s.text += ev.delta; slots.set(idx, s); }
|
|
165
|
+
buffer += ev.delta;
|
|
166
|
+
yield { type: "text_delta", text: ev.delta };
|
|
167
|
+
break;
|
|
168
|
+
}
|
|
169
|
+
case "response.reasoning_summary_text.delta":
|
|
170
|
+
case "response.reasoning_text.delta": if (ev.delta) yield { type: "reasoning_delta", text: ev.delta }; break;
|
|
171
|
+
case "response.reasoning_summary_part.done": yield { type: "reasoning_delta", text: "\n\n" }; break;
|
|
172
|
+
case "response.function_call_arguments.delta": {
|
|
173
|
+
if (!ev.delta) break;
|
|
174
|
+
const s: Slot = slots.get(idx) ?? { kind: "function_call", callId: "", name: "", args: "" };
|
|
175
|
+
if (s.kind !== "function_call") break;
|
|
176
|
+
s.args += ev.delta;
|
|
177
|
+
slots.set(idx, s);
|
|
178
|
+
if (s.callId && s.name) yield { type: "tool_call_delta", id: s.callId, tool: s.name, argsDelta: ev.delta };
|
|
179
|
+
break;
|
|
180
|
+
}
|
|
181
|
+
case "response.function_call_arguments.done": { const s = slots.get(idx); if (s?.kind === "function_call" && typeof ev.arguments === "string") s.args = ev.arguments; break; }
|
|
182
|
+
case "response.output_item.done": {
|
|
183
|
+
if (ev.item) { const p = finishedPart(ev.item, slots.get(idx), idx); if (p) parts.set(idx, p); }
|
|
184
|
+
slots.delete(idx);
|
|
185
|
+
break;
|
|
186
|
+
}
|
|
187
|
+
case "response.completed":
|
|
188
|
+
case "response.incomplete": {
|
|
189
|
+
if (ev.response?.usage) usage = turnUsage(ev.response.usage);
|
|
190
|
+
const reason = ev.response?.incomplete_details?.reason;
|
|
191
|
+
terminal = ev.type === "response.completed" ? { stop: "end_turn" } : reason === "max_output_tokens" ? { stop: "length" } : { stop: "error", error: `Response incomplete: ${reason ?? "no reason given"}` };
|
|
192
|
+
break;
|
|
193
|
+
}
|
|
194
|
+
case "response.failed": { const e = ev.response?.error; terminal = { stop: "error", error: `${e?.code || "unknown"}: ${e?.message || "no message"}` }; break; }
|
|
195
|
+
case "error": terminal = { stop: "error", error: `${ev.code || "unknown"}: ${ev.message || "no message"}` }; break;
|
|
196
|
+
default: break; // response.created / in_progress / content_part.* / *_part.added / *.done text echoes: nothing to do
|
|
197
|
+
}
|
|
198
|
+
}
|
|
199
|
+
if (!terminal) terminal = { stop: "error", error: NO_TERMINAL_EVENT }; // pi shared.ts:757-759 — never end_turn
|
|
200
|
+
const ordered = [...parts].sort((a, b) => a[0] - b[0]).map(([, p]) => p);
|
|
201
|
+
turn = terminal.stop === "error"
|
|
202
|
+
? { parts: [], stopReason: "error", usage, error: terminal.error }
|
|
203
|
+
: { parts: ordered, stopReason: terminal.stop === "end_turn" && ordered.some((p) => p.kind === "tool_call") ? "tool_use" : terminal.stop, usage };
|
|
204
|
+
} catch (e) {
|
|
205
|
+
turn = failedTurn(e, options?.signal, buffer); // mid-stream abort: the text already streamed survives, tool fragments do not
|
|
206
|
+
}
|
|
207
|
+
yield { type: "turn", turn };
|
|
208
|
+
};
|
|
209
|
+
}
|