@aituber-onair/chat 0.1.0

package/dist/esm/services/providers/gemini/GeminiChatService.d.ts

@@ -0,0 +1,104 @@
+import { ChatService } from '../../ChatService';
+import { Message, MessageWithVision, ToolChatCompletion, ToolDefinition, MCPServerConfig } from '../../../types';
+import { ChatResponseLength } from '../../../constants/chat';
+/**
+ * Gemini implementation of ChatService
+ */
+export declare class GeminiChatService implements ChatService {
+    /** Provider name */
+    readonly provider: string;
+    private apiKey;
+    private model;
+    private visionModel;
+    private tools;
+    private mcpServers;
+    private mcpToolSchemas;
+    private mcpSchemasInitialized;
+    private responseLength?;
+    /** id(OpenAI) → name(Gemini) mapping */
+    private callIdMap;
+    private safeJsonParse;
+    private normalizeToolResult;
+    /**
+     * camelCase → snake_case conversion (v1beta)
+     */
+    private adaptKeysForApi;
+    /**
+     * Constructor
+     * @param apiKey Google API key
+     * @param model Name of the model to use
+     * @param visionModel Name of the vision model
+     * @param tools Array of tool definitions
+     * @param mcpServers Array of MCP server configurations
+     */
+    constructor(apiKey: string, model?: string, visionModel?: string, tools?: ToolDefinition[], mcpServers?: MCPServerConfig[], responseLength?: ChatResponseLength);
+    /**
+     * Get the current model name
+     * @returns Model name
+     */
+    getModel(): string;
+    /**
+     * Get the current vision model name
+     * @returns Vision model name
+     */
+    getVisionModel(): string;
+    /**
+     * Get configured MCP servers
+     * @returns Array of MCP server configurations
+     */
+    getMCPServers(): MCPServerConfig[];
+    /**
+     * Add MCP server configuration
+     * @param serverConfig MCP server configuration
+     */
+    addMCPServer(serverConfig: MCPServerConfig): void;
+    /**
+     * Remove MCP server by name
+     * @param serverName Name of the server to remove
+     */
+    removeMCPServer(serverName: string): void;
+    /**
+     * Check if MCP servers are configured
+     * @returns True if MCP servers are configured
+     */
+    hasMCPServers(): boolean;
+    /**
+     * Initialize MCP tool schemas by fetching from servers
+     * @private
+     */
+    private initializeMCPSchemas;
+    /**
+     * Process chat messages
+     * @param messages Array of messages to send
+     * @param onPartialResponse Callback to receive each part of streaming response
+     * @param onCompleteResponse Callback to execute when response is complete
+     */
+    processChat(messages: Message[], onPartialResponse: (text: string) => void, onCompleteResponse: (text: string) => Promise<void>): Promise<void>;
+    processVisionChat(messages: MessageWithVision[], onPartialResponse: (text: string) => void, onCompleteResponse: (text: string) => Promise<void>): Promise<void>;
+    private convertMessagesToGeminiFormat;
+    private callGemini;
+    /**
+     * Convert AITuber OnAir vision messages to Gemini format
+     * @param messages Array of vision messages
+     * @returns Gemini formatted vision messages
+     */
+    private convertVisionMessagesToGeminiFormat;
+    /**
+     * Convert Blob to Base64 string
+     * @param blob Image blob
+     * @returns Promise with base64 encoded string
+     */
+    private blobToBase64;
+    /**
+     * Map AITuber OnAir roles to Gemini roles
+     * @param role AITuber OnAir role
+     * @returns Gemini role
+     */
+    private mapRoleToGemini;
+    private parseStream;
+    private parseOneShot;
+    chatOnce(messages: Message[], stream?: boolean, onPartialResponse?: (t: string) => void, maxTokens?: number): Promise<ToolChatCompletion>;
+    visionChatOnce(messages: MessageWithVision[], stream?: boolean, onPartialResponse?: (t: string) => void, maxTokens?: number): Promise<ToolChatCompletion>;
+    private genUUID;
+}
+//# sourceMappingURL=GeminiChatService.d.ts.map

