This commit is contained in:
王春跃 2026-01-30 19:32:28 +08:00 committed by GitHub
commit c0ab92fe2c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
10 changed files with 292 additions and 0 deletions

View File

@ -86,6 +86,21 @@ const OLLAMA_DEFAULT_COST = {
cacheWrite: 0, cacheWrite: 0,
}; };
const DEEPSEEK_API_BASE_URL = "https://api.deepseek.com";
const DEEPSEEK_CHAT_MODEL_ID = "deepseek-chat";
const DEEPSEEK_REASONER_MODEL_ID = "deepseek-reasoner";
const DEEPSEEK_DEFAULT_CONTEXT_WINDOW = 128000;
const DEEPSEEK_CHAT_MAX_TOKENS = 8192;
const DEEPSEEK_REASONER_MAX_TOKENS = 64000;
// DeepSeek pricing (per 1M tokens ) - Updated 2026-01-29
// https://api-docs.deepseek.com/quick_start/pricing
const DEEPSEEK_API_COST = {
input: 0.28, // Input (Cache Miss ): $0.28 per 1M tokens
output: 0.42, // Output: $0.42 per 1M tokens
cacheRead: 0.028, // Input (Cache Hit): $0.028 per 1M tokens
cacheWrite: 0.28, // Treat same as regular input for write
};
interface OllamaModel { interface OllamaModel {
name: string; name: string;
modified_at: string; modified_at: string;
@ -388,6 +403,33 @@ async function buildOllamaProvider(): Promise<ProviderConfig> {
}; };
} }
function buildDeepSeekProvider(): ProviderConfig {
return {
baseUrl: DEEPSEEK_API_BASE_URL,
api: "openai-completions",
models: [
{
id: DEEPSEEK_CHAT_MODEL_ID,
name: "DeepSeek Chat (V3.2)",
reasoning: false,
input: ["text"],
cost: DEEPSEEK_API_COST,
contextWindow: DEEPSEEK_DEFAULT_CONTEXT_WINDOW,
maxTokens: DEEPSEEK_CHAT_MAX_TOKENS,
},
{
id: DEEPSEEK_REASONER_MODEL_ID,
name: "DeepSeek Reasoner (V3.2)",
reasoning: true,
input: ["text"],
cost: DEEPSEEK_API_COST,
contextWindow: DEEPSEEK_DEFAULT_CONTEXT_WINDOW,
maxTokens: DEEPSEEK_REASONER_MAX_TOKENS,
},
],
};
}
export async function resolveImplicitProviders(params: { export async function resolveImplicitProviders(params: {
agentDir: string; agentDir: string;
}): Promise<ModelsConfig["providers"]> { }): Promise<ModelsConfig["providers"]> {
@ -454,6 +496,14 @@ export async function resolveImplicitProviders(params: {
providers.ollama = { ...(await buildOllamaProvider()), apiKey: ollamaKey }; providers.ollama = { ...(await buildOllamaProvider()), apiKey: ollamaKey };
} }
// DeepSeek provider
const deepseekKey =
resolveEnvApiKeyVarName("deepseek") ??
resolveApiKeyFromProfiles({ provider: "deepseek", store: authStore });
if (deepseekKey) {
providers.deepseek = { ...buildDeepSeekProvider(), apiKey: deepseekKey };
}
return providers; return providers;
} }

View File

