@ax-llm/ax 9.0.0

package/build/module/src/ai/cohere/info.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"info.js","sourceRoot":"","sources":["../../../../../src/ai/cohere/info.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAE3D,MAAM,CAAC,MAAM,eAAe,GAAoB;IAC9C;QACE,IAAI,EAAE,WAAW,CAAC,YAAY;QAC9B,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,EAAE;KAC7B;IACD;QACE,IAAI,EAAE,WAAW,CAAC,QAAQ;QAC1B,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,WAAW,CAAC,OAAO;QACzB,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,WAAW,CAAC,YAAY;QAC9B,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,gBAAgB,CAAC,oBAAoB;QAC3C,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,gBAAgB,CAAC,eAAe;QACtC,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,gBAAgB,CAAC,oBAAoB;QAC3C,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,gBAAgB,CAAC,yBAAyB;QAChD,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;CACF,CAAC"}

package/build/module/src/ai/cohere/types.d.ts

@@ -0,0 +1,85 @@
+import type { TextModelConfig } from '../types.js';
+/**
+ * Cohere: Models for text generation
+ * @export
+ */
+export declare enum CohereModel {
+    CommandRPlus = "command-r-plus",
+    CommandR = "command-r",
+    Command = "command",
+    CommandLight = "command-light"
+}
+/**
+ * Cohere: Models for use in embeddings
+ * @export
+ */
+export declare enum CohereEmbedModel {
+    EmbedEnglishV30 = "embed-english-v3.0",
+    EmbedEnglishLightV30 = "embed-english-light-v3.0",
+    EmbedMultiLingualV30 = "embed-multilingual-v3.0",
+    EmbedMultiLingualLightV30 = "embed-multilingual-light-v3.0"
+}
+/**
+ * Cohere: Model options for text generation
+ * @export
+ */
+export type CohereConfig = TextModelConfig & {
+    model: CohereModel;
+    embedModel: CohereEmbedModel;
+};
+export type CohereChatRequest = {
+    message: string;
+    preamble?: string;
+    chat_history: {
+        role: 'CHATBOT' | 'SYSTEM' | 'USER';
+        message: string;
+    }[];
+    model: CohereModel | string;
+    max_tokens?: number;
+    temperature?: number;
+    k?: number;
+    p?: number;
+    frequency_penalty?: number;
+    presence_penalty?: number;
+    end_sequences?: readonly string[];
+    stop_sequences?: string[];
+    tools?: {
+        name: string;
+        description: string;
+        parameter_definitions: Record<string, {
+            description: string;
+            type: string;
+            required: boolean;
+        }>;
+    }[];
+    tool_results?: {
+        call: {
+            name: string;
+            parameters: object;
+        };
+        outputs: object[];
+    }[];
+};
+export type CohereChatResponse = {
+    generation_id: string;
+    text: string;
+    finish_reason: 'COMPLETE' | 'ERROR' | 'ERROR_TOXIC' | 'ERROR_LIMIT' | 'USER_CANCEL' | 'MAX_TOKENS';
+    tool_calls: {
+        name: string;
+        parameters: object;
+    }[];
+};
+export type CohereChatResponseDelta = CohereChatResponse & {
+    event_type: 'stream-start' | 'text-generation' | 'tool-calls-generation' | 'stream-end';
+};
+export type CohereEmbedRequest = {
+    texts: readonly string[];
+    model: CohereModel | string;
+    truncate: string;
+};
+export type CohereEmbedResponse = {
+    id: string;
+    texts: string[];
+    model: string;
+    embeddings: number[][];
+};

package/build/module/src/ai/cohere/types.js

@@ -0,0 +1,23 @@
+/**
+ * Cohere: Models for text generation
+ * @export
+ */
+export var CohereModel;
+(function (CohereModel) {
+    CohereModel["CommandRPlus"] = "command-r-plus";
+    CohereModel["CommandR"] = "command-r";
+    CohereModel["Command"] = "command";
+    CohereModel["CommandLight"] = "command-light";
+})(CohereModel || (CohereModel = {}));
+/**
+ * Cohere: Models for use in embeddings
+ * @export
+ */
+export var CohereEmbedModel;
+(function (CohereEmbedModel) {
+    CohereEmbedModel["EmbedEnglishV30"] = "embed-english-v3.0";
+    CohereEmbedModel["EmbedEnglishLightV30"] = "embed-english-light-v3.0";
+    CohereEmbedModel["EmbedMultiLingualV30"] = "embed-multilingual-v3.0";
+    CohereEmbedModel["EmbedMultiLingualLightV30"] = "embed-multilingual-light-v3.0";
+})(CohereEmbedModel || (CohereEmbedModel = {}));
+//# sourceMappingURL=types.js.map

