aiwrapper
Version:
A Universal AI Wrapper for JavaScript & TypeScript
49 lines (48 loc) • 2.06 kB
JavaScript
var __defProp = Object.defineProperty;
var __getOwnPropSymbols = Object.getOwnPropertySymbols;
var __hasOwnProp = Object.prototype.hasOwnProperty;
var __propIsEnum = Object.prototype.propertyIsEnumerable;
var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
var __spreadValues = (a, b) => {
for (var prop in b || (b = {}))
if (__hasOwnProp.call(b, prop))
__defNormalProp(a, prop, b[prop]);
if (__getOwnPropSymbols)
for (var prop of __getOwnPropSymbols(b)) {
if (__propIsEnum.call(b, prop))
__defNormalProp(a, prop, b[prop]);
}
return a;
};
import { OpenAIChatCompletionsLang } from "../openai/openai-chat-completions-lang.js";
class KimiLang extends OpenAIChatCompletionsLang {
constructor(options) {
const modelName = options.model || "kimi-k2.5";
super({
apiKey: options.apiKey,
model: modelName,
systemPrompt: options.systemPrompt || "",
maxTokens: options.maxTokens,
baseURL: "https://api.moonshot.ai/v1",
bodyProperties: __spreadValues(__spreadValues(__spreadValues(__spreadValues({}, options.bodyProperties || {}), options.thinking ? { thinking: options.thinking } : {}), options.promptCacheKey ? { prompt_cache_key: options.promptCacheKey } : {}), options.safetyIdentifier ? { safety_identifier: options.safetyIdentifier } : {}),
defaultOptions: options.defaultOptions
});
}
transformBody(body) {
const transformedBody = super.transformBody(body);
if (this._config.maxTokens === void 0) {
delete transformedBody.max_tokens;
delete transformedBody.max_completion_tokens;
return transformedBody;
}
if (typeof transformedBody.max_tokens === "number" && transformedBody.max_completion_tokens === void 0) {
transformedBody.max_completion_tokens = transformedBody.max_tokens;
}
delete transformedBody.max_tokens;
return transformedBody;
}
}
export {
KimiLang
};
//# sourceMappingURL=kimi-lang.js.map