@ -28,4 +28,32 @@ describe("ensureSkillsWatcher", () => {
expect(ignored.some((re) => re.test("/tmp/workspace/skills/.git/config"))).toBe(true); expect(ignored.some((re) => re.test("/tmp/workspace/skills/.git/config"))).toBe(true);
expect(ignored.some((re) => re.test("/tmp/.hidden/skills/index.md"))).toBe(false); expect(ignored.some((re) => re.test("/tmp/.hidden/skills/index.md"))).toBe(false);
}); });
it("ignores Python virtual environments and cache directories", async () => {
const mod = await import("./refresh.js");
const ignored = mod.DEFAULT_SKILLS_WATCH_IGNORED;
// Python virtual environments
expect(
ignored.some((re) => re.test("/tmp/workspace/skills/.venv/lib/python3.9/site-packages/")),
).toBe(true);
expect(
ignored.some((re) => re.test("/tmp/workspace/skills/venv/lib/python3.9/site-packages/")),
).toBe(true);
expect(ignored.some((re) => re.test("/tmp/workspace/skills/env/bin/python"))).toBe(true);
// Python cache directories
expect(
ignored.some((re) => re.test("/tmp/workspace/skills/__pycache__/module.cpython-39.pyc")),
).toBe(true);
expect(ignored.some((re) => re.test("/tmp/workspace/skills/.pytest_cache/"))).toBe(true);
expect(ignored.some((re) => re.test("/tmp/workspace/skills/.tox/py39/lib/"))).toBe(true);
expect(ignored.some((re) => re.test("/tmp/workspace/skills/package.egg-info/PKG-INFO"))).toBe(
true,
);
// Should not ignore files that just contain these names
expect(ignored.some((re) => re.test("/tmp/workspace/skills/my_venv_config.json"))).toBe(false);
expect(ignored.some((re) => re.test("/tmp/workspace/skills/pytest.ini"))).toBe(false);
});
}); });

View File