package/build/module/src/ai/cohere/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../src/ai/cohere/types.ts"],"names":[],"mappings":"AAEA;;;GAGG;AACH,MAAM,CAAN,IAAY,WAKX;AALD,WAAY,WAAW;IACrB,8CAA+B,CAAA;IAC/B,qCAAsB,CAAA;IACtB,kCAAmB,CAAA;IACnB,6CAA8B,CAAA;AAChC,CAAC,EALW,WAAW,KAAX,WAAW,QAKtB;AAED;;;GAGG;AACH,MAAM,CAAN,IAAY,gBAKX;AALD,WAAY,gBAAgB;IAC1B,0DAAsC,CAAA;IACtC,qEAAiD,CAAA;IACjD,oEAAgD,CAAA;IAChD,+EAA2D,CAAA;AAC7D,CAAC,EALW,gBAAgB,KAAhB,gBAAgB,QAK3B"}

package/build/module/src/ai/deepseek/api.d.ts

@@ -0,0 +1,27 @@
+import type { AIServiceOptions } from '../../text/types.js';
+import { OpenAI } from '../openai/api.js';
+import type { OpenAIConfig } from '../openai/types.js';
+type DeepSeekConfig = OpenAIConfig;
+/**
+ * DeepSeek: Default Model options for text generation
+ * @export
+ */
+export declare const DeepSeekDefaultConfig: () => DeepSeekConfig;
+/**
+ * DeepSeek: Default Model options for text generation
+ * @export
+ */
+export declare const DeepSeekCodeConfig: () => DeepSeekConfig;
+export interface DeepSeekArgs {
+    apiKey: string;
+    config: Readonly<DeepSeekConfig>;
+    options?: Readonly<AIServiceOptions>;
+}
+/**
+ * DeepSeek: AI Service
+ * @export
+ */
+export declare class DeepSeek extends OpenAI {
+    constructor({ apiKey, config, options }: Readonly<DeepSeekArgs>);
+}
+export {};

package/build/module/src/ai/deepseek/api.js

@@ -0,0 +1,38 @@
+import { BaseAIDefaultConfig, BaseAIDefaultCreativeConfig } from '../base.js';
+import { OpenAI } from '../openai/api.js';
+import { DeepSeekModel } from './types.js';
+/**
+ * DeepSeek: Default Model options for text generation
+ * @export
+ */
+export const DeepSeekDefaultConfig = () => structuredClone({
+    model: DeepSeekModel.DeepSeekChat,
+    ...BaseAIDefaultConfig()
+});
+/**
+ * DeepSeek: Default Model options for text generation
+ * @export
+ */
+export const DeepSeekCodeConfig = () => structuredClone({
+    model: DeepSeekModel.DeepSeekCoder,
+    ...BaseAIDefaultCreativeConfig()
+});
+/**
+ * DeepSeek: AI Service
+ * @export
+ */
+export class DeepSeek extends OpenAI {
+    constructor({ apiKey, config = DeepSeekDefaultConfig(), options }) {
+        if (!apiKey || apiKey === '') {
+            throw new Error('DeepSeek API key not set');
+        }
+        super({
+            apiKey,
+            config,
+            options,
+            apiURL: 'https://api.deepseek.com'
+        });
+        super.setName('DeepSeek');
+    }
+}
+//# sourceMappingURL=api.js.map

package/build/module/src/ai/deepseek/api.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../../src/ai/deepseek/api.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,2BAA2B,EAAE,MAAM,YAAY,CAAC;AAC9E,OAAO,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAC;AAG1C,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAI3C;;;GAGG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,GAAmB,EAAE,CACxD,eAAe,CAAC;IACd,KAAK,EAAE,aAAa,CAAC,YAAY;IACjC,GAAG,mBAAmB,EAAE;CACzB,CAAC,CAAC;AAEL;;;GAGG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,GAAmB,EAAE,CACrD,eAAe,CAAC;IACd,KAAK,EAAE,aAAa,CAAC,aAAa;IAClC,GAAG,2BAA2B,EAAE;CACjC,CAAC,CAAC;AAQL;;;GAGG;AACH,MAAM,OAAO,QAAS,SAAQ,MAAM;IAClC,YAAY,EACV,MAAM,EACN,MAAM,GAAG,qBAAqB,EAAE,EAChC,OAAO,EACgB;QACvB,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAC9C,CAAC;QAED,KAAK,CAAC;YACJ,MAAM;YACN,MAAM;YACN,OAAO;YACP,MAAM,EAAE,0BAA0B;SACnC,CAAC,CAAC;QAEH,KAAK,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAC5B,CAAC;CACF"}

package/build/module/src/ai/deepseek/index.d.ts

@@ -0,0 +1,2 @@
+export * from './api.js';
+export * from './types.js';

package/build/module/src/ai/deepseek/index.js

@@ -0,0 +1,3 @@
+export * from './api.js';
+export * from './types.js';
+//# sourceMappingURL=index.js.map

package/build/module/src/ai/deepseek/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/ai/deepseek/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAC;AACzB,cAAc,YAAY,CAAC"}

package/build/module/src/ai/deepseek/info.d.ts

@@ -0,0 +1,2 @@
+import type { TextModelInfo } from '../types.js';
+export declare const textModelInfos: TextModelInfo[];

