78 lines
2.3 KiB
TypeScript
78 lines
2.3 KiB
TypeScript
import type { ModelDefinitionConfig } from "../config/types.js";
|
|
import {
|
|
buildVolcModelDefinition,
|
|
VOLC_MODEL_GLM_4_7,
|
|
VOLC_MODEL_KIMI_K2_5,
|
|
VOLC_SHARED_CODING_MODEL_CATALOG,
|
|
} from "./volc-models.shared.js";
|
|
|
|
export const DOUBAO_BASE_URL = "https://ark.cn-beijing.volces.com/api/v3";
|
|
export const DOUBAO_CODING_BASE_URL = "https://ark.cn-beijing.volces.com/api/coding/v3";
|
|
export const DOUBAO_DEFAULT_MODEL_ID = "doubao-seed-1-8-251228";
|
|
export const DOUBAO_CODING_DEFAULT_MODEL_ID = "ark-code-latest";
|
|
export const DOUBAO_DEFAULT_MODEL_REF = `volcengine/${DOUBAO_DEFAULT_MODEL_ID}`;
|
|
|
|
// Volcano Engine Doubao pricing (approximate, adjust based on actual pricing)
|
|
export const DOUBAO_DEFAULT_COST = {
|
|
input: 0.0001, // ¥0.0001 per 1K tokens
|
|
output: 0.0002, // ¥0.0002 per 1K tokens
|
|
cacheRead: 0,
|
|
cacheWrite: 0,
|
|
};
|
|
|
|
/**
|
|
* Complete catalog of Volcano Engine models.
|
|
*
|
|
* Volcano Engine provides access to models
|
|
* through the API. Authentication requires a Volcano Engine API Key.
|
|
*/
|
|
export const DOUBAO_MODEL_CATALOG = [
|
|
{
|
|
id: "doubao-seed-code-preview-251028",
|
|
name: "doubao-seed-code-preview-251028",
|
|
reasoning: false,
|
|
input: ["text", "image"] as const,
|
|
contextWindow: 256000,
|
|
maxTokens: 4096,
|
|
},
|
|
{
|
|
id: "doubao-seed-1-8-251228",
|
|
name: "Doubao Seed 1.8",
|
|
reasoning: false,
|
|
input: ["text", "image"] as const,
|
|
contextWindow: 256000,
|
|
maxTokens: 4096,
|
|
},
|
|
VOLC_MODEL_KIMI_K2_5,
|
|
VOLC_MODEL_GLM_4_7,
|
|
{
|
|
id: "deepseek-v3-2-251201",
|
|
name: "DeepSeek V3.2",
|
|
reasoning: false,
|
|
input: ["text", "image"] as const,
|
|
contextWindow: 128000,
|
|
maxTokens: 4096,
|
|
},
|
|
] as const;
|
|
|
|
export type DoubaoCatalogEntry = (typeof DOUBAO_MODEL_CATALOG)[number];
|
|
export type DoubaoCodingCatalogEntry = (typeof DOUBAO_CODING_MODEL_CATALOG)[number];
|
|
|
|
export function buildDoubaoModelDefinition(
|
|
entry: DoubaoCatalogEntry | DoubaoCodingCatalogEntry,
|
|
): ModelDefinitionConfig {
|
|
return buildVolcModelDefinition(entry, DOUBAO_DEFAULT_COST);
|
|
}
|
|
|
|
export const DOUBAO_CODING_MODEL_CATALOG = [
|
|
...VOLC_SHARED_CODING_MODEL_CATALOG,
|
|
{
|
|
id: "doubao-seed-code-preview-251028",
|
|
name: "Doubao Seed Code Preview",
|
|
reasoning: false,
|
|
input: ["text"] as const,
|
|
contextWindow: 256000,
|
|
maxTokens: 4096,
|
|
},
|
|
] as const;
|