@ -31,6 +31,15 @@ export const DEFAULT_SKILLS_WATCH_IGNORED: RegExp[] = [
/(^|[\\/])\.git([\\/]|$)/, /(^|[\\/])\.git([\\/]|$)/,
/(^|[\\/])node_modules([\\/]|$)/, /(^|[\\/])node_modules([\\/]|$)/,
/(^|[\\/])dist([\\/]|$)/, /(^|[\\/])dist([\\/]|$)/,
// Python virtual environments
/(^|[\\/])\.venv([\\/]|$)/,
/(^|[\\/])venv([\\/]|$)/,
/(^|[\\/])env([\\/]|$)/,
// Python cache directories
/(^|[\\/])__pycache__([\\/]|$)/,
/(^|[\\/])\.pytest_cache([\\/]|$)/,
/(^|[\\/])\.tox([\\/]|$)/,
/(^|[\\/])[^/\\]*\.egg-info([\\/]|$)/,
]; ];
function bumpVersion(current: number): number { function bumpVersion(current: number): number {

View File

@ -21,6 +21,7 @@ export type AuthChoiceGroupId =
| "minimax" | "minimax"
| "synthetic" | "synthetic"
| "venice" | "venice"
| "deepseek"
| "qwen"; | "qwen";
export type AuthChoiceGroup = { export type AuthChoiceGroup = {
@ -60,6 +61,12 @@ const AUTH_CHOICE_GROUP_DEFS: {
hint: "OAuth", hint: "OAuth",
choices: ["qwen-portal"], choices: ["qwen-portal"],
}, },
{
value: "deepseek",
label: "DeepSeek",
hint: "V3.2 (chat + reasoning)",
choices: ["deepseek-api-key"],
},
{ {
value: "synthetic", value: "synthetic",
label: "Synthetic", label: "Synthetic",
@ -194,6 +201,11 @@ export function buildAuthChoiceOptions(params: {
label: "MiniMax M2.1 Lightning", label: "MiniMax M2.1 Lightning",
hint: "Faster, higher output cost", hint: "Faster, higher output cost",
}); });
options.push({
value: "deepseek-api-key",
label: "DeepSeek API key",
hint: "V3.2 chat and reasoning models",
});
if (params.includeSkip) { if (params.includeSkip) {
options.push({ value: "skip", label: "Skip for now" }); options.push({ value: "skip", label: "Skip for now" });
} }

View File

@ -13,6 +13,8 @@ import {
} from "./google-gemini-model-default.js"; } from "./google-gemini-model-default.js";
import { import {
applyAuthProfileConfig, applyAuthProfileConfig,
applyDeepSeekConfig,
applyDeepSeekProviderConfig,
applyKimiCodeConfig, applyKimiCodeConfig,
applyKimiCodeProviderConfig, applyKimiCodeProviderConfig,
applyMoonshotConfig, applyMoonshotConfig,
@ -30,6 +32,7 @@ import {
applyXiaomiConfig, applyXiaomiConfig,
applyXiaomiProviderConfig, applyXiaomiProviderConfig,
applyZaiConfig, applyZaiConfig,
DEEPSEEK_DEFAULT_MODEL_REF,
KIMI_CODE_MODEL_REF, KIMI_CODE_MODEL_REF,
MOONSHOT_DEFAULT_MODEL_REF, MOONSHOT_DEFAULT_MODEL_REF,
OPENROUTER_DEFAULT_MODEL_REF, OPENROUTER_DEFAULT_MODEL_REF,
@ -37,6 +40,7 @@ import {
VENICE_DEFAULT_MODEL_REF, VENICE_DEFAULT_MODEL_REF,
VERCEL_AI_GATEWAY_DEFAULT_MODEL_REF, VERCEL_AI_GATEWAY_DEFAULT_MODEL_REF,
XIAOMI_DEFAULT_MODEL_REF, XIAOMI_DEFAULT_MODEL_REF,
setDeepSeekApiKey,
setGeminiApiKey, setGeminiApiKey,
setKimiCodeApiKey, setKimiCodeApiKey,
setMoonshotApiKey, setMoonshotApiKey,
@ -89,6 +93,8 @@ export async function applyAuthChoiceApiProviders(
authChoice = "synthetic-api-key"; authChoice = "synthetic-api-key";
} else if (params.opts.tokenProvider === "venice") { } else if (params.opts.tokenProvider === "venice") {
authChoice = "venice-api-key"; authChoice = "venice-api-key";
} else if (params.opts.tokenProvider === "deepseek") {
authChoice = "deepseek-api-key";
} else if (params.opts.tokenProvider === "opencode") { } else if (params.opts.tokenProvider === "opencode") {
authChoice = "opencode-zen"; authChoice = "opencode-zen";
} }
@ -633,5 +639,52 @@ export async function applyAuthChoiceApiProviders(
return { config: nextConfig, agentModelOverride }; return { config: nextConfig, agentModelOverride };
} }
if (authChoice === "deepseek-api-key") {
let hasCredential = false;
if (!hasCredential && params.opts?.token && params.opts?.tokenProvider === "deepseek") {
await setDeepSeekApiKey(normalizeApiKeyInput(params.opts.token), params.agentDir);
hasCredential = true;
}
const envKey = resolveEnvApiKey("deepseek");
if (envKey) {
const useExisting = await params.prompter.confirm({
message: `Use existing DEEPSEEK_API_KEY (${envKey.source}, ${formatApiKeyPreview(envKey.apiKey)})?`,
initialValue: true,
});
if (useExisting) {
await setDeepSeekApiKey(envKey.apiKey, params.agentDir);
hasCredential = true;
}
}
if (!hasCredential) {
const key = await params.prompter.text({
message: "Enter DeepSeek API key",
validate: validateApiKeyInput,
});
await setDeepSeekApiKey(normalizeApiKeyInput(String(key)), params.agentDir);
}
nextConfig = applyAuthProfileConfig(nextConfig, {
profileId: "deepseek:default",
provider: "deepseek",
mode: "api_key",
});
{
const applied = await applyDefaultModelChoice({
config: nextConfig,
setDefaultModel: params.setDefaultModel,
defaultModel: DEEPSEEK_DEFAULT_MODEL_REF,
applyDefaultConfig: applyDeepSeekConfig,
applyProviderConfig: applyDeepSeekProviderConfig,
noteAgentModel,
prompter: params.prompter,
});
nextConfig = applied.config;
agentModelOverride = applied.agentModelOverride ?? agentModelOverride;
}
return { config: nextConfig, agentModelOverride };
}
return null; return null;
} }

View File

@ -19,8 +19,13 @@ import {
ZAI_DEFAULT_MODEL_REF, ZAI_DEFAULT_MODEL_REF,
} from "./onboard-auth.credentials.js"; } from "./onboard-auth.credentials.js";
import { import {
buildDeepSeekModelDefinition,
buildKimiCodeModelDefinition, buildKimiCodeModelDefinition,
buildMoonshotModelDefinition, buildMoonshotModelDefinition,
DEEPSEEK_BASE_URL,
DEEPSEEK_CHAT_MODEL_ID,
DEEPSEEK_DEFAULT_MODEL_REF,
DEEPSEEK_REASONER_MODEL_ID,
KIMI_CODE_BASE_URL, KIMI_CODE_BASE_URL,
KIMI_CODE_MODEL_ID, KIMI_CODE_MODEL_ID,
KIMI_CODE_MODEL_REF, KIMI_CODE_MODEL_REF,
@ -532,3 +537,75 @@ export function applyAuthProfileConfig(
}, },
}; };
} }
export function applyDeepSeekProviderConfig(cfg: MoltbotConfig): MoltbotConfig {
const models = { ...cfg.agents?.defaults?.models };
models[DEEPSEEK_DEFAULT_MODEL_REF] = {
...models[DEEPSEEK_DEFAULT_MODEL_REF],
alias: models[DEEPSEEK_DEFAULT_MODEL_REF]?.alias ?? "DeepSeek",
};
const providers = { ...cfg.models?.providers };
const existingProvider = providers.deepseek;
const existingModels = Array.isArray(existingProvider?.models) ? existingProvider.models : [];
const deepseekModels = [
buildDeepSeekModelDefinition(DEEPSEEK_CHAT_MODEL_ID),
buildDeepSeekModelDefinition(DEEPSEEK_REASONER_MODEL_ID),
];
const mergedModels = [
...existingModels,
...deepseekModels.filter(
(model) => !existingModels.some((existing) => existing.id === model.id),
),
];
const { apiKey: existingApiKey, ...existingProviderRest } = (existingProvider ?? {}) as Record<
string,
unknown
> as { apiKey?: string };
const resolvedApiKey = typeof existingApiKey === "string" ? existingApiKey : undefined;
const normalizedApiKey = resolvedApiKey?.trim();
providers.deepseek = {
...existingProviderRest,
baseUrl: DEEPSEEK_BASE_URL,
api: "openai-completions",
...(normalizedApiKey ? { apiKey: normalizedApiKey } : {}),
models: mergedModels.length > 0 ? mergedModels : deepseekModels,
};
return {
...cfg,
agents: {
...cfg.agents,
defaults: {
...cfg.agents?.defaults,
models,
},
},
models: {
mode: cfg.models?.mode ?? "merge",
providers,
},
};
}
export function applyDeepSeekConfig(cfg: MoltbotConfig): MoltbotConfig {
const next = applyDeepSeekProviderConfig(cfg);
const existingModel = next.agents?.defaults?.model;
return {
...next,
agents: {
...next.agents,
defaults: {
...next.agents?.defaults,
model: {
...(existingModel && "fallbacks" in (existingModel as Record<string, unknown>)
? {
fallbacks: (existingModel as { fallbacks?: string[] }).fallbacks,
}
: undefined),
primary: DEEPSEEK_DEFAULT_MODEL_REF,
},
},
},
};
}