package/build/module/src/ai/deepseek/info.js

@@ -0,0 +1,16 @@
+import { DeepSeekModel } from './types.js';
+export const textModelInfos = [
+    {
+        name: DeepSeekModel.DeepSeekChat,
+        currency: 'USD',
+        promptTokenCostPer1M: 0.14,
+        completionTokenCostPer1M: 0.28
+    },
+    {
+        name: DeepSeekModel.DeepSeekCoder,
+        currency: 'USD',
+        promptTokenCostPer1M: 0.14,
+        completionTokenCostPer1M: 0.28
+    }
+];
+//# sourceMappingURL=info.js.map

package/build/module/src/ai/deepseek/info.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"info.js","sourceRoot":"","sources":["../../../../../src/ai/deepseek/info.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAE3C,MAAM,CAAC,MAAM,cAAc,GAAoB;IAC7C;QACE,IAAI,EAAE,aAAa,CAAC,YAAY;QAChC,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,IAAI;QAC1B,wBAAwB,EAAE,IAAI;KAC/B;IACD;QACE,IAAI,EAAE,aAAa,CAAC,aAAa;QACjC,QAAQ,EAAE,KAAK;QACf,oBAAoB,EAAE,IAAI;QAC1B,wBAAwB,EAAE,IAAI;KAC/B;CACF,CAAC"}

package/build/module/src/ai/deepseek/types.d.ts

@@ -0,0 +1,8 @@
+/**
+ * DeepSeek: Models for text generation
+ * @export
+ */
+export declare enum DeepSeekModel {
+    DeepSeekChat = "deepseek-chat",
+    DeepSeekCoder = "deepseek-coder"
+}

package/build/module/src/ai/deepseek/types.js

@@ -0,0 +1,10 @@
+/**
+ * DeepSeek: Models for text generation
+ * @export
+ */
+export var DeepSeekModel;
+(function (DeepSeekModel) {
+    DeepSeekModel["DeepSeekChat"] = "deepseek-chat";
+    DeepSeekModel["DeepSeekCoder"] = "deepseek-coder";
+})(DeepSeekModel || (DeepSeekModel = {}));
+//# sourceMappingURL=types.js.map

package/build/module/src/ai/deepseek/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../../src/ai/deepseek/types.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,MAAM,CAAN,IAAY,aAGX;AAHD,WAAY,aAAa;IACvB,+CAA8B,CAAA;IAC9B,iDAAgC,CAAA;AAClC,CAAC,EAHW,aAAa,KAAb,aAAa,QAGxB"}

package/build/module/src/ai/google-gemini/api.d.ts

@@ -0,0 +1,34 @@
+import type { AIServiceOptions } from '../../text/types.js';
+import type { AITextChatRequest, AITextEmbedRequest } from '../../types/index.js';
+import type { API } from '../../util/apicall.js';
+import { BaseAI } from '../base.js';
+import type { EmbedResponse, TextModelConfig, TextResponse } from '../types.js';
+import { type GoogleGeminiBatchEmbedRequest, type GoogleGeminiBatchEmbedResponse, type GoogleGeminiChatRequest, type GoogleGeminiChatResponse, type GoogleGeminiChatResponseDelta, type GoogleGeminiConfig } from './types.js';
+/**
+ * GoogleGemini: Default Model options for text generation
+ * @export
+ */
+export declare const GoogleGeminiDefaultConfig: () => GoogleGeminiConfig;
+export declare const GoogleGeminiDefaultCreativeConfig: () => GoogleGeminiConfig;
+export interface GoogleGeminiArgs {
+    apiKey: string;
+    projectId?: string;
+    region?: string;
+    config: Readonly<GoogleGeminiConfig>;
+    options?: Readonly<AIServiceOptions>;
+}
+/**
+ * GoogleGemini: AI Service
+ * @export
+ */
+export declare class GoogleGemini extends BaseAI<GoogleGeminiChatRequest, GoogleGeminiBatchEmbedRequest, GoogleGeminiChatResponse, GoogleGeminiChatResponseDelta, GoogleGeminiBatchEmbedResponse> {
+    private config;
+    private apiKey;
+    constructor({ apiKey, projectId, region, config, options }: Readonly<GoogleGeminiArgs>);
+    getModelConfig(): TextModelConfig;
+    generateChatReq: (req: Readonly<AITextChatRequest>) => [API, GoogleGeminiChatRequest];
+    generateEmbedReq: (req: Readonly<AITextEmbedRequest>) => [API, GoogleGeminiBatchEmbedRequest];
+    generateChatResp: (resp: Readonly<GoogleGeminiChatResponse>) => TextResponse;
+    generateChatStreamResp: (resp: Readonly<GoogleGeminiChatResponseDelta>) => TextResponse;
+    generateEmbedResp: (resp: Readonly<GoogleGeminiBatchEmbedResponse>) => EmbedResponse;
+}

package/build/module/src/ai/google-gemini/api.js

