poe-code 0.1.26 → 1.0.0
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/README.md +110 -29
- package/dist/bin/poe-claude.js +16 -0
- package/dist/bin/poe-codex.js +16 -0
- package/dist/bin/poe-kimi-cli.js +16 -0
- package/dist/bin/poe-opencode.js +16 -0
- package/dist/cli/binary-aliases.d.ts +7 -0
- package/dist/cli/binary-aliases.js +26 -0
- package/dist/cli/binary-aliases.js.map +1 -0
- package/dist/cli/bootstrap.d.ts +4 -0
- package/dist/cli/bootstrap.js +78 -0
- package/dist/cli/bootstrap.js.map +1 -0
- package/dist/cli/command-runner.d.ts +1 -1
- package/dist/cli/command-runner.js +16 -2
- package/dist/cli/command-runner.js.map +1 -1
- package/dist/cli/commands/configure-payload.d.ts +15 -0
- package/dist/cli/commands/configure-payload.js +31 -0
- package/dist/cli/commands/configure-payload.js.map +1 -0
- package/dist/cli/commands/configure.d.ts +4 -3
- package/dist/cli/commands/configure.js +102 -101
- package/dist/cli/commands/configure.js.map +1 -1
- package/dist/cli/commands/ensure-isolated-config.d.ts +12 -0
- package/dist/cli/commands/ensure-isolated-config.js +53 -0
- package/dist/cli/commands/ensure-isolated-config.js.map +1 -0
- package/dist/cli/commands/generate.d.ts +9 -0
- package/dist/cli/commands/generate.js +323 -0
- package/dist/cli/commands/generate.js.map +1 -0
- package/dist/cli/commands/install.d.ts +4 -0
- package/dist/cli/commands/install.js +40 -0
- package/dist/cli/commands/install.js.map +1 -0
- package/dist/cli/commands/login.js +62 -8
- package/dist/cli/commands/login.js.map +1 -1
- package/dist/cli/commands/{root.d.ts → mcp.d.ts} +1 -1
- package/dist/cli/commands/mcp.js +168 -0
- package/dist/cli/commands/mcp.js.map +1 -0
- package/dist/cli/commands/shared.d.ts +16 -11
- package/dist/cli/commands/shared.js +45 -24
- package/dist/cli/commands/shared.js.map +1 -1
- package/dist/cli/commands/spawn.d.ts +14 -4
- package/dist/cli/commands/spawn.js +92 -77
- package/dist/cli/commands/spawn.js.map +1 -1
- package/dist/cli/commands/test.d.ts +6 -4
- package/dist/cli/commands/test.js +111 -16
- package/dist/cli/commands/test.js.map +1 -1
- package/dist/cli/commands/unconfigure.d.ts +7 -0
- package/dist/cli/commands/unconfigure.js +116 -0
- package/dist/cli/commands/unconfigure.js.map +1 -0
- package/dist/cli/commands/version.d.ts +3 -0
- package/dist/cli/commands/version.js +31 -0
- package/dist/cli/commands/version.js.map +1 -0
- package/dist/cli/commands/wrap.d.ts +3 -0
- package/dist/cli/commands/wrap.js +61 -0
- package/dist/cli/commands/wrap.js.map +1 -0
- package/dist/cli/constants.d.ts +23 -9
- package/dist/cli/constants.js +42 -9
- package/dist/cli/constants.js.map +1 -1
- package/dist/cli/container.d.ts +4 -10
- package/dist/cli/container.js +18 -22
- package/dist/cli/container.js.map +1 -1
- package/dist/cli/context.d.ts +9 -13
- package/dist/cli/context.js +45 -152
- package/dist/cli/context.js.map +1 -1
- package/dist/cli/environment.d.ts +2 -0
- package/dist/cli/environment.js +80 -0
- package/dist/cli/environment.js.map +1 -1
- package/dist/cli/error-logger.d.ts +18 -2
- package/dist/cli/error-logger.js +25 -7
- package/dist/cli/error-logger.js.map +1 -1
- package/dist/cli/errors.d.ts +14 -10
- package/dist/cli/errors.js +25 -12
- package/dist/cli/errors.js.map +1 -1
- package/dist/cli/exit-signals.d.ts +4 -0
- package/dist/cli/exit-signals.js +8 -0
- package/dist/cli/exit-signals.js.map +1 -0
- package/dist/cli/http.d.ts +1 -0
- package/dist/cli/isolated-env-runner.d.ts +10 -0
- package/dist/cli/isolated-env-runner.js +46 -0
- package/dist/cli/isolated-env-runner.js.map +1 -0
- package/dist/cli/isolated-env.d.ts +22 -0
- package/dist/cli/isolated-env.js +176 -0
- package/dist/cli/isolated-env.js.map +1 -0
- package/dist/cli/logger.d.ts +11 -1
- package/dist/cli/logger.js +101 -16
- package/dist/cli/logger.js.map +1 -1
- package/dist/cli/mcp-output-format.d.ts +2 -0
- package/dist/cli/mcp-output-format.js +22 -0
- package/dist/cli/mcp-output-format.js.map +1 -0
- package/dist/cli/mcp-server.d.ts +21 -0
- package/dist/cli/mcp-server.js +286 -0
- package/dist/cli/mcp-server.js.map +1 -0
- package/dist/cli/options.d.ts +18 -3
- package/dist/cli/options.js +39 -12
- package/dist/cli/options.js.map +1 -1
- package/dist/cli/poe-code-command-runner.d.ts +6 -0
- package/dist/cli/poe-code-command-runner.js +57 -0
- package/dist/cli/poe-code-command-runner.js.map +1 -0
- package/dist/cli/program.js +134 -23
- package/dist/cli/program.js.map +1 -1
- package/dist/cli/prompt-runner.d.ts +10 -0
- package/dist/cli/prompt-runner.js +79 -0
- package/dist/cli/prompt-runner.js.map +1 -0
- package/dist/cli/prompts.d.ts +23 -7
- package/dist/cli/prompts.js +17 -33
- package/dist/cli/prompts.js.map +1 -1
- package/dist/cli/service-registry.d.ts +86 -21
- package/dist/cli/service-registry.js +60 -12
- package/dist/cli/service-registry.js.map +1 -1
- package/dist/cli/ui/service-menu.d.ts +7 -0
- package/dist/cli/ui/service-menu.js +43 -0
- package/dist/cli/ui/service-menu.js.map +1 -0
- package/dist/index.d.ts +6 -2
- package/dist/index.js +7 -74
- package/dist/index.js.map +1 -1
- package/dist/providers/claude-code.d.ts +13 -0
- package/dist/providers/claude-code.js +136 -0
- package/dist/providers/claude-code.js.map +1 -0
- package/dist/providers/codex.d.ts +17 -0
- package/dist/providers/codex.js +144 -0
- package/dist/providers/codex.js.map +1 -0
- package/dist/providers/create-provider.d.ts +29 -0
- package/dist/providers/create-provider.js +63 -0
- package/dist/providers/create-provider.js.map +1 -0
- package/dist/providers/index.d.ts +2 -2
- package/dist/providers/index.js +5 -10
- package/dist/providers/index.js.map +1 -1
- package/dist/providers/kimi.d.ts +4 -0
- package/dist/providers/kimi.js +123 -0
- package/dist/providers/kimi.js.map +1 -0
- package/dist/providers/opencode.d.ts +3 -0
- package/dist/providers/opencode.js +119 -0
- package/dist/providers/opencode.js.map +1 -0
- package/dist/providers/spawn-options.d.ts +12 -0
- package/dist/providers/spawn-options.js +2 -0
- package/dist/providers/spawn-options.js.map +1 -0
- package/dist/sdk/container.d.ts +18 -0
- package/dist/sdk/container.js +127 -0
- package/dist/sdk/container.js.map +1 -0
- package/dist/sdk/credentials.d.ts +9 -0
- package/dist/sdk/credentials.js +35 -0
- package/dist/sdk/credentials.js.map +1 -0
- package/dist/sdk/generate.d.ts +5 -0
- package/dist/sdk/generate.js +97 -0
- package/dist/sdk/generate.js.map +1 -0
- package/dist/sdk/spawn-core.d.ts +23 -0
- package/dist/sdk/spawn-core.js +105 -0
- package/dist/sdk/spawn-core.js.map +1 -0
- package/dist/sdk/spawn.d.ts +23 -0
- package/dist/sdk/spawn.js +41 -0
- package/dist/sdk/spawn.js.map +1 -0
- package/dist/sdk/types.d.ts +38 -0
- package/dist/sdk/types.js +2 -0
- package/dist/sdk/types.js.map +1 -0
- package/dist/services/client-instance.d.ts +12 -0
- package/dist/services/client-instance.js +36 -0
- package/dist/services/client-instance.js.map +1 -0
- package/dist/services/credentials.d.ts +13 -0
- package/dist/services/credentials.js +132 -14
- package/dist/services/credentials.js.map +1 -1
- package/dist/services/llm-client.d.ts +22 -0
- package/dist/services/llm-client.js +138 -0
- package/dist/services/llm-client.js.map +1 -0
- package/dist/services/media-download.d.ts +17 -0
- package/dist/services/media-download.js +42 -0
- package/dist/services/media-download.js.map +1 -0
- package/dist/services/model-strategy.d.ts +2 -2
- package/dist/services/model-strategy.js +36 -31
- package/dist/services/model-strategy.js.map +1 -1
- package/dist/services/mutation-events.d.ts +4 -0
- package/dist/services/mutation-events.js +53 -0
- package/dist/services/mutation-events.js.map +1 -0
- package/dist/services/service-install.d.ts +5 -4
- package/dist/services/service-install.js +36 -13
- package/dist/services/service-install.js.map +1 -1
- package/dist/services/version.d.ts +11 -0
- package/dist/services/version.js +28 -0
- package/dist/services/version.js.map +1 -0
- package/dist/templates/codex/config.toml.hbs +3 -3
- package/dist/tools/label-generator.d.ts +11 -0
- package/dist/tools/label-generator.js +78 -0
- package/dist/tools/label-generator.js.map +1 -0
- package/dist/utils/command-checks.d.ts +45 -0
- package/dist/utils/command-checks.js +120 -0
- package/dist/utils/command-checks.js.map +1 -0
- package/dist/utils/dry-run.d.ts +1 -0
- package/dist/utils/dry-run.js +116 -7
- package/dist/utils/dry-run.js.map +1 -1
- package/dist/utils/execution-context.d.ts +40 -0
- package/dist/utils/execution-context.js +128 -0
- package/dist/utils/execution-context.js.map +1 -0
- package/dist/utils/json.js +3 -0
- package/dist/utils/json.js.map +1 -1
- package/dist/utils/templates.d.ts +1 -0
- package/dist/utils/templates.js +3 -4
- package/dist/utils/templates.js.map +1 -1
- package/package.json +52 -46
- package/dist/cli/api-client.d.ts +0 -14
- package/dist/cli/api-client.js +0 -140
- package/dist/cli/api-client.js.map +0 -1
- package/dist/cli/chat.d.ts +0 -32
- package/dist/cli/chat.js +0 -2
- package/dist/cli/chat.js.map +0 -1
- package/dist/cli/commands/agent.d.ts +0 -18
- package/dist/cli/commands/agent.js +0 -84
- package/dist/cli/commands/agent.js.map +0 -1
- package/dist/cli/commands/configure-agents.d.ts +0 -3
- package/dist/cli/commands/configure-agents.js +0 -72
- package/dist/cli/commands/configure-agents.js.map +0 -1
- package/dist/cli/commands/init.d.ts +0 -8
- package/dist/cli/commands/init.js +0 -45
- package/dist/cli/commands/init.js.map +0 -1
- package/dist/cli/commands/interactive.d.ts +0 -3
- package/dist/cli/commands/interactive.js +0 -17
- package/dist/cli/commands/interactive.js.map +0 -1
- package/dist/cli/commands/logout.d.ts +0 -3
- package/dist/cli/commands/logout.js +0 -31
- package/dist/cli/commands/logout.js.map +0 -1
- package/dist/cli/commands/prerequisites.d.ts +0 -3
- package/dist/cli/commands/prerequisites.js +0 -23
- package/dist/cli/commands/prerequisites.js.map +0 -1
- package/dist/cli/commands/query.d.ts +0 -7
- package/dist/cli/commands/query.js +0 -46
- package/dist/cli/commands/query.js.map +0 -1
- package/dist/cli/commands/remove.d.ts +0 -6
- package/dist/cli/commands/remove.js +0 -80
- package/dist/cli/commands/remove.js.map +0 -1
- package/dist/cli/commands/root.js +0 -7
- package/dist/cli/commands/root.js.map +0 -1
- package/dist/cli/commands/spawn-worktree.d.ts +0 -6
- package/dist/cli/commands/spawn-worktree.js +0 -91
- package/dist/cli/commands/spawn-worktree.js.map +0 -1
- package/dist/cli/file-mentions.d.ts +0 -13
- package/dist/cli/file-mentions.js +0 -87
- package/dist/cli/file-mentions.js.map +0 -1
- package/dist/cli/file-picker-state.d.ts +0 -14
- package/dist/cli/file-picker-state.js +0 -23
- package/dist/cli/file-picker-state.js.map +0 -1
- package/dist/cli/interactive-command-runner.d.ts +0 -11
- package/dist/cli/interactive-command-runner.js +0 -142
- package/dist/cli/interactive-command-runner.js.map +0 -1
- package/dist/cli/interactive-launcher.d.ts +0 -2
- package/dist/cli/interactive-launcher.js +0 -404
- package/dist/cli/interactive-launcher.js.map +0 -1
- package/dist/cli/interactive-tasks.d.ts +0 -7
- package/dist/cli/interactive-tasks.js +0 -199
- package/dist/cli/interactive-tasks.js.map +0 -1
- package/dist/cli/interactive.d.ts +0 -19
- package/dist/cli/interactive.js +0 -479
- package/dist/cli/interactive.js.map +0 -1
- package/dist/cli/markdown-renderer.d.ts +0 -27
- package/dist/cli/markdown-renderer.js +0 -178
- package/dist/cli/markdown-renderer.js.map +0 -1
- package/dist/cli/markdown-renderer.test.d.ts +0 -1
- package/dist/cli/markdown-renderer.test.js +0 -105
- package/dist/cli/markdown-renderer.test.js.map +0 -1
- package/dist/cli/telemetry.d.ts +0 -15
- package/dist/cli/telemetry.js +0 -45
- package/dist/cli/telemetry.js.map +0 -1
- package/dist/commands/init.d.ts +0 -9
- package/dist/commands/init.js +0 -43
- package/dist/commands/init.js.map +0 -1
- package/dist/commands/publish-placeholder.d.ts +0 -8
- package/dist/commands/publish-placeholder.js +0 -50
- package/dist/commands/publish-placeholder.js.map +0 -1
- package/dist/commands/spawn-worktree.d.ts +0 -25
- package/dist/commands/spawn-worktree.js +0 -139
- package/dist/commands/spawn-worktree.js.map +0 -1
- package/dist/providers/claude-code-adapter.d.ts +0 -20
- package/dist/providers/claude-code-adapter.js +0 -48
- package/dist/providers/claude-code-adapter.js.map +0 -1
- package/dist/providers/codex-adapter.d.ts +0 -19
- package/dist/providers/codex-adapter.js +0 -49
- package/dist/providers/codex-adapter.js.map +0 -1
- package/dist/providers/opencode-adapter.d.ts +0 -18
- package/dist/providers/opencode-adapter.js +0 -45
- package/dist/providers/opencode-adapter.js.map +0 -1
- package/dist/providers/roo-code-adapter.d.ts +0 -19
- package/dist/providers/roo-code-adapter.js +0 -70
- package/dist/providers/roo-code-adapter.js.map +0 -1
- package/dist/services/agent-config-manager.d.ts +0 -32
- package/dist/services/agent-config-manager.js +0 -130
- package/dist/services/agent-config-manager.js.map +0 -1
- package/dist/services/agent-registry.d.ts +0 -27
- package/dist/services/agent-registry.js +0 -89
- package/dist/services/agent-registry.js.map +0 -1
- package/dist/services/agent-session.d.ts +0 -24
- package/dist/services/agent-session.js +0 -107
- package/dist/services/agent-session.js.map +0 -1
- package/dist/services/agent-task-registry.d.ts +0 -131
- package/dist/services/agent-task-registry.js +0 -491
- package/dist/services/agent-task-registry.js.map +0 -1
- package/dist/services/chat.d.ts +0 -94
- package/dist/services/chat.js +0 -241
- package/dist/services/chat.js.map +0 -1
- package/dist/services/claude-code.d.ts +0 -28
- package/dist/services/claude-code.js +0 -155
- package/dist/services/claude-code.js.map +0 -1
- package/dist/services/codex.d.ts +0 -27
- package/dist/services/codex.js +0 -231
- package/dist/services/codex.js.map +0 -1
- package/dist/services/mcp-client.d.ts +0 -38
- package/dist/services/mcp-client.js +0 -170
- package/dist/services/mcp-client.js.map +0 -1
- package/dist/services/mcp-manager.d.ts +0 -28
- package/dist/services/mcp-manager.js +0 -157
- package/dist/services/mcp-manager.js.map +0 -1
- package/dist/services/opencode.d.ts +0 -25
- package/dist/services/opencode.js +0 -161
- package/dist/services/opencode.js.map +0 -1
- package/dist/services/poe-code.d.ts +0 -7
- package/dist/services/poe-code.js +0 -5
- package/dist/services/poe-code.js.map +0 -1
- package/dist/services/roo-code.d.ts +0 -22
- package/dist/services/roo-code.js +0 -134
- package/dist/services/roo-code.js.map +0 -1
- package/dist/services/service-manifest.d.ts +0 -122
- package/dist/services/service-manifest.js +0 -368
- package/dist/services/service-manifest.js.map +0 -1
- package/dist/services/task-logger.d.ts +0 -24
- package/dist/services/task-logger.js +0 -73
- package/dist/services/task-logger.js.map +0 -1
- package/dist/services/task-runner.d.ts +0 -19
- package/dist/services/task-runner.js +0 -157
- package/dist/services/task-runner.js.map +0 -1
- package/dist/services/tools.d.ts +0 -76
- package/dist/services/tools.js +0 -870
- package/dist/services/tools.js.map +0 -1
- package/dist/templates/claude-code/anthropic_key.sh.hbs +0 -2
- package/dist/utils/e2e-flags.d.ts +0 -1
- package/dist/utils/e2e-flags.js +0 -9
- package/dist/utils/e2e-flags.js.map +0 -1
- package/dist/utils/prerequisites.d.ts +0 -41
- package/dist/utils/prerequisites.js +0 -92
- package/dist/utils/prerequisites.js.map +0 -1
- package/dist/utils/toml.d.ts +0 -8
- package/dist/utils/toml.js +0 -30
- package/dist/utils/toml.js.map +0 -1
- package/dist/utils/worktree.d.ts +0 -40
- package/dist/utils/worktree.js +0 -179
- package/dist/utils/worktree.js.map +0 -1
- package/shared/conversation-layout.cjs +0 -11
- package/shared/conversation-layout.d.ts +0 -7
- package/shared/conversation-layout.js +0 -11
- package/shared/package.json +0 -13
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAC1C,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAChD,OAAO,EAAE,WAAW,EAAE,MAAM,WAAW,CAAC;AAGxC,MAAM,UAAU,mBAAmB;IACjC,OAAO,CAAC,iBAAiB,EAAE,YAAY,EAAE,eAAe,EAAE,WAAW,CAAC,CAAC;AACzE,CAAC"}
|
|
@@ -0,0 +1,4 @@
|
|
|
1
|
+
import { type ServiceInstallDefinition } from "../services/service-install.js";
|
|
2
|
+
import type { ProviderSpawnOptions, ModelConfigureOptions, EmptyProviderOptions } from "./spawn-options.js";
|
|
3
|
+
export declare const KIMI_INSTALL_DEFINITION: ServiceInstallDefinition;
|
|
4
|
+
export declare const kimiService: import("../cli/service-registry.js").ProviderService<ModelConfigureOptions, EmptyProviderOptions, ProviderSpawnOptions>;
|
|
@@ -0,0 +1,123 @@
|
|
|
1
|
+
import { createBinaryExistsCheck, createCommandExpectationCheck } from "../utils/command-checks.js";
|
|
2
|
+
import { configMutation, fileMutation } from "@poe-code/config-mutations";
|
|
3
|
+
import { KIMI_MODELS, DEFAULT_KIMI_MODEL, PROVIDER_NAME, stripModelNamespace } from "../cli/constants.js";
|
|
4
|
+
import { createProvider } from "./create-provider.js";
|
|
5
|
+
import { kimiAgent } from "@poe-code/agent-defs";
|
|
6
|
+
export const KIMI_INSTALL_DEFINITION = {
|
|
7
|
+
id: "kimi",
|
|
8
|
+
summary: "Kimi CLI",
|
|
9
|
+
check: createBinaryExistsCheck("kimi", "kimi-cli-binary", "Kimi CLI binary must exist"),
|
|
10
|
+
steps: [
|
|
11
|
+
{
|
|
12
|
+
id: "install-kimi-cli-uv",
|
|
13
|
+
command: "uv",
|
|
14
|
+
args: ["tool", "install", "--python", "3.13", "kimi-cli"]
|
|
15
|
+
}
|
|
16
|
+
],
|
|
17
|
+
successMessage: "Installed Kimi CLI via uv."
|
|
18
|
+
};
|
|
19
|
+
function providerModel(model) {
|
|
20
|
+
const stripped = stripModelNamespace(model);
|
|
21
|
+
return `${PROVIDER_NAME}/${stripped}`;
|
|
22
|
+
}
|
|
23
|
+
function buildKimiArgs(prompt, extraArgs) {
|
|
24
|
+
return ["--quiet", "-p", prompt, ...(extraArgs ?? [])];
|
|
25
|
+
}
|
|
26
|
+
export const kimiService = createProvider({
|
|
27
|
+
...kimiAgent,
|
|
28
|
+
disabled: false,
|
|
29
|
+
supportsStdinPrompt: false,
|
|
30
|
+
configurePrompts: {
|
|
31
|
+
model: {
|
|
32
|
+
label: "Kimi default model",
|
|
33
|
+
defaultValue: DEFAULT_KIMI_MODEL,
|
|
34
|
+
choices: KIMI_MODELS.map((id) => ({
|
|
35
|
+
title: id,
|
|
36
|
+
value: id
|
|
37
|
+
}))
|
|
38
|
+
}
|
|
39
|
+
},
|
|
40
|
+
isolatedEnv: {
|
|
41
|
+
// Use "kimi-cli" to avoid stripAgentHome stripping ".kimi" from paths
|
|
42
|
+
agentBinary: "kimi-cli",
|
|
43
|
+
configProbe: { kind: "isolatedFile", relativePath: ".kimi/config.toml" },
|
|
44
|
+
env: {
|
|
45
|
+
HOME: { kind: "isolatedDir" }
|
|
46
|
+
}
|
|
47
|
+
},
|
|
48
|
+
test(context) {
|
|
49
|
+
return context.runCheck(createCommandExpectationCheck({
|
|
50
|
+
id: "kimi-cli-health",
|
|
51
|
+
command: "kimi",
|
|
52
|
+
args: buildKimiArgs("Output exactly: KIMI_OK"),
|
|
53
|
+
expectedOutput: "KIMI_OK"
|
|
54
|
+
}));
|
|
55
|
+
},
|
|
56
|
+
manifest: {
|
|
57
|
+
configure: [
|
|
58
|
+
fileMutation.ensureDirectory({ path: "~/.kimi" }),
|
|
59
|
+
configMutation.merge({
|
|
60
|
+
target: "~/.kimi/config.toml",
|
|
61
|
+
pruneByPrefix: { models: `${PROVIDER_NAME}/` },
|
|
62
|
+
value: (ctx) => {
|
|
63
|
+
const { model, apiKey, env } = (ctx ?? {});
|
|
64
|
+
const selectedModel = model ?? DEFAULT_KIMI_MODEL;
|
|
65
|
+
const models = {};
|
|
66
|
+
for (const m of KIMI_MODELS) {
|
|
67
|
+
models[providerModel(m)] = {
|
|
68
|
+
provider: PROVIDER_NAME,
|
|
69
|
+
model: stripModelNamespace(m),
|
|
70
|
+
max_context_size: 256000
|
|
71
|
+
};
|
|
72
|
+
}
|
|
73
|
+
return {
|
|
74
|
+
default_model: providerModel(selectedModel),
|
|
75
|
+
default_thinking: true,
|
|
76
|
+
models,
|
|
77
|
+
providers: {
|
|
78
|
+
[PROVIDER_NAME]: {
|
|
79
|
+
type: "openai_legacy",
|
|
80
|
+
base_url: env.poeApiBaseUrl,
|
|
81
|
+
api_key: apiKey ?? ""
|
|
82
|
+
}
|
|
83
|
+
}
|
|
84
|
+
};
|
|
85
|
+
}
|
|
86
|
+
})
|
|
87
|
+
],
|
|
88
|
+
unconfigure: [
|
|
89
|
+
configMutation.transform({
|
|
90
|
+
target: "~/.kimi/config.toml",
|
|
91
|
+
transform: (document) => {
|
|
92
|
+
const providers = document.providers;
|
|
93
|
+
if (!providers || typeof providers !== "object") {
|
|
94
|
+
return { changed: false, content: document };
|
|
95
|
+
}
|
|
96
|
+
if (!(PROVIDER_NAME in providers)) {
|
|
97
|
+
return { changed: false, content: document };
|
|
98
|
+
}
|
|
99
|
+
const { [PROVIDER_NAME]: ignoredProvider, ...rest } = providers;
|
|
100
|
+
void ignoredProvider;
|
|
101
|
+
const updatedProviders = rest;
|
|
102
|
+
if (Object.keys(updatedProviders).length === 0) {
|
|
103
|
+
const { providers: ignoredProviders, ...docWithoutProviders } = document;
|
|
104
|
+
void ignoredProviders;
|
|
105
|
+
return { changed: true, content: docWithoutProviders };
|
|
106
|
+
}
|
|
107
|
+
return { changed: true, content: { ...document, providers: updatedProviders } };
|
|
108
|
+
}
|
|
109
|
+
})
|
|
110
|
+
]
|
|
111
|
+
},
|
|
112
|
+
install: KIMI_INSTALL_DEFINITION,
|
|
113
|
+
spawn(context, options) {
|
|
114
|
+
const args = buildKimiArgs(options.prompt, options.args);
|
|
115
|
+
if (options.cwd) {
|
|
116
|
+
return context.command.runCommand("kimi", args, {
|
|
117
|
+
cwd: options.cwd
|
|
118
|
+
});
|
|
119
|
+
}
|
|
120
|
+
return context.command.runCommand("kimi", args);
|
|
121
|
+
}
|
|
122
|
+
});
|
|
123
|
+
//# sourceMappingURL=kimi.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"kimi.js","sourceRoot":"","sources":["../../src/providers/kimi.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,uBAAuB,EACvB,6BAA6B,EAC9B,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,cAAc,EACd,YAAY,EAEb,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EAAE,WAAW,EAAE,kBAAkB,EAAE,aAAa,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAC1G,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAMtD,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AAEjD,MAAM,CAAC,MAAM,uBAAuB,GAA6B;IAC/D,EAAE,EAAE,MAAM;IACV,OAAO,EAAE,UAAU;IACnB,KAAK,EAAE,uBAAuB,CAC5B,MAAM,EACN,iBAAiB,EACjB,4BAA4B,CAC7B;IACD,KAAK,EAAE;QACL;YACE,EAAE,EAAE,qBAAqB;YACzB,OAAO,EAAE,IAAI;YACb,IAAI,EAAE,CAAC,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,EAAE,UAAU,CAAC;SAC1D;KACF;IACD,cAAc,EAAE,4BAA4B;CAC7C,CAAC;AAEF,SAAS,aAAa,CAAC,KAAa;IAClC,MAAM,QAAQ,GAAG,mBAAmB,CAAC,KAAK,CAAC,CAAC;IAC5C,OAAO,GAAG,aAAa,IAAI,QAAQ,EAAE,CAAC;AACxC,CAAC;AAED,SAAS,aAAa,CAAC,MAAc,EAAE,SAAoB;IACzD,OAAO,CAAC,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC,CAAC;AACzD,CAAC;AAED,MAAM,CAAC,MAAM,WAAW,GAAG,cAAc,CAIvC;IACA,GAAG,SAAS;IACZ,QAAQ,EAAE,KAAK;IACf,mBAAmB,EAAE,KAAK;IAC1B,gBAAgB,EAAE;QAChB,KAAK,EAAE;YACL,KAAK,EAAE,oBAAoB;YAC3B,YAAY,EAAE,kBAAkB;YAChC,OAAO,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBAChC,KAAK,EAAE,EAAE;gBACT,KAAK,EAAE,EAAE;aACV,CAAC,CAAC;SACJ;KACF;IACD,WAAW,EAAE;QACX,sEAAsE;QACtE,WAAW,EAAE,UAAU;QACvB,WAAW,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,YAAY,EAAE,mBAAmB,EAAE;QACxE,GAAG,EAAE;YACH,IAAI,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE;SAC9B;KACF;IACD,IAAI,CAAC,OAAO;QACV,OAAO,OAAO,CAAC,QAAQ,CACrB,6BAA6B,CAAC;YAC5B,EAAE,EAAE,iBAAiB;YACrB,OAAO,EAAE,MAAM;YACf,IAAI,EAAE,aAAa,CAAC,yBAAyB,CAAC;YAC9C,cAAc,EAAE,SAAS;SAC1B,CAAC,CACH,CAAC;IACJ,CAAC;IACD,QAAQ,EAAE;QACR,SAAS,EAAE;YACT,YAAY,CAAC,eAAe,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;YACjD,cAAc,CAAC,KAAK,CAAC;gBACnB,MAAM,EAAE,qBAAqB;gBAC7B,aAAa,EAAE,EAAE,MAAM,EAAE,GAAG,aAAa,GAAG,EAAE;gBAC9C,KAAK,EAAE,CAAC,GAAG,EAAE,EAAE;oBACb,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,IAAI,EAAE,CAIxC,CAAC;oBACF,MAAM,aAAa,GAAG,KAAK,IAAI,kBAAkB,CAAC;oBAElD,MAAM,MAAM,GAAiB,EAAE,CAAC;oBAChC,KAAK,MAAM,CAAC,IAAI,WAAW,EAAE,CAAC;wBAC5B,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,GAAG;4BACzB,QAAQ,EAAE,aAAa;4BACvB,KAAK,EAAE,mBAAmB,CAAC,CAAC,CAAC;4BAC7B,gBAAgB,EAAE,MAAM;yBACzB,CAAC;oBACJ,CAAC;oBAED,OAAO;wBACL,aAAa,EAAE,aAAa,CAAC,aAAa,CAAC;wBAC3C,gBAAgB,EAAE,IAAI;wBACtB,MAAM;wBACN,SAAS,EAAE;4BACT,CAAC,aAAa,CAAC,EAAE;gCACf,IAAI,EAAE,eAAe;gCACrB,QAAQ,EAAE,GAAG,CAAC,aAAa;gCAC3B,OAAO,EAAE,MAAM,IAAI,EAAE;6BACtB;yBACF;qBACF,CAAC;gBACJ,CAAC;aACF,CAAC;SACH;QACD,WAAW,EAAE;YACX,cAAc,CAAC,SAAS,CAAC;gBACvB,MAAM,EAAE,qBAAqB;gBAC7B,SAAS,EAAE,CAAC,QAAQ,EAAE,EAAE;oBACtB,MAAM,SAAS,GAAG,QAAQ,CAAC,SAAqC,CAAC;oBACjE,IAAI,CAAC,SAAS,IAAI,OAAO,SAAS,KAAK,QAAQ,EAAE,CAAC;wBAChD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;oBAC/C,CAAC;oBACD,IAAI,CAAC,CAAC,aAAa,IAAI,SAAS,CAAC,EAAE,CAAC;wBAClC,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC;oBAC/C,CAAC;oBACD,MAAM,EAAE,CAAC,aAAa,CAAC,EAAE,eAAe,EAAE,GAAG,IAAI,EAAE,GAAG,SAAS,CAAC;oBAChE,KAAK,eAAe,CAAC;oBACrB,MAAM,gBAAgB,GAAG,IAAoB,CAAC;oBAC9C,IAAI,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;wBAC/C,MAAM,EAAE,SAAS,EAAE,gBAAgB,EAAE,GAAG,mBAAmB,EAAE,GAAG,QAAQ,CAAC;wBACzE,KAAK,gBAAgB,CAAC;wBACtB,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,mBAAmB,EAAE,CAAC;oBACzD,CAAC;oBACD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,GAAG,QAAQ,EAAE,SAAS,EAAE,gBAAgB,EAAE,EAAE,CAAC;gBAClF,CAAC;aACF,CAAC;SACH;KACF;IACD,OAAO,EAAE,uBAAuB;IAChC,KAAK,CAAC,OAAO,EAAE,OAAO;QACpB,MAAM,IAAI,GAAG,aAAa,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;QACzD,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;YAChB,OAAO,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,EAAE,IAAI,EAAE;gBAC9C,GAAG,EAAE,OAAO,CAAC,GAAG;aACjB,CAAC,CAAC;QACL,CAAC;QACD,OAAO,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;IAClD,CAAC;CACF,CAAC,CAAC"}
|
|
@@ -0,0 +1,119 @@
|
|
|
1
|
+
import { DEFAULT_FRONTIER_MODEL, FRONTIER_MODELS, PROVIDER_NAME } from "../cli/constants.js";
|
|
2
|
+
import { createBinaryExistsCheck, createCommandExpectationCheck } from "../utils/command-checks.js";
|
|
3
|
+
import { configMutation, fileMutation } from "@poe-code/config-mutations";
|
|
4
|
+
import { createProvider } from "./create-provider.js";
|
|
5
|
+
import { openCodeAgent } from "@poe-code/agent-defs";
|
|
6
|
+
function providerModel(model) {
|
|
7
|
+
const value = model ?? DEFAULT_FRONTIER_MODEL;
|
|
8
|
+
const prefix = `${PROVIDER_NAME}/`;
|
|
9
|
+
return value.startsWith(prefix) ? value : `${prefix}${value}`;
|
|
10
|
+
}
|
|
11
|
+
export const OPEN_CODE_INSTALL_DEFINITION = {
|
|
12
|
+
id: "opencode",
|
|
13
|
+
summary: "OpenCode CLI",
|
|
14
|
+
check: createBinaryExistsCheck("opencode", "opencode-cli-binary", "OpenCode CLI binary must exist"),
|
|
15
|
+
steps: [
|
|
16
|
+
{
|
|
17
|
+
id: "install-opencode-cli-npm",
|
|
18
|
+
command: "npm",
|
|
19
|
+
args: ["install", "-g", "opencode-ai"]
|
|
20
|
+
}
|
|
21
|
+
],
|
|
22
|
+
successMessage: "Installed OpenCode CLI via npm."
|
|
23
|
+
};
|
|
24
|
+
function getModelArgs(model) {
|
|
25
|
+
return ["--model", providerModel(model)];
|
|
26
|
+
}
|
|
27
|
+
export const openCodeService = createProvider({
|
|
28
|
+
...openCodeAgent,
|
|
29
|
+
supportsStdinPrompt: false,
|
|
30
|
+
configurePrompts: {
|
|
31
|
+
model: {
|
|
32
|
+
label: "OpenCode model",
|
|
33
|
+
defaultValue: DEFAULT_FRONTIER_MODEL,
|
|
34
|
+
choices: FRONTIER_MODELS.map((id) => ({
|
|
35
|
+
title: id,
|
|
36
|
+
value: id
|
|
37
|
+
}))
|
|
38
|
+
}
|
|
39
|
+
},
|
|
40
|
+
isolatedEnv: {
|
|
41
|
+
agentBinary: openCodeAgent.binaryName,
|
|
42
|
+
configProbe: {
|
|
43
|
+
kind: "isolatedFile",
|
|
44
|
+
relativePath: ".config/opencode/config.json"
|
|
45
|
+
},
|
|
46
|
+
env: {
|
|
47
|
+
XDG_CONFIG_HOME: { kind: "isolatedDir", relativePath: ".config" },
|
|
48
|
+
XDG_DATA_HOME: { kind: "isolatedDir", relativePath: ".local/share" }
|
|
49
|
+
}
|
|
50
|
+
},
|
|
51
|
+
manifest: {
|
|
52
|
+
configure: [
|
|
53
|
+
fileMutation.ensureDirectory({ path: "~/.config/opencode" }),
|
|
54
|
+
configMutation.merge({
|
|
55
|
+
target: "~/.config/opencode/config.json",
|
|
56
|
+
value: (ctx) => {
|
|
57
|
+
const { model } = (ctx ?? {});
|
|
58
|
+
return {
|
|
59
|
+
$schema: "https://opencode.ai/config.json",
|
|
60
|
+
model: providerModel(model),
|
|
61
|
+
enabled_providers: [PROVIDER_NAME]
|
|
62
|
+
};
|
|
63
|
+
}
|
|
64
|
+
}),
|
|
65
|
+
fileMutation.ensureDirectory({ path: "~/.opencode-data" }),
|
|
66
|
+
configMutation.merge({
|
|
67
|
+
target: "~/.opencode-data/auth.json",
|
|
68
|
+
value: (ctx) => {
|
|
69
|
+
const { apiKey } = (ctx ?? {});
|
|
70
|
+
return {
|
|
71
|
+
[PROVIDER_NAME]: {
|
|
72
|
+
type: "api",
|
|
73
|
+
key: apiKey ?? ""
|
|
74
|
+
}
|
|
75
|
+
};
|
|
76
|
+
}
|
|
77
|
+
})
|
|
78
|
+
],
|
|
79
|
+
unconfigure: [
|
|
80
|
+
configMutation.prune({
|
|
81
|
+
target: "~/.config/opencode/config.json",
|
|
82
|
+
shape: { enabled_providers: true }
|
|
83
|
+
}),
|
|
84
|
+
configMutation.prune({
|
|
85
|
+
target: "~/.opencode-data/auth.json",
|
|
86
|
+
shape: { [PROVIDER_NAME]: true }
|
|
87
|
+
})
|
|
88
|
+
]
|
|
89
|
+
},
|
|
90
|
+
install: OPEN_CODE_INSTALL_DEFINITION,
|
|
91
|
+
test(context) {
|
|
92
|
+
return context.runCheck(createCommandExpectationCheck({
|
|
93
|
+
id: "opencode-cli-health",
|
|
94
|
+
command: "opencode",
|
|
95
|
+
args: [
|
|
96
|
+
...getModelArgs(DEFAULT_FRONTIER_MODEL),
|
|
97
|
+
"run",
|
|
98
|
+
"Output exactly: OPEN_CODE_OK"
|
|
99
|
+
],
|
|
100
|
+
expectedOutput: "OPEN_CODE_OK"
|
|
101
|
+
}));
|
|
102
|
+
},
|
|
103
|
+
spawn(context, options) {
|
|
104
|
+
const opts = (options ?? {});
|
|
105
|
+
const args = [
|
|
106
|
+
...getModelArgs(opts.model),
|
|
107
|
+
"run",
|
|
108
|
+
opts.prompt,
|
|
109
|
+
...(opts.args ?? [])
|
|
110
|
+
];
|
|
111
|
+
if (opts.cwd) {
|
|
112
|
+
return context.command.runCommand("poe-code", ["wrap", "opencode", ...args], {
|
|
113
|
+
cwd: opts.cwd
|
|
114
|
+
});
|
|
115
|
+
}
|
|
116
|
+
return context.command.runCommand("poe-code", ["wrap", "opencode", ...args]);
|
|
117
|
+
}
|
|
118
|
+
});
|
|
119
|
+
//# sourceMappingURL=opencode.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"opencode.js","sourceRoot":"","sources":["../../src/providers/opencode.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,sBAAsB,EACtB,eAAe,EACf,aAAa,EACd,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,uBAAuB,EACvB,6BAA6B,EAC9B,MAAM,4BAA4B,CAAC;AAEpC,OAAO,EACL,cAAc,EACd,YAAY,EACb,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEtD,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAErD,SAAS,aAAa,CAAC,KAAc;IACnC,MAAM,KAAK,GAAG,KAAK,IAAI,sBAAsB,CAAC;IAC9C,MAAM,MAAM,GAAG,GAAG,aAAa,GAAG,CAAC;IACnC,OAAO,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,MAAM,GAAG,KAAK,EAAE,CAAC;AAChE,CAAC;AAED,MAAM,CAAC,MAAM,4BAA4B,GAA6B;IACpE,EAAE,EAAE,UAAU;IACd,OAAO,EAAE,cAAc;IACvB,KAAK,EAAE,uBAAuB,CAC5B,UAAU,EACV,qBAAqB,EACrB,gCAAgC,CACjC;IACD,KAAK,EAAE;QACL;YACE,EAAE,EAAE,0BAA0B;YAC9B,OAAO,EAAE,KAAK;YACd,IAAI,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,aAAa,CAAC;SACvC;KACF;IACD,cAAc,EAAE,iCAAiC;CAClD,CAAC;AAEF,SAAS,YAAY,CAAC,KAAc;IAClC,OAAO,CAAC,SAAS,EAAE,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;AAC3C,CAAC;AAED,MAAM,CAAC,MAAM,eAAe,GAAG,cAAc,CAAC;IAC5C,GAAG,aAAa;IAChB,mBAAmB,EAAE,KAAK;IAC1B,gBAAgB,EAAE;QAChB,KAAK,EAAE;YACL,KAAK,EAAE,gBAAgB;YACvB,YAAY,EAAE,sBAAsB;YACpC,OAAO,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;gBACpC,KAAK,EAAE,EAAE;gBACT,KAAK,EAAE,EAAE;aACV,CAAC,CAAC;SACJ;KACF;IACD,WAAW,EAAE;QACX,WAAW,EAAE,aAAa,CAAC,UAAW;QACtC,WAAW,EAAE;YACX,IAAI,EAAE,cAAc;YACpB,YAAY,EAAE,8BAA8B;SAC7C;QACD,GAAG,EAAE;YACH,eAAe,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,YAAY,EAAE,SAAS,EAAE;YACjE,aAAa,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,YAAY,EAAE,cAAc,EAAE;SACrE;KACF;IACD,QAAQ,EAAE;QACR,SAAS,EAAE;YACT,YAAY,CAAC,eAAe,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,CAAC;YAC5D,cAAc,CAAC,KAAK,CAAC;gBACnB,MAAM,EAAE,gCAAgC;gBACxC,KAAK,EAAE,CAAC,GAAG,EAAE,EAAE;oBACb,MAAM,EAAE,KAAK,EAAE,GAAG,CAAC,GAAG,IAAI,EAAE,CAAuB,CAAC;oBACpD,OAAO;wBACL,OAAO,EAAE,iCAAiC;wBAC1C,KAAK,EAAE,aAAa,CAAC,KAAK,CAAC;wBAC3B,iBAAiB,EAAE,CAAC,aAAa,CAAC;qBACnC,CAAC;gBACJ,CAAC;aACF,CAAC;YACF,YAAY,CAAC,eAAe,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,CAAC;YAC1D,cAAc,CAAC,KAAK,CAAC;gBACnB,MAAM,EAAE,4BAA4B;gBACpC,KAAK,EAAE,CAAC,GAAG,EAAE,EAAE;oBACb,MAAM,EAAE,MAAM,EAAE,GAAG,CAAC,GAAG,IAAI,EAAE,CAAwB,CAAC;oBACtD,OAAO;wBACL,CAAC,aAAa,CAAC,EAAE;4BACf,IAAI,EAAE,KAAK;4BACX,GAAG,EAAE,MAAM,IAAI,EAAE;yBAClB;qBACF,CAAC;gBACJ,CAAC;aACF,CAAC;SACH;QACD,WAAW,EAAE;YACX,cAAc,CAAC,KAAK,CAAC;gBACnB,MAAM,EAAE,gCAAgC;gBACxC,KAAK,EAAE,EAAE,iBAAiB,EAAE,IAAI,EAAE;aACnC,CAAC;YACF,cAAc,CAAC,KAAK,CAAC;gBACnB,MAAM,EAAE,4BAA4B;gBACpC,KAAK,EAAE,EAAE,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE;aACjC,CAAC;SACH;KACF;IACD,OAAO,EAAE,4BAA4B;IACrC,IAAI,CAAC,OAAO;QACV,OAAO,OAAO,CAAC,QAAQ,CACrB,6BAA6B,CAAC;YAC5B,EAAE,EAAE,qBAAqB;YACzB,OAAO,EAAE,UAAU;YACnB,IAAI,EAAE;gBACJ,GAAG,YAAY,CAAC,sBAAsB,CAAC;gBACvC,KAAK;gBACL,8BAA8B;aAC/B;YACD,cAAc,EAAE,cAAc;SAC/B,CAAC,CACH,CAAC;IACJ,CAAC;IACD,KAAK,CAAC,OAAO,EAAE,OAAO;QACpB,MAAM,IAAI,GAAG,CAAC,OAAO,IAAI,EAAE,CAAyB,CAAC;QACrD,MAAM,IAAI,GAAG;YACX,GAAG,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC;YAC3B,KAAK;YACL,IAAI,CAAC,MAAM;YACX,GAAG,CAAC,IAAI,CAAC,IAAI,IAAI,EAAE,CAAC;SACrB,CAAC;QACF,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACb,OAAO,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,UAAU,EAAE,CAAC,MAAM,EAAE,UAAU,EAAE,GAAG,IAAI,CAAC,EAAE;gBAC3E,GAAG,EAAE,IAAI,CAAC,GAAG;aACd,CAAC,CAAC;QACL,CAAC;QACD,OAAO,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,UAAU,EAAE,CAAC,MAAM,EAAE,UAAU,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC;IAC/E,CAAC;CACF,CAAC,CAAC"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
export interface SpawnCommandOptions {
|
|
2
|
+
prompt: string;
|
|
3
|
+
args?: string[];
|
|
4
|
+
model?: string;
|
|
5
|
+
cwd?: string;
|
|
6
|
+
useStdin?: boolean;
|
|
7
|
+
}
|
|
8
|
+
export type ProviderSpawnOptions<Extra extends Record<string, unknown> = Record<string, never>> = SpawnCommandOptions & Extra;
|
|
9
|
+
export interface ModelConfigureOptions {
|
|
10
|
+
model: string;
|
|
11
|
+
}
|
|
12
|
+
export type EmptyProviderOptions = Record<string, never>;
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"spawn-options.js","sourceRoot":"","sources":["../../src/providers/spawn-options.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import type { CliContainer } from "../cli/container.js";
|
|
2
|
+
export interface SdkContainerOptions {
|
|
3
|
+
/** Working directory (defaults to process.cwd()) */
|
|
4
|
+
cwd?: string;
|
|
5
|
+
/** Home directory (defaults to os.homedir()) */
|
|
6
|
+
homeDir?: string;
|
|
7
|
+
/** Environment variables (defaults to process.env) */
|
|
8
|
+
variables?: Record<string, string | undefined>;
|
|
9
|
+
/** Enable verbose logging (defaults to false) */
|
|
10
|
+
verbose?: boolean;
|
|
11
|
+
}
|
|
12
|
+
/**
|
|
13
|
+
* Creates a lightweight container for SDK usage.
|
|
14
|
+
* Uses real file system and command runner.
|
|
15
|
+
* No prompts needed (non-interactive).
|
|
16
|
+
* Minimal logger (silent by default).
|
|
17
|
+
*/
|
|
18
|
+
export declare function createSdkContainer(options?: SdkContainerOptions): CliContainer;
|
|
@@ -0,0 +1,127 @@
|
|
|
1
|
+
import * as fs from "node:fs/promises";
|
|
2
|
+
import * as os from "node:os";
|
|
3
|
+
import { createCliEnvironment } from "../cli/environment.js";
|
|
4
|
+
import { createServiceRegistry } from "../cli/service-registry.js";
|
|
5
|
+
import { createCommandContextFactory } from "../cli/context.js";
|
|
6
|
+
import { createPromptLibrary } from "../cli/prompts.js";
|
|
7
|
+
import { createOptionResolvers } from "../cli/options.js";
|
|
8
|
+
import { createLoggerFactory } from "../cli/logger.js";
|
|
9
|
+
import { ErrorLogger } from "../cli/error-logger.js";
|
|
10
|
+
import { createDefaultCommandRunner } from "../cli/command-runner.js";
|
|
11
|
+
import { getDefaultProviders } from "../providers/index.js";
|
|
12
|
+
import { createPoeCodeCommandRunner } from "../cli/poe-code-command-runner.js";
|
|
13
|
+
import { loadCredentials, saveCredentials } from "../services/credentials.js";
|
|
14
|
+
import * as nodeFsSync from "node:fs";
|
|
15
|
+
/**
|
|
16
|
+
* Creates a lightweight container for SDK usage.
|
|
17
|
+
* Uses real file system and command runner.
|
|
18
|
+
* No prompts needed (non-interactive).
|
|
19
|
+
* Minimal logger (silent by default).
|
|
20
|
+
*/
|
|
21
|
+
export function createSdkContainer(options) {
|
|
22
|
+
const cwd = options?.cwd ?? process.cwd();
|
|
23
|
+
const homeDir = options?.homeDir ?? os.homedir();
|
|
24
|
+
const variables = options?.variables ?? process.env;
|
|
25
|
+
const verbose = options?.verbose ?? false;
|
|
26
|
+
const environment = createCliEnvironment({
|
|
27
|
+
cwd,
|
|
28
|
+
homeDir,
|
|
29
|
+
platform: process.platform,
|
|
30
|
+
variables
|
|
31
|
+
});
|
|
32
|
+
// Silent logger for SDK - only emits if verbose
|
|
33
|
+
const silentEmitter = verbose ? undefined : () => { };
|
|
34
|
+
const loggerFactory = createLoggerFactory(silentEmitter);
|
|
35
|
+
// Create error logger with sync fs
|
|
36
|
+
const errorLogger = new ErrorLogger({
|
|
37
|
+
fs: nodeFsSync,
|
|
38
|
+
logDir: environment.logDir,
|
|
39
|
+
logToStderr: false
|
|
40
|
+
});
|
|
41
|
+
loggerFactory.setErrorLogger(errorLogger);
|
|
42
|
+
// Create async file system adapter
|
|
43
|
+
const asyncFs = {
|
|
44
|
+
readFile: ((path, encoding) => {
|
|
45
|
+
if (encoding) {
|
|
46
|
+
return fs.readFile(path, encoding);
|
|
47
|
+
}
|
|
48
|
+
return fs.readFile(path);
|
|
49
|
+
}),
|
|
50
|
+
writeFile: (path, data, opts) => fs.writeFile(path, data, opts),
|
|
51
|
+
mkdir: (path, opts) => fs.mkdir(path, opts).then(() => { }),
|
|
52
|
+
stat: (path) => fs.stat(path),
|
|
53
|
+
rm: (path, opts) => fs.rm(path, opts),
|
|
54
|
+
unlink: (path) => fs.unlink(path),
|
|
55
|
+
readdir: (path) => fs.readdir(path),
|
|
56
|
+
copyFile: (src, dest) => fs.copyFile(src, dest),
|
|
57
|
+
chmod: (path, mode) => fs.chmod(path, mode)
|
|
58
|
+
};
|
|
59
|
+
const contextFactory = createCommandContextFactory({ fs: asyncFs });
|
|
60
|
+
// No-op prompts for SDK (non-interactive)
|
|
61
|
+
const noopPrompts = async () => {
|
|
62
|
+
throw new Error("SDK does not support interactive prompts");
|
|
63
|
+
};
|
|
64
|
+
const promptLibrary = createPromptLibrary();
|
|
65
|
+
const optionResolvers = createOptionResolvers({
|
|
66
|
+
prompts: noopPrompts,
|
|
67
|
+
promptLibrary,
|
|
68
|
+
apiKeyStore: {
|
|
69
|
+
read: () => loadCredentials({
|
|
70
|
+
fs: asyncFs,
|
|
71
|
+
filePath: environment.credentialsPath
|
|
72
|
+
}),
|
|
73
|
+
write: (value) => saveCredentials({
|
|
74
|
+
fs: asyncFs,
|
|
75
|
+
filePath: environment.credentialsPath,
|
|
76
|
+
apiKey: value
|
|
77
|
+
})
|
|
78
|
+
}
|
|
79
|
+
});
|
|
80
|
+
const registry = createServiceRegistry();
|
|
81
|
+
const providers = getDefaultProviders().filter((adapter) => !adapter.disabled);
|
|
82
|
+
for (const adapter of providers) {
|
|
83
|
+
registry.register(adapter);
|
|
84
|
+
}
|
|
85
|
+
const baseRunner = createDefaultCommandRunner();
|
|
86
|
+
// Create container with wrapped runner
|
|
87
|
+
let container = null;
|
|
88
|
+
const wrappedRunner = createPoeCodeCommandRunner({
|
|
89
|
+
getContainer: () => container,
|
|
90
|
+
baseRunner
|
|
91
|
+
});
|
|
92
|
+
// HTTP client using global fetch
|
|
93
|
+
const httpClient = async (url, init) => {
|
|
94
|
+
const response = await globalThis.fetch(url, init);
|
|
95
|
+
return {
|
|
96
|
+
ok: response.ok,
|
|
97
|
+
status: response.status,
|
|
98
|
+
json: () => response.json()
|
|
99
|
+
};
|
|
100
|
+
};
|
|
101
|
+
container = {
|
|
102
|
+
env: environment,
|
|
103
|
+
fs: asyncFs,
|
|
104
|
+
prompts: noopPrompts,
|
|
105
|
+
promptLibrary,
|
|
106
|
+
loggerFactory,
|
|
107
|
+
errorLogger,
|
|
108
|
+
options: optionResolvers,
|
|
109
|
+
contextFactory,
|
|
110
|
+
registry,
|
|
111
|
+
httpClient,
|
|
112
|
+
commandRunner: wrappedRunner,
|
|
113
|
+
providers,
|
|
114
|
+
dependencies: {
|
|
115
|
+
fs: asyncFs,
|
|
116
|
+
prompts: noopPrompts,
|
|
117
|
+
env: {
|
|
118
|
+
cwd,
|
|
119
|
+
homeDir,
|
|
120
|
+
platform: process.platform,
|
|
121
|
+
variables
|
|
122
|
+
}
|
|
123
|
+
}
|
|
124
|
+
};
|
|
125
|
+
return container;
|
|
126
|
+
}
|
|
127
|
+
//# sourceMappingURL=container.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"container.js","sourceRoot":"","sources":["../../src/sdk/container.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACvC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAG9B,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAC7D,OAAO,EAAE,qBAAqB,EAAE,MAAM,4BAA4B,CAAC;AACnE,OAAO,EAAE,2BAA2B,EAAE,MAAM,mBAAmB,CAAC;AAChE,OAAO,EAAE,mBAAmB,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,qBAAqB,EAAE,MAAM,mBAAmB,CAAC;AAC1D,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AACvD,OAAO,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AACrD,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,0BAA0B,EAAE,MAAM,mCAAmC,CAAC;AAC/E,OAAO,EACL,eAAe,EACf,eAAe,EAChB,MAAM,4BAA4B,CAAC;AACpC,OAAO,KAAK,UAAU,MAAM,SAAS,CAAC;AAatC;;;;;GAKG;AACH,MAAM,UAAU,kBAAkB,CAAC,OAA6B;IAC9D,MAAM,GAAG,GAAG,OAAO,EAAE,GAAG,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;IAC1C,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC,OAAO,EAAE,CAAC;IACjD,MAAM,SAAS,GAAG,OAAO,EAAE,SAAS,IAAI,OAAO,CAAC,GAAG,CAAC;IACpD,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,KAAK,CAAC;IAE1C,MAAM,WAAW,GAAG,oBAAoB,CAAC;QACvC,GAAG;QACH,OAAO;QACP,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,SAAS;KACV,CAAC,CAAC;IAEH,gDAAgD;IAChD,MAAM,aAAa,GAAG,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;IACrD,MAAM,aAAa,GAAG,mBAAmB,CAAC,aAAa,CAAC,CAAC;IAEzD,mCAAmC;IACnC,MAAM,WAAW,GAAG,IAAI,WAAW,CAAC;QAClC,EAAE,EAAE,UAAiB;QACrB,MAAM,EAAE,WAAW,CAAC,MAAM;QAC1B,WAAW,EAAE,KAAK;KACnB,CAAC,CAAC;IACH,aAAa,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;IAE1C,mCAAmC;IACnC,MAAM,OAAO,GAAe;QAC1B,QAAQ,EAAE,CAAC,CAAC,IAAY,EAAE,QAAyB,EAAE,EAAE;YACrD,IAAI,QAAQ,EAAE,CAAC;gBACb,OAAO,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;YACrC,CAAC;YACD,OAAO,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC3B,CAAC,CAA2B;QAC5B,SAAS,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE,CAC9B,EAAE,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC;QAChC,KAAK,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC;QAC1D,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;QAC7B,EAAE,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC;QACrC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC;QACjC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC;QACnC,QAAQ,EAAE,CAAC,GAAG,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,GAAG,EAAE,IAAI,CAAC;QAC/C,KAAK,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC;KAC5C,CAAC;IAEF,MAAM,cAAc,GAAG,2BAA2B,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;IAEpE,0CAA0C;IAC1C,MAAM,WAAW,GAAG,KAAK,IAAI,EAAE;QAC7B,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;IAC9D,CAAC,CAAC;IAEF,MAAM,aAAa,GAAG,mBAAmB,EAAE,CAAC;IAE5C,MAAM,eAAe,GAAG,qBAAqB,CAAC;QAC5C,OAAO,EAAE,WAAW;QACpB,aAAa;QACb,WAAW,EAAE;YACX,IAAI,EAAE,GAAG,EAAE,CACT,eAAe,CAAC;gBACd,EAAE,EAAE,OAAO;gBACX,QAAQ,EAAE,WAAW,CAAC,eAAe;aACtC,CAAC;YACJ,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE,CACf,eAAe,CAAC;gBACd,EAAE,EAAE,OAAO;gBACX,QAAQ,EAAE,WAAW,CAAC,eAAe;gBACrC,MAAM,EAAE,KAAK;aACd,CAAC;SACL;KACF,CAAC,CAAC;IAEH,MAAM,QAAQ,GAAG,qBAAqB,EAAE,CAAC;IAEzC,MAAM,SAAS,GAAG,mBAAmB,EAAE,CAAC,MAAM,CAC5C,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,QAAQ,CAC/B,CAAC;IACF,KAAK,MAAM,OAAO,IAAI,SAAS,EAAE,CAAC;QAChC,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IAC7B,CAAC;IAED,MAAM,UAAU,GAAG,0BAA0B,EAAE,CAAC;IAEhD,uCAAuC;IACvC,IAAI,SAAS,GAAiB,IAA+B,CAAC;IAC9D,MAAM,aAAa,GAAG,0BAA0B,CAAC;QAC/C,YAAY,EAAE,GAAG,EAAE,CAAC,SAAS;QAC7B,UAAU;KACX,CAAC,CAAC;IAEH,iCAAiC;IACjC,MAAM,UAAU,GAAG,KAAK,EAAE,GAAW,EAAE,IAAkB,EAAE,EAAE;QAC3D,MAAM,QAAQ,GAAG,MAAM,UAAU,CAAC,KAAK,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QACnD,OAAO;YACL,EAAE,EAAE,QAAQ,CAAC,EAAE;YACf,MAAM,EAAE,QAAQ,CAAC,MAAM;YACvB,IAAI,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE;SAC5B,CAAC;IACJ,CAAC,CAAC;IAEF,SAAS,GAAG;QACV,GAAG,EAAE,WAAW;QAChB,EAAE,EAAE,OAAO;QACX,OAAO,EAAE,WAAW;QACpB,aAAa;QACb,aAAa;QACb,WAAW;QACX,OAAO,EAAE,eAAe;QACxB,cAAc;QACd,QAAQ;QACR,UAAU;QACV,aAAa,EAAE,aAAa;QAC5B,SAAS;QACT,YAAY,EAAE;YACZ,EAAE,EAAE,OAAO;YACX,OAAO,EAAE,WAAW;YACpB,GAAG,EAAE;gBACH,GAAG;gBACH,OAAO;gBACP,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,SAAS;aACV;SACF;KACF,CAAC;IAEF,OAAO,SAAS,CAAC;AACnB,CAAC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Reads the Poe API key with the following priority:
|
|
3
|
+
* 1. `POE_API_KEY` environment variable (if set)
|
|
4
|
+
* 2. Credentials file (`~/.poe-code/credentials.json`)
|
|
5
|
+
*
|
|
6
|
+
* @returns The API key
|
|
7
|
+
* @throws Error if no credentials found
|
|
8
|
+
*/
|
|
9
|
+
export declare function getPoeApiKey(): Promise<string>;
|
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
import * as fs from "node:fs/promises";
|
|
2
|
+
import * as os from "node:os";
|
|
3
|
+
import * as path from "node:path";
|
|
4
|
+
const CREDENTIALS_RELATIVE_PATH = ".poe-code/credentials.json";
|
|
5
|
+
/**
|
|
6
|
+
* Reads the Poe API key with the following priority:
|
|
7
|
+
* 1. `POE_API_KEY` environment variable (if set)
|
|
8
|
+
* 2. Credentials file (`~/.poe-code/credentials.json`)
|
|
9
|
+
*
|
|
10
|
+
* @returns The API key
|
|
11
|
+
* @throws Error if no credentials found
|
|
12
|
+
*/
|
|
13
|
+
export async function getPoeApiKey() {
|
|
14
|
+
const envKey = process.env.POE_API_KEY;
|
|
15
|
+
if (typeof envKey === "string" && envKey.trim().length > 0) {
|
|
16
|
+
return envKey.trim();
|
|
17
|
+
}
|
|
18
|
+
const homeDir = os.homedir();
|
|
19
|
+
const credentialsPath = path.join(homeDir, CREDENTIALS_RELATIVE_PATH);
|
|
20
|
+
try {
|
|
21
|
+
const content = await fs.readFile(credentialsPath, "utf8");
|
|
22
|
+
const parsed = JSON.parse(content);
|
|
23
|
+
if (typeof parsed === "object" &&
|
|
24
|
+
parsed !== null &&
|
|
25
|
+
typeof parsed.apiKey === "string" &&
|
|
26
|
+
parsed.apiKey.length > 0) {
|
|
27
|
+
return parsed.apiKey;
|
|
28
|
+
}
|
|
29
|
+
}
|
|
30
|
+
catch {
|
|
31
|
+
// File doesn't exist or is invalid
|
|
32
|
+
}
|
|
33
|
+
throw new Error("No API key found. Set POE_API_KEY or run 'poe-code login'.");
|
|
34
|
+
}
|
|
35
|
+
//# sourceMappingURL=credentials.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"credentials.js","sourceRoot":"","sources":["../../src/sdk/credentials.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACvC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAElC,MAAM,yBAAyB,GAAG,4BAA4B,CAAC;AAE/D;;;;;;;GAOG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY;IAChC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC;IACvC,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3D,OAAO,MAAM,CAAC,IAAI,EAAE,CAAC;IACvB,CAAC;IAED,MAAM,OAAO,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC;IAC7B,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,yBAAyB,CAAC,CAAC;IAEtE,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,eAAe,EAAE,MAAM,CAAC,CAAC;QAC3D,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QACnC,IACE,OAAO,MAAM,KAAK,QAAQ;YAC1B,MAAM,KAAK,IAAI;YACf,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ;YACjC,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EACxB,CAAC;YACD,OAAO,MAAM,CAAC,MAAM,CAAC;QACvB,CAAC;IACH,CAAC;IAAC,MAAM,CAAC;QACP,mCAAmC;IACrC,CAAC;IAED,MAAM,IAAI,KAAK,CACb,4DAA4D,CAC7D,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
import type { GenerateOptions, GenerateResult, MediaGenerateOptions, MediaGenerateResult } from "./types.js";
|
|
2
|
+
export declare function generate(prompt: string, options?: GenerateOptions): Promise<GenerateResult>;
|
|
3
|
+
export declare function generateImage(prompt: string, options?: MediaGenerateOptions): Promise<MediaGenerateResult>;
|
|
4
|
+
export declare function generateVideo(prompt: string, options?: MediaGenerateOptions): Promise<MediaGenerateResult>;
|
|
5
|
+
export declare function generateAudio(prompt: string, options?: MediaGenerateOptions): Promise<MediaGenerateResult>;
|
|
@@ -0,0 +1,97 @@
|
|
|
1
|
+
import { DEFAULT_TEXT_MODEL, DEFAULT_IMAGE_BOT, DEFAULT_VIDEO_BOT, DEFAULT_AUDIO_BOT } from "../cli/constants.js";
|
|
2
|
+
import { getPoeApiKey } from "./credentials.js";
|
|
3
|
+
import { getGlobalClient, setGlobalClient } from "../services/client-instance.js";
|
|
4
|
+
import { createPoeClient } from "../services/llm-client.js";
|
|
5
|
+
const MODEL_ENV_KEYS = {
|
|
6
|
+
text: "POE_TEXT_MODEL",
|
|
7
|
+
image: "POE_IMAGE_MODEL",
|
|
8
|
+
video: "POE_VIDEO_MODEL",
|
|
9
|
+
audio: "POE_AUDIO_MODEL"
|
|
10
|
+
};
|
|
11
|
+
const DEFAULT_MODELS = {
|
|
12
|
+
text: DEFAULT_TEXT_MODEL,
|
|
13
|
+
image: DEFAULT_IMAGE_BOT,
|
|
14
|
+
video: DEFAULT_VIDEO_BOT,
|
|
15
|
+
audio: DEFAULT_AUDIO_BOT
|
|
16
|
+
};
|
|
17
|
+
export async function generate(prompt, options = {}) {
|
|
18
|
+
if (!prompt) {
|
|
19
|
+
throw new Error("Prompt is required");
|
|
20
|
+
}
|
|
21
|
+
const client = await resolveClient();
|
|
22
|
+
const model = resolveSdkModel("text", options);
|
|
23
|
+
const response = await client.text({
|
|
24
|
+
model,
|
|
25
|
+
prompt,
|
|
26
|
+
params: options.params
|
|
27
|
+
});
|
|
28
|
+
if (!response.content) {
|
|
29
|
+
throw new Error("No response from LLM");
|
|
30
|
+
}
|
|
31
|
+
return { content: response.content };
|
|
32
|
+
}
|
|
33
|
+
export async function generateImage(prompt, options = {}) {
|
|
34
|
+
return generateMedia("image", prompt, options);
|
|
35
|
+
}
|
|
36
|
+
export async function generateVideo(prompt, options = {}) {
|
|
37
|
+
return generateMedia("video", prompt, options);
|
|
38
|
+
}
|
|
39
|
+
export async function generateAudio(prompt, options = {}) {
|
|
40
|
+
return generateMedia("audio", prompt, options);
|
|
41
|
+
}
|
|
42
|
+
async function generateMedia(type, prompt, options) {
|
|
43
|
+
if (!prompt) {
|
|
44
|
+
throw new Error("Prompt is required");
|
|
45
|
+
}
|
|
46
|
+
const client = await resolveClient();
|
|
47
|
+
const model = resolveSdkModel(type, options);
|
|
48
|
+
const response = await client.media(type, {
|
|
49
|
+
model,
|
|
50
|
+
prompt,
|
|
51
|
+
params: options.params
|
|
52
|
+
});
|
|
53
|
+
if (!response.url) {
|
|
54
|
+
throw new Error(`No ${type} URL returned`);
|
|
55
|
+
}
|
|
56
|
+
return { url: response.url, mimeType: response.mimeType };
|
|
57
|
+
}
|
|
58
|
+
async function resolveClient() {
|
|
59
|
+
try {
|
|
60
|
+
return getGlobalClient();
|
|
61
|
+
}
|
|
62
|
+
catch {
|
|
63
|
+
const apiKey = await getPoeApiKey();
|
|
64
|
+
const baseUrl = normalizeBaseUrl(process.env.POE_API_BASE_URL);
|
|
65
|
+
const client = createPoeClient({
|
|
66
|
+
apiKey,
|
|
67
|
+
baseUrl
|
|
68
|
+
});
|
|
69
|
+
setGlobalClient(client);
|
|
70
|
+
return client;
|
|
71
|
+
}
|
|
72
|
+
}
|
|
73
|
+
function resolveSdkModel(type, options) {
|
|
74
|
+
if (options.model) {
|
|
75
|
+
return options.model;
|
|
76
|
+
}
|
|
77
|
+
const envKey = MODEL_ENV_KEYS[type];
|
|
78
|
+
const envModel = normalizeEnvModel(process.env[envKey]);
|
|
79
|
+
if (envModel) {
|
|
80
|
+
return envModel;
|
|
81
|
+
}
|
|
82
|
+
return DEFAULT_MODELS[type];
|
|
83
|
+
}
|
|
84
|
+
function normalizeEnvModel(value) {
|
|
85
|
+
if (typeof value !== "string") {
|
|
86
|
+
return undefined;
|
|
87
|
+
}
|
|
88
|
+
const trimmed = value.trim();
|
|
89
|
+
return trimmed.length > 0 ? trimmed : undefined;
|
|
90
|
+
}
|
|
91
|
+
function normalizeBaseUrl(value) {
|
|
92
|
+
if (typeof value === "string" && value.trim().length > 0) {
|
|
93
|
+
return value.trim();
|
|
94
|
+
}
|
|
95
|
+
return "https://api.poe.com/v1";
|
|
96
|
+
}
|
|
97
|
+
//# sourceMappingURL=generate.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"generate.js","sourceRoot":"","sources":["../../src/sdk/generate.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,kBAAkB,EAClB,iBAAiB,EACjB,iBAAiB,EACjB,iBAAiB,EAClB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AAClF,OAAO,EAAE,eAAe,EAAE,MAAM,2BAA2B,CAAC;AAa5D,MAAM,cAAc,GAAiC;IACnD,IAAI,EAAE,gBAAgB;IACtB,KAAK,EAAE,iBAAiB;IACxB,KAAK,EAAE,iBAAiB;IACxB,KAAK,EAAE,iBAAiB;CACzB,CAAC;AAEF,MAAM,cAAc,GAAiC;IACnD,IAAI,EAAE,kBAAkB;IACxB,KAAK,EAAE,iBAAiB;IACxB,KAAK,EAAE,iBAAiB;IACxB,KAAK,EAAE,iBAAiB;CACzB,CAAC;AAEF,MAAM,CAAC,KAAK,UAAU,QAAQ,CAC5B,MAAc,EACd,UAA2B,EAAE;IAE7B,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;IACxC,CAAC;IACD,MAAM,MAAM,GAAG,MAAM,aAAa,EAAE,CAAC;IACrC,MAAM,KAAK,GAAG,eAAe,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC/C,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC;QACjC,KAAK;QACL,MAAM;QACN,MAAM,EAAE,OAAO,CAAC,MAAM;KACvB,CAAC,CAAC;IACH,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;QACtB,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;IAC1C,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,QAAQ,CAAC,OAAO,EAAE,CAAC;AACvC,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,MAAc,EACd,UAAgC,EAAE;IAElC,OAAO,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;AACjD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,MAAc,EACd,UAAgC,EAAE;IAElC,OAAO,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;AACjD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,MAAc,EACd,UAAgC,EAAE;IAElC,OAAO,aAAa,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;AACjD,CAAC;AAED,KAAK,UAAU,aAAa,CAC1B,IAAe,EACf,MAAc,EACd,OAA6B;IAE7B,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;IACxC,CAAC;IACD,MAAM,MAAM,GAAG,MAAM,aAAa,EAAE,CAAC;IACrC,MAAM,KAAK,GAAG,eAAe,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAC7C,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE;QACxC,KAAK;QACL,MAAM;QACN,MAAM,EAAE,OAAO,CAAC,MAAM;KACvB,CAAC,CAAC;IACH,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,CAAC;QAClB,MAAM,IAAI,KAAK,CAAC,MAAM,IAAI,eAAe,CAAC,CAAC;IAC7C,CAAC;IACD,OAAO,EAAE,GAAG,EAAE,QAAQ,CAAC,GAAG,EAAE,QAAQ,EAAE,QAAQ,CAAC,QAAQ,EAAE,CAAC;AAC5D,CAAC;AAED,KAAK,UAAU,aAAa;IAC1B,IAAI,CAAC;QACH,OAAO,eAAe,EAAE,CAAC;IAC3B,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,MAAM,GAAG,MAAM,YAAY,EAAE,CAAC;QACpC,MAAM,OAAO,GAAG,gBAAgB,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG,eAAe,CAAC;YAC7B,MAAM;YACN,OAAO;SACR,CAAC,CAAC;QACH,eAAe,CAAC,MAAM,CAAC,CAAC;QACxB,OAAO,MAAM,CAAC;IAChB,CAAC;AACH,CAAC;AAED,SAAS,eAAe,CACtB,IAAkB,EAClB,OAAwB;IAExB,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QAClB,OAAO,OAAO,CAAC,KAAK,CAAC;IACvB,CAAC;IACD,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;IACpC,MAAM,QAAQ,GAAG,iBAAiB,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;IACxD,IAAI,QAAQ,EAAE,CAAC;QACb,OAAO,QAAQ,CAAC;IAClB,CAAC;IACD,OAAO,cAAc,CAAC,IAAI,CAAC,CAAC;AAC9B,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAyB;IAClD,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,SAAS,CAAC;IACnB,CAAC;IACD,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;IAC7B,OAAO,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC;AAClD,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAyB;IACjD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACzD,OAAO,KAAK,CAAC,IAAI,EAAE,CAAC;IACtB,CAAC;IACD,OAAO,wBAAwB,CAAC;AAClC,CAAC"}
|
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
import type { CliContainer } from "../cli/container.js";
|
|
2
|
+
import type { SpawnResult } from "./types.js";
|
|
3
|
+
export interface SpawnCoreOptions {
|
|
4
|
+
/** The prompt to send to the provider */
|
|
5
|
+
prompt: string;
|
|
6
|
+
/** Working directory for the service CLI */
|
|
7
|
+
cwd?: string;
|
|
8
|
+
/** Model identifier override */
|
|
9
|
+
model?: string;
|
|
10
|
+
/** Additional arguments forwarded to the CLI */
|
|
11
|
+
args?: string[];
|
|
12
|
+
/** Whether prompt was read from stdin */
|
|
13
|
+
useStdin?: boolean;
|
|
14
|
+
}
|
|
15
|
+
export interface SpawnCoreFlags {
|
|
16
|
+
dryRun: boolean;
|
|
17
|
+
verbose: boolean;
|
|
18
|
+
}
|
|
19
|
+
/**
|
|
20
|
+
* Core spawn implementation used by both SDK and CLI.
|
|
21
|
+
* Accepts an existing container to avoid creating a new one.
|
|
22
|
+
*/
|
|
23
|
+
export declare function spawnCore(container: CliContainer, service: string, options: SpawnCoreOptions, flags?: SpawnCoreFlags): Promise<SpawnResult>;
|