View File

@ -177,3 +177,16 @@ export async function setOpencodeZenApiKey(key: string, agentDir?: string) {
agentDir: resolveAuthAgentDir(agentDir), agentDir: resolveAuthAgentDir(agentDir),
}); });
} }
export async function setDeepSeekApiKey(key: string, agentDir?: string) {
// Write to resolved agent dir so gateway finds credentials on startup.
upsertAuthProfile({
profileId: "deepseek:default",
credential: {
type: "api_key",
provider: "deepseek",
key,
},
agentDir: resolveAuthAgentDir(agentDir),
});
}

View File

@ -116,3 +116,42 @@ export function buildKimiCodeModelDefinition(): ModelDefinitionConfig {
compat: KIMI_CODE_COMPAT, compat: KIMI_CODE_COMPAT,
}; };
} }
export const DEEPSEEK_BASE_URL = "https://api.deepseek.com";
export const DEEPSEEK_CHAT_MODEL_ID = "deepseek-chat";
export const DEEPSEEK_REASONER_MODEL_ID = "deepseek-reasoner";
export const DEEPSEEK_DEFAULT_MODEL_REF = `deepseek/${DEEPSEEK_CHAT_MODEL_ID}`;
export const DEEPSEEK_REASONER_MODEL_REF = `deepseek/${DEEPSEEK_REASONER_MODEL_ID}`;
// DeepSeek pricing (per 1M tokens )
const DEEPSEEK_DEFAULT_COST = {
input: 0.28, // Input (Cache Miss)
output: 0.42, // Output
cacheRead: 0.028, // Input (Cache Hit)
cacheWrite: 0.28, // Treat same as regular input for write
};
export function buildDeepSeekModelDefinition(modelId: string): ModelDefinitionConfig {
if (modelId === DEEPSEEK_CHAT_MODEL_ID) {
return {
id: DEEPSEEK_CHAT_MODEL_ID,
name: "DeepSeek Chat (V3.2)",
reasoning: false,
input: ["text"],
cost: DEEPSEEK_DEFAULT_COST,
contextWindow: 128000,
maxTokens: 8192,
};
} else if (modelId === DEEPSEEK_REASONER_MODEL_ID) {
return {
id: DEEPSEEK_REASONER_MODEL_ID,
name: "DeepSeek Reasoner (V3.2)",
reasoning: true,
input: ["text"],
cost: DEEPSEEK_DEFAULT_COST,
contextWindow: 128000,
maxTokens: 64000,
};
}
throw new Error(`Unknown DeepSeek model: ${modelId}`);
}