@@ -0,0 +1,268 @@
+import { BaseAI, BaseAIDefaultConfig, BaseAIDefaultCreativeConfig } from '../base.js';
+import { modelInfoGoogleGemini } from './info.js';
+import { GoogleGeminiEmbedModels, GoogleGeminiModel, GoogleGeminiSafetyCategory, GoogleGeminiSafetyThreshold } from './types.js';
+const safetySettings = [
+    {
+        category: GoogleGeminiSafetyCategory.HarmCategoryHarassment,
+        threshold: GoogleGeminiSafetyThreshold.BlockNone
+    },
+    {
+        category: GoogleGeminiSafetyCategory.HarmCategoryHateSpeech,
+        threshold: GoogleGeminiSafetyThreshold.BlockNone
+    },
+    {
+        category: GoogleGeminiSafetyCategory.HarmCategorySexuallyExplicit,
+        threshold: GoogleGeminiSafetyThreshold.BlockNone
+    },
+    {
+        category: GoogleGeminiSafetyCategory.HarmCategoryDangerousContent,
+        threshold: GoogleGeminiSafetyThreshold.BlockNone
+    }
+];
+/**
+ * GoogleGemini: Default Model options for text generation
+ * @export
+ */
+export const GoogleGeminiDefaultConfig = () => structuredClone({
+    model: GoogleGeminiModel.Gemini15Pro,
+    embedModel: GoogleGeminiEmbedModels.Embedding001,
+    safetySettings,
+    ...BaseAIDefaultConfig()
+});
+export const GoogleGeminiDefaultCreativeConfig = () => structuredClone({
+    model: GoogleGeminiModel.Gemini15Flash,
+    embedModel: GoogleGeminiEmbedModels.Embedding001,
+    safetySettings,
+    ...BaseAIDefaultCreativeConfig()
+});
+/**
+ * GoogleGemini: AI Service
+ * @export
+ */
+export class GoogleGemini extends BaseAI {
+    config;
+    apiKey;
+    constructor({ apiKey, projectId, region, config = GoogleGeminiDefaultConfig(), options }) {
+        if (!apiKey || apiKey === '') {
+            throw new Error('GoogleGemini AI API key not set');
+        }
+        let apiURL = 'https://generativelanguage.googleapis.com/v1beta';
+        if (projectId && region) {
+            apiURL = `POST https://${region}-aiplatform.googleapis.com/v1/projects/${projectId}/locations/{REGION}/publishers/google/`;
+        }
+        super({
+            name: 'GoogleGeminiAI',
+            apiURL,
+            headers: {},
+            modelInfo: modelInfoGoogleGemini,
+            models: { model: config.model, embedModel: config.embedModel },
+            options,
+            supportFor: { functions: true, streaming: true }
+        });
+        this.config = config;
+        this.apiKey = apiKey;
+    }
+    getModelConfig() {
+        const { config } = this;
+        return {
+            maxTokens: config.maxTokens,
+            temperature: config.temperature,
+            topP: config.topP,
+            topK: config.topK
+        };
+    }
+    generateChatReq = (req) => {
+        const model = req.modelInfo?.name ?? this.config.model;
+        const stream = req.modelConfig?.stream ?? this.config.stream;
+        if (!req.chatPrompt || req.chatPrompt.length === 0) {
+            throw new Error('Chat prompt is empty');
+        }
+        const apiConfig = {
+            name: stream
+                ? `/models/${model}:streamGenerateContent?alt=sse&key=${this.apiKey}`
+                : `/models/${model}:generateContent?key=${this.apiKey}`
+        };
+        const systemPrompts = req.chatPrompt
+            .filter((p) => p.role === 'system')
+            .map((p) => p.content);
+        const systemInstruction = systemPrompts.length > 0
+            ? {
+                role: 'user',
+                parts: [{ text: systemPrompts.join(' ') }]
+            }
+            : undefined;
+        const contents = req.chatPrompt
+            .filter((p) => p.role !== 'system')
+            .map(({ role, ...prompt }, i) => {
+            if (role === 'user') {
+                if (!prompt.content) {
+                    throw new Error(`Chat prompt content is empty (index: ${i})`);
+                }
+                return {
+                    role: 'user',
+                    parts: [{ text: prompt.content }]
+                };
+            }
+            if (role === 'assistant') {
+                const text = prompt.content ? [{ text: prompt.content }] : [];
+                let functionCalls = [];
+                if ('functionCalls' in prompt) {
+                    functionCalls =
+                        prompt.functionCalls?.map((f) => {
+                            const args = typeof f.function.arguments === 'string'
+                                ? JSON.parse(f.function.arguments)
+                                : f.function.arguments;
+                            return {
+                                functionCall: {
+                                    name: f.function.name,
+                                    args: args
+                                }
+                            };
+                        }) ?? [];
+                }
+                return {
+                    role: 'model',
+                    parts: text ? text : functionCalls
+                };
+            }
+            if (role === 'function') {
+                if ('functionId' in prompt) {
+                    return {
+                        role: 'function',
+                        parts: [
+                            {
+                                functionResponse: {
+                                    name: prompt.functionId,
+                                    response: { result: prompt.content }
+                                }
+                            }
+                        ]
+                    };
+                }
+                throw new Error(`Chat prompt functionId is empty (index: ${i})`);
+            }
+            throw new Error(`Chat prompt role not supported: ${role} (index: ${i})`);
+        });
+        const tools = req.functions
+            ? [
+                {
+                    functionDeclarations: req.functions ?? []
+                }
+            ]
+            : undefined;
+        let tool_config;
+        if (req.functionCall) {
+            if (req.functionCall === 'none') {
+                tool_config = { function_calling_config: { mode: 'NONE' } };
+            }
+            else if (req.functionCall === 'auto') {
+                tool_config = { function_calling_config: { mode: 'AUTO' } };
+            }
+            else if (req.functionCall === 'required') {
+                tool_config = {
+                    function_calling_config: { mode: 'ANY' }
+                };
+            }
+            else {
+                tool_config = {
+                    function_calling_config: {
+                        mode: 'ANY',
+                        allowed_function_names: [req.functionCall.function.name]
+                    }
+                };
+            }
+        }
+        const generationConfig = {
+            maxOutputTokens: req.modelConfig?.maxTokens ?? this.config.maxTokens,
+            temperature: req.modelConfig?.temperature ?? this.config.temperature,
+            topP: req.modelConfig?.topP ?? this.config.topP,
+            topK: req.modelConfig?.topK ?? this.config.topK,
+            candidateCount: 1,
+            stopSequences: req.modelConfig?.stopSequences ?? this.config.stopSequences
+        };
+        const safetySettings = this.config.safetySettings;
+        const reqValue = {
+            contents,
+            tools,
+            tool_config,
+            systemInstruction,
+            generationConfig,
+            safetySettings
+        };
+        return [apiConfig, reqValue];
+    };
+    generateEmbedReq = (req) => {
+        const model = req.embedModelInfo?.name ?? this.config.embedModel;
+        if (!model) {
+            throw new Error('Embed model not set');
+        }
+        if (!req.texts || req.texts.length === 0) {
+            throw new Error('Embed texts is empty');
+        }
+        const apiConfig = {
+            name: `/models/${model}:batchEmbedText?key=${this.apiKey}`
+        };
+        const reqValue = {
+            requests: req.texts.map((text) => ({ model, text }))
+        };
+        return [apiConfig, reqValue];
+    };
+    generateChatResp = (resp) => {
+        const results = resp.candidates.map((candidate) => {
+            const result = { content: null };
+            switch (candidate.finishReason) {
+                case 'MAX_TOKENS':
+                    result.finishReason = 'length';
+                    break;
+                case 'STOP':
+                    result.finishReason = 'stop';
+                    break;
+                case 'SAFETY':
+                    throw new Error('Finish reason: SAFETY');
+                case 'RECITATION':
+                    throw new Error('Finish reason: RECITATION');
+            }
+            for (const part of candidate.content.parts) {
+                if ('text' in part) {
+                    result.content = part.text;
+                    continue;
+                }
+                if ('functionCall' in part) {
+                    result.functionCalls = [
+                        {
+                            id: part.functionCall.name,
+                            type: 'function',
+                            function: {
+                                name: part.functionCall.name,
+                                arguments: part.functionCall.args
+                            }
+                        }
+                    ];
+                }
+            }
+            return result;
+        });
+        let modelUsage;
+        if (resp.usageMetadata) {
+            modelUsage = {
+                totalTokens: resp.usageMetadata.totalTokenCount,
+                promptTokens: resp.usageMetadata.promptTokenCount,
+                completionTokens: resp.usageMetadata.candidatesTokenCount
+            };
+        }
+        return {
+            results,
+            modelUsage
+        };
+    };
+    generateChatStreamResp = (resp) => {
+        return this.generateChatResp(resp);
+    };
+    generateEmbedResp = (resp) => {
+        const embeddings = resp.embeddings.map((embedding) => embedding.value);
+        return {
+            embeddings
+        };
+    };
+}
+//# sourceMappingURL=api.js.map

