openclaw/extensions/kilocode/provider-catalog.ts

35 lines
1.2 KiB
TypeScript

import { type ModelProviderConfig } from "openclaw/plugin-sdk/provider-model-shared";
import {
discoverKilocodeModels,
KILOCODE_BASE_URL as LOCAL_KILOCODE_BASE_URL,
KILOCODE_DEFAULT_CONTEXT_WINDOW as LOCAL_KILOCODE_DEFAULT_CONTEXT_WINDOW,
KILOCODE_DEFAULT_COST as LOCAL_KILOCODE_DEFAULT_COST,
KILOCODE_DEFAULT_MAX_TOKENS as LOCAL_KILOCODE_DEFAULT_MAX_TOKENS,
KILOCODE_MODEL_CATALOG as LOCAL_KILOCODE_MODEL_CATALOG,
} from "./provider-models.js";
export function buildKilocodeProvider(): ModelProviderConfig {
return {
baseUrl: LOCAL_KILOCODE_BASE_URL,
api: "openai-completions",
models: LOCAL_KILOCODE_MODEL_CATALOG.map((model) => ({
id: model.id,
name: model.name,
reasoning: model.reasoning,
input: model.input,
cost: LOCAL_KILOCODE_DEFAULT_COST,
contextWindow: model.contextWindow ?? LOCAL_KILOCODE_DEFAULT_CONTEXT_WINDOW,
maxTokens: model.maxTokens ?? LOCAL_KILOCODE_DEFAULT_MAX_TOKENS,
})),
};
}
export async function buildKilocodeProviderWithDiscovery(): Promise<ModelProviderConfig> {
const models = await discoverKilocodeModels();
return {
baseUrl: LOCAL_KILOCODE_BASE_URL,
api: "openai-completions",
models,
};
}