View File

@ -5,6 +5,8 @@ export {
export { VENICE_DEFAULT_MODEL_ID, VENICE_DEFAULT_MODEL_REF } from "../agents/venice-models.js"; export { VENICE_DEFAULT_MODEL_ID, VENICE_DEFAULT_MODEL_REF } from "../agents/venice-models.js";
export { export {
applyAuthProfileConfig, applyAuthProfileConfig,
applyDeepSeekConfig,
applyDeepSeekProviderConfig,
applyKimiCodeConfig, applyKimiCodeConfig,
applyKimiCodeProviderConfig, applyKimiCodeProviderConfig,
applyMoonshotConfig, applyMoonshotConfig,
@ -37,6 +39,7 @@ export {
export { export {
OPENROUTER_DEFAULT_MODEL_REF, OPENROUTER_DEFAULT_MODEL_REF,
setAnthropicApiKey, setAnthropicApiKey,
setDeepSeekApiKey,
setGeminiApiKey, setGeminiApiKey,
setKimiCodeApiKey, setKimiCodeApiKey,
setMinimaxApiKey, setMinimaxApiKey,
@ -54,11 +57,17 @@ export {
ZAI_DEFAULT_MODEL_REF, ZAI_DEFAULT_MODEL_REF,
} from "./onboard-auth.credentials.js"; } from "./onboard-auth.credentials.js";
export { export {
buildDeepSeekModelDefinition,
buildKimiCodeModelDefinition, buildKimiCodeModelDefinition,
buildMinimaxApiModelDefinition, buildMinimaxApiModelDefinition,
buildMinimaxModelDefinition, buildMinimaxModelDefinition,
buildMoonshotModelDefinition, buildMoonshotModelDefinition,
DEFAULT_MINIMAX_BASE_URL, DEFAULT_MINIMAX_BASE_URL,
DEEPSEEK_BASE_URL,
DEEPSEEK_CHAT_MODEL_ID,
DEEPSEEK_DEFAULT_MODEL_REF,
DEEPSEEK_REASONER_MODEL_ID,
DEEPSEEK_REASONER_MODEL_REF,
KIMI_CODE_BASE_URL, KIMI_CODE_BASE_URL,
KIMI_CODE_MODEL_ID, KIMI_CODE_MODEL_ID,
KIMI_CODE_MODEL_REF, KIMI_CODE_MODEL_REF,

View File

@ -17,6 +17,7 @@ export type AuthChoice =
| "kimi-code-api-key" | "kimi-code-api-key"
| "synthetic-api-key" | "synthetic-api-key"
| "venice-api-key" | "venice-api-key"
| "deepseek-api-key"
| "codex-cli" | "codex-cli"
| "apiKey" | "apiKey"
| "gemini-api-key" | "gemini-api-key"
@ -72,6 +73,7 @@ export type OnboardOptions = {
minimaxApiKey?: string; minimaxApiKey?: string;
syntheticApiKey?: string; syntheticApiKey?: string;
veniceApiKey?: string; veniceApiKey?: string;
deepseekApiKey?: string;
opencodeZenApiKey?: string; opencodeZenApiKey?: string;
gatewayPort?: number; gatewayPort?: number;
gatewayBind?: GatewayBind; gatewayBind?: GatewayBind;