koishi-plugin-chatluna-google-gemini-adapter 1.0.0-beta.2 → 1.0.0-beta.20

package/lib/index.d.ts

@@ -1,12 +1,111 @@
-import { ChatLunaPlugin } from 'koishi-plugin-chatluna/lib/services/chat';
-import { Context, Logger, Schema } from 'koishi';
-export declare let logger: Logger;
-export declare function apply(ctx: Context, config: Config): void;
-export interface Config extends ChatLunaPlugin.Config {
-    apiKeys: [string, string][];
-    maxTokens: number;
-    temperature: number;
-}
-export declare const Config: Schema<Config>;
-export declare const inject: string[];
-export declare const name = "chatluna-gemini-adapter";
+import { ChatLunaPlugin } from 'koishi-plugin-chatluna/services/chat';
+import { Context, Logger, Schema } from 'koishi';
+import { AIMessageChunk, BaseMessage, ChatMessageChunk, HumanMessageChunk, MessageType, SystemMessageChunk } from '@langchain/core/messages';
+import { StructuredTool } from '@langchain/core/tools';
+import { ChatGenerationChunk } from '@langchain/core/outputs';
+import { EmbeddingsRequester, EmbeddingsRequestParams, ModelRequester, ModelRequestParams } from 'koishi-plugin-chatluna/llm-core/platform/api';
+import { ClientConfig } from 'koishi-plugin-chatluna/llm-core/platform/config';
+import { PlatformModelAndEmbeddingsClient } from 'koishi-plugin-chatluna/llm-core/platform/client';
+import { ChatHubBaseEmbeddings, ChatLunaChatModel } from 'koishi-plugin-chatluna/llm-core/platform/model';
+import { ModelInfo } from 'koishi-plugin-chatluna/llm-core/platform/types';
+export let logger: Logger;
+export function apply(ctx: Context, config: Config): void;
+export interface Config extends ChatLunaPlugin.Config {
+    apiKeys: [string, string][];
+    maxTokens: number;
+    temperature: number;
+}
+export const Config: Schema<Config>;
+export const inject: string[];
+export const name = "chatluna-google-gemini-adapter";
+export interface ChatCompletionResponseMessage {
+    role: string;
+    parts?: ChatPart[];
+}
+export type ChatPart = ChatMessagePart | ChatUploadDataPart | ChatFunctionCallingPart | ChatFunctionResponsePart;
+export type ChatMessagePart = {
+    text: string;
+};
+export type ChatUploadDataPart = {
+    inline_data: {
+        mime_type: string;
+        data?: string;
+    };
+};
+export type ChatFunctionCallingPart = {
+    functionCall: {
+        name: string;
+        args?: any;
+    };
+};
+export type ChatFunctionResponsePart = {
+    functionResponse: {
+        name: string;
+        response: any;
+    };
+};
+export interface ChatResponse {
+    candidates: {
+        content: ChatCompletionResponseMessage;
+        finishReason: string;
+        index: number;
+        safetyRatings: {
+            category: string;
+            probability: string;
+        }[];
+    }[];
+    promptFeedback: {
+        safetyRatings: {
+            category: string;
+            probability: string;
+        }[];
+    };
+}
+export interface ChatCompletionFunction {
+    name: string;
+    description?: string;
+    parameters?: {
+        [key: string]: any;
+    };
+}
+export interface ChatCompletionMessageFunctionCall {
+    name: string;
+    args?: any;
+}
+export interface CreateEmbeddingResponse {
+    embeddings: {
+        values: number[];
+    }[];
+}
+export type ChatCompletionResponseMessageRoleEnum = 'system' | 'model' | 'user' | 'function';
+export function langchainMessageToGeminiMessage(messages: BaseMessage[], model?: string): Promise<ChatCompletionResponseMessage[]>;
+export function partAsType<T extends ChatPart>(part: ChatPart): T;
+export function formatToolsToGeminiAITools(tools: StructuredTool[]): ChatCompletionFunction[];
+export function formatToolToGeminiAITool(tool: StructuredTool): ChatCompletionFunction;
+export function messageTypeToGeminiRole(type: MessageType): ChatCompletionResponseMessageRoleEnum;
+export function convertDeltaToMessageChunk(delta: Record<string, any>, defaultRole?: ChatCompletionResponseMessageRoleEnum): HumanMessageChunk | AIMessageChunk | SystemMessageChunk | ChatMessageChunk;
+export class GeminiRequester extends ModelRequester implements EmbeddingsRequester {
+    private _config;
+    private _plugin;
+    constructor(_config: ClientConfig, _plugin: ChatLunaPlugin);
+    completionStream(params: ModelRequestParams): AsyncGenerator<ChatGenerationChunk>;
+    embeddings(params: EmbeddingsRequestParams): Promise<number[] | number[][]>;
+    getModels(): Promise<string[]>;
+    private _post;
+    private _get;
+    private _concatUrl;
+    private _buildHeaders;
+    init(): Promise<void>;
+    dispose(): Promise<void>;
+}
+export class GeminiClient extends PlatformModelAndEmbeddingsClient {
+    private _config;
+    platform: string;
+    private _requester;
+    private _models;
+    constructor(ctx: Context, _config: Config, clientConfig: ClientConfig, plugin: ChatLunaPlugin);
+    init(): Promise<void>;
+    refreshModels(): Promise<ModelInfo[]>;
+    getModels(): Promise<ModelInfo[]>;
+    protected _createModel(model: string): ChatLunaChatModel | ChatHubBaseEmbeddings;
+}