package/dist/esm/services/providers/gemini/GeminiChatService.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"GeminiChatService.d.ts","sourceRoot":"","sources":["../../../../../src/services/providers/gemini/GeminiChatService.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EACL,OAAO,EACP,iBAAiB,EAEjB,kBAAkB,EAClB,cAAc,EACd,eAAe,EAChB,MAAM,gBAAgB,CAAC;AAMxB,OAAO,EACL,kBAAkB,EAEnB,MAAM,yBAAyB,CAAC;AAKjC;;GAEG;AACH,qBAAa,iBAAkB,YAAW,WAAW;IACnD,oBAAoB;IACpB,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAY;IAErC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,KAAK,CAAS;IACtB,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,KAAK,CAAmB;IAChC,OAAO,CAAC,UAAU,CAAoB;IACtC,OAAO,CAAC,cAAc,CAAwB;IAC9C,OAAO,CAAC,qBAAqB,CAAkB;IAC/C,OAAO,CAAC,cAAc,CAAC,CAAqB;IAE5C,wCAAwC;IACxC,OAAO,CAAC,SAAS,CAA6B;IAK9C,OAAO,CAAC,aAAa;IAQrB,OAAO,CAAC,mBAAmB;IAM3B;;OAEG;IACH,OAAO,CAAC,eAAe;IAoBvB;;;;;;;OAOG;gBAED,MAAM,EAAE,MAAM,EACd,KAAK,GAAE,MAAoC,EAC3C,WAAW,GAAE,MAAoC,EACjD,KAAK,GAAE,cAAc,EAAO,EAC5B,UAAU,GAAE,eAAe,EAAO,EAClC,cAAc,CAAC,EAAE,kBAAkB;IAiBrC;;;OAGG;IACH,QAAQ,IAAI,MAAM;IAIlB;;;OAGG;IACH,cAAc,IAAI,MAAM;IAIxB;;;OAGG;IACH,aAAa,IAAI,eAAe,EAAE;IAIlC;;;OAGG;IACH,YAAY,CAAC,YAAY,EAAE,eAAe,GAAG,IAAI;IAMjD;;;OAGG;IACH,eAAe,CAAC,UAAU,EAAE,MAAM,GAAG,IAAI;IAQzC;;;OAGG;IACH,aAAa,IAAI,OAAO;IAIxB;;;OAGG;YACW,oBAAoB;IAwClC;;;;;OAKG;IACG,WAAW,CACf,QAAQ,EAAE,OAAO,EAAE,EACnB,iBAAiB,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,EACzC,kBAAkB,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,GAClD,OAAO,CAAC,IAAI,CAAC;IAqCV,iBAAiB,CACrB,QAAQ,EAAE,iBAAiB,EAAE,EAC7B,iBAAiB,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,IAAI,EACzC,kBAAkB,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,GAClD,OAAO,CAAC,IAAI,CAAC;IAqChB,OAAO,CAAC,6BAA6B;YAsEvB,UAAU;IAmHxB;;;;OAIG;YACW,mCAAmC;IA8GjD;;;;OAIG;IACH,OAAO,CAAC,YAAY;IASpB;;;;OAIG;IACH,OAAO,CAAC,eAAe;YAgBT,WAAW;IAiFzB,OAAO,CAAC,YAAY;IAwCd,QAAQ,CACZ,QAAQ,EAAE,OAAO,EAAE,EACnB,MAAM,GAAE,OAAc,EACtB,iBAAiB,GAAE,CAAC,CAAC,EAAE,MAAM,KAAK,IAAe,EACjD,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,kBAAkB,CAAC;IAUxB,cAAc,CAClB,QAAQ,EAAE,iBAAiB,EAAE,EAC7B,MAAM,GAAE,OAAe,EACvB,iBAAiB,GAAE,CAAC,CAAC,EAAE,MAAM,KAAK,IAAe,EACjD,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,kBAAkB,CAAC;IAe9B,OAAO,CAAC,OAAO;CAShB"}