package/build/module/src/ai/google-gemini/api.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"api.js","sourceRoot":"","sources":["../../../../../src/ai/google-gemini/api.ts"],"names":[],"mappings":"AAMA,OAAO,EACL,MAAM,EACN,mBAAmB,EACnB,2BAA2B,EAC5B,MAAM,YAAY,CAAC;AASpB,OAAO,EAAE,qBAAqB,EAAE,MAAM,WAAW,CAAC;AAClD,OAAO,EAOL,uBAAuB,EACvB,iBAAiB,EACjB,0BAA0B,EAE1B,2BAA2B,EAC5B,MAAM,YAAY,CAAC;AAEpB,MAAM,cAAc,GAA+B;IACjD;QACE,QAAQ,EAAE,0BAA0B,CAAC,sBAAsB;QAC3D,SAAS,EAAE,2BAA2B,CAAC,SAAS;KACjD;IACD;QACE,QAAQ,EAAE,0BAA0B,CAAC,sBAAsB;QAC3D,SAAS,EAAE,2BAA2B,CAAC,SAAS;KACjD;IACD;QACE,QAAQ,EAAE,0BAA0B,CAAC,4BAA4B;QACjE,SAAS,EAAE,2BAA2B,CAAC,SAAS;KACjD;IACD;QACE,QAAQ,EAAE,0BAA0B,CAAC,4BAA4B;QACjE,SAAS,EAAE,2BAA2B,CAAC,SAAS;KACjD;CACF,CAAC;AAEF;;;GAGG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,GAAuB,EAAE,CAChE,eAAe,CAAC;IACd,KAAK,EAAE,iBAAiB,CAAC,WAAW;IACpC,UAAU,EAAE,uBAAuB,CAAC,YAAY;IAChD,cAAc;IACd,GAAG,mBAAmB,EAAE;CACzB,CAAC,CAAC;AAEL,MAAM,CAAC,MAAM,iCAAiC,GAAG,GAAuB,EAAE,CACxE,eAAe,CAAC;IACd,KAAK,EAAE,iBAAiB,CAAC,aAAa;IACtC,UAAU,EAAE,uBAAuB,CAAC,YAAY;IAChD,cAAc;IACd,GAAG,2BAA2B,EAAE;CACjC,CAAC,CAAC;AAUL;;;GAGG;AACH,MAAM,OAAO,YAAa,SAAQ,MAMjC;IACS,MAAM,CAAqB;IAC3B,MAAM,CAAS;IAEvB,YAAY,EACV,MAAM,EACN,SAAS,EACT,MAAM,EACN,MAAM,GAAG,yBAAyB,EAAE,EACpC,OAAO,EACoB;QAC3B,IAAI,CAAC,MAAM,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACrD,CAAC;QAED,IAAI,MAAM,GAAG,kDAAkD,CAAC;QAEhE,IAAI,SAAS,IAAI,MAAM,EAAE,CAAC;YACxB,MAAM,GAAG,gBAAgB,MAAM,0CAA0C,SAAS,wCAAwC,CAAC;QAC7H,CAAC;QAED,KAAK,CAAC;YACJ,IAAI,EAAE,gBAAgB;YACtB,MAAM;YACN,OAAO,EAAE,EAAE;YACX,SAAS,EAAE,qBAAqB;YAChC,MAAM,EAAE,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE;YAC9D,OAAO;YACP,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE;SACjD,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAEQ,cAAc;QACrB,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;QACxB,OAAO;YACL,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,IAAI,EAAE,MAAM,CAAC,IAAI;SACC,CAAC;IACvB,CAAC;IAEQ,eAAe,GAAG,CACzB,GAAgC,EACA,EAAE;QAClC,MAAM,KAAK,GAAG,GAAG,CAAC,SAAS,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QACvD,MAAM,MAAM,GAAG,GAAG,CAAC,WAAW,EAAE,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;QAE7D,IAAI,CAAC,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACnD,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC1C,CAAC;QAED,MAAM,SAAS,GAAG;YAChB,IAAI,EAAE,MAAM;gBACV,CAAC,CAAC,WAAW,KAAK,sCAAsC,IAAI,CAAC,MAAM,EAAE;gBACrE,CAAC,CAAC,WAAW,KAAK,wBAAwB,IAAI,CAAC,MAAM,EAAE;SAC1D,CAAC;QAEF,MAAM,aAAa,GAAG,GAAG,CAAC,UAAU;aACjC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;aAClC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QAEzB,MAAM,iBAAiB,GACrB,aAAa,CAAC,MAAM,GAAG,CAAC;YACtB,CAAC,CAAC;gBACE,IAAI,EAAE,MAAe;gBACrB,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,aAAa,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;aAC3C;YACH,CAAC,CAAC,SAAS,CAAC;QAEhB,MAAM,QAAQ,GAAG,GAAG,CAAC,UAAU;aAC5B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;aAClC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,GAAG,MAAM,EAAE,EAAE,CAAC,EAAE,EAAE;YAC9B,IAAI,IAAI,KAAK,MAAM,EAAE,CAAC;gBACpB,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;oBACpB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,GAAG,CAAC,CAAC;gBAChE,CAAC;gBACD,OAAO;oBACL,IAAI,EAAE,MAAe;oBACrB,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC;iBAClC,CAAC;YACJ,CAAC;YAED,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;gBACzB,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBAE9D,IAAI,aAAa,GAKX,EAAE,CAAC;gBAET,IAAI,eAAe,IAAI,MAAM,EAAE,CAAC;oBAC9B,aAAa;wBACX,MAAM,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;4BAC9B,MAAM,IAAI,GACR,OAAO,CAAC,CAAC,QAAQ,CAAC,SAAS,KAAK,QAAQ;gCACtC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC;gCAClC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC;4BAC3B,OAAO;gCACL,YAAY,EAAE;oCACZ,IAAI,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI;oCACrB,IAAI,EAAE,IAAI;iCACX;6BACF,CAAC;wBACJ,CAAC,CAAC,IAAI,EAAE,CAAC;gBACb,CAAC;gBACD,OAAO;oBACL,IAAI,EAAE,OAAgB;oBACtB,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,aAAa;iBACnC,CAAC;YACJ,CAAC;YAED,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;gBACxB,IAAI,YAAY,IAAI,MAAM,EAAE,CAAC;oBAC3B,OAAO;wBACL,IAAI,EAAE,UAAmB;wBACzB,KAAK,EAAE;4BACL;gCACE,gBAAgB,EAAE;oCAChB,IAAI,EAAE,MAAM,CAAC,UAAU;oCACvB,QAAQ,EAAE,EAAE,MAAM,EAAE,MAAM,CAAC,OAAO,EAAE;iCACrC;6BACF;yBACF;qBACF,CAAC;gBACJ,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,GAAG,CAAC,CAAC;YACnE,CAAC;YAED,MAAM,IAAI,KAAK,CACb,mCAAmC,IAAI,YAAY,CAAC,GAAG,CACxD,CAAC;QACJ,CAAC,CAAC,CAAC;QAEL,MAAM,KAAK,GAAG,GAAG,CAAC,SAAS;YACzB,CAAC,CAAC;gBACE;oBACE,oBAAoB,EAAE,GAAG,CAAC,SAAS,IAAI,EAAE;iBAC1C;aACF;YACH,CAAC,CAAC,SAAS,CAAC;QAEd,IAAI,WAAW,CAAC;QAChB,IAAI,GAAG,CAAC,YAAY,EAAE,CAAC;YACrB,IAAI,GAAG,CAAC,YAAY,KAAK,MAAM,EAAE,CAAC;gBAChC,WAAW,GAAG,EAAE,uBAAuB,EAAE,EAAE,IAAI,EAAE,MAAe,EAAE,EAAE,CAAC;YACvE,CAAC;iBAAM,IAAI,GAAG,CAAC,YAAY,KAAK,MAAM,EAAE,CAAC;gBACvC,WAAW,GAAG,EAAE,uBAAuB,EAAE,EAAE,IAAI,EAAE,MAAe,EAAE,EAAE,CAAC;YACvE,CAAC;iBAAM,IAAI,GAAG,CAAC,YAAY,KAAK,UAAU,EAAE,CAAC;gBAC3C,WAAW,GAAG;oBACZ,uBAAuB,EAAE,EAAE,IAAI,EAAE,KAAc,EAAE;iBAClD,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,WAAW,GAAG;oBACZ,uBAAuB,EAAE;wBACvB,IAAI,EAAE,KAAc;wBACpB,sBAAsB,EAAE,CAAC,GAAG,CAAC,YAAY,CAAC,QAAQ,CAAC,IAAI,CAAC;qBACzD;iBACF,CAAC;YACJ,CAAC;QACH,CAAC;QAED,MAAM,gBAAgB,GAAG;YACvB,eAAe,EAAE,GAAG,CAAC,WAAW,EAAE,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS;YACpE,WAAW,EAAE,GAAG,CAAC,WAAW,EAAE,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW;YACpE,IAAI,EAAE,GAAG,CAAC,WAAW,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI;YAC/C,IAAI,EAAE,GAAG,CAAC,WAAW,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI;YAC/C,cAAc,EAAE,CAAC;YACjB,aAAa,EAAE,GAAG,CAAC,WAAW,EAAE,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa;SAC3E,CAAC;QAEF,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC;QAElD,MAAM,QAAQ,GAA4B;YACxC,QAAQ;YACR,KAAK;YACL,WAAW;YACX,iBAAiB;YACjB,gBAAgB;YAChB,cAAc;SACf,CAAC;QAEF,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAC/B,CAAC,CAAC;IAEO,gBAAgB,GAAG,CAC1B,GAAiC,EACK,EAAE;QACxC,MAAM,KAAK,GAAG,GAAG,CAAC,cAAc,EAAE,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;QAEjE,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;QACzC,CAAC;QAED,IAAI,CAAC,GAAG,CAAC,KAAK,IAAI,GAAG,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACzC,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC1C,CAAC;QAED,MAAM,SAAS,GAAG;YAChB,IAAI,EAAE,WAAW,KAAK,uBAAuB,IAAI,CAAC,MAAM,EAAE;SAC3D,CAAC;QAEF,MAAM,QAAQ,GAAkC;YAC9C,QAAQ,EAAE,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;SACrD,CAAC;QAEF,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAC/B,CAAC,CAAC;IAEO,gBAAgB,GAAG,CAC1B,IAAwC,EAC1B,EAAE;QAChB,MAAM,OAAO,GAAyB,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE;YACtE,MAAM,MAAM,GAAuB,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YAErD,QAAQ,SAAS,CAAC,YAAY,EAAE,CAAC;gBAC/B,KAAK,YAAY;oBACf,MAAM,CAAC,YAAY,GAAG,QAAQ,CAAC;oBAC/B,MAAM;gBACR,KAAK,MAAM;oBACT,MAAM,CAAC,YAAY,GAAG,MAAM,CAAC;oBAC7B,MAAM;gBACR,KAAK,QAAQ;oBACX,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;gBAC3C,KAAK,YAAY;oBACf,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;YACjD,CAAC;YAED,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;gBAC3C,IAAI,MAAM,IAAI,IAAI,EAAE,CAAC;oBACnB,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC;oBAC3B,SAAS;gBACX,CAAC;gBACD,IAAI,cAAc,IAAI,IAAI,EAAE,CAAC;oBAC3B,MAAM,CAAC,aAAa,GAAG;wBACrB;4BACE,EAAE,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;4BAC1B,IAAI,EAAE,UAAU;4BAChB,QAAQ,EAAE;gCACR,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;gCAC5B,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;6BAClC;yBACF;qBACF,CAAC;gBACJ,CAAC;YACH,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,IAAI,UAAkC,CAAC;QACvC,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,UAAU,GAAG;gBACX,WAAW,EAAE,IAAI,CAAC,aAAa,CAAC,eAAe;gBAC/C,YAAY,EAAE,IAAI,CAAC,aAAa,CAAC,gBAAgB;gBACjD,gBAAgB,EAAE,IAAI,CAAC,aAAa,CAAC,oBAAoB;aAC1D,CAAC;QACJ,CAAC;QACD,OAAO;YACL,OAAO;YACP,UAAU;SACX,CAAC;IACJ,CAAC,CAAC;IAEO,sBAAsB,GAAG,CAChC,IAA6C,EAC/B,EAAE;QAChB,OAAO,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC;IACrC,CAAC,CAAC;IAEO,iBAAiB,GAAG,CAC3B,IAA8C,EAC/B,EAAE;QACjB,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAEvE,OAAO;YACL,UAAU;SACX,CAAC;IACJ,CAAC,CAAC;CACH"}

package/build/module/src/ai/google-gemini/index.d.ts

@@ -0,0 +1,2 @@
+export * from './api.js';
+export * from './types.js';

package/build/module/src/ai/google-gemini/index.js

@@ -0,0 +1,3 @@
+export * from './api.js';
+export * from './types.js';
+//# sourceMappingURL=index.js.map

package/build/module/src/ai/google-gemini/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/ai/google-gemini/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAC;AACzB,cAAc,YAAY,CAAC"}

package/build/module/src/ai/google-gemini/info.d.ts

@@ -0,0 +1,6 @@
+import type { TextModelInfo } from '../types.js';
+/**
+ * GoogleGemini: Model information
+ * @export
+ */
+export declare const modelInfoGoogleGemini: TextModelInfo[];

package/build/module/src/ai/google-gemini/info.js

@@ -0,0 +1,29 @@
+import { GoogleGeminiModel } from './types.js';
+/**
+ * GoogleGemini: Model information
+ * @export
+ */
+export const modelInfoGoogleGemini = [
+    {
+        name: GoogleGeminiModel.Gemini15Flash,
+        currency: 'usd',
+        characterIsToken: true,
+        promptTokenCostPer1M: 0.35,
+        completionTokenCostPer1M: 0.7
+    },
+    {
+        name: GoogleGeminiModel.Gemini15Pro,
+        currency: 'usd',
+        characterIsToken: true,
+        promptTokenCostPer1M: 3.5,
+        completionTokenCostPer1M: 1.75
+    },
+    {
+        name: GoogleGeminiModel.Gemini1Pro,
+        currency: 'usd',
+        characterIsToken: true,
+        promptTokenCostPer1M: 0.5,
+        completionTokenCostPer1M: 1.5
+    }
+];
+//# sourceMappingURL=info.js.map

package/build/module/src/ai/google-gemini/info.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"info.js","sourceRoot":"","sources":["../../../../../src/ai/google-gemini/info.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAE/C;;;GAGG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAoB;IACpD;QACE,IAAI,EAAE,iBAAiB,CAAC,aAAa;QACrC,QAAQ,EAAE,KAAK;QACf,gBAAgB,EAAE,IAAI;QACtB,oBAAoB,EAAE,IAAI;QAC1B,wBAAwB,EAAE,GAAG;KAC9B;IACD;QACE,IAAI,EAAE,iBAAiB,CAAC,WAAW;QACnC,QAAQ,EAAE,KAAK;QACf,gBAAgB,EAAE,IAAI;QACtB,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,IAAI;KAC/B;IACD;QACE,IAAI,EAAE,iBAAiB,CAAC,UAAU;QAClC,QAAQ,EAAE,KAAK;QACf,gBAAgB,EAAE,IAAI;QACtB,oBAAoB,EAAE,GAAG;QACzB,wBAAwB,EAAE,GAAG;KAC9B;CACF,CAAC"}