支持角色id与性格配置

2025-08-11 14:29:39 +08:00
parent 0f8902405d
commit 1c33f4345b
11 changed files with 773 additions and 81 deletions
@@ -0,0 +1,165 @@
+import { GoogleGenAI } from "@google/genai";
+import { RoleConfig } from "./RoleConfig";
+
+// API配置
+const API_CONFIG = {
+    apiKey: "AIzaSyBJT_68Fc-sKPp_lYSbQmDck0otsd3uKn8",
+    model: "gemini-2.5-flash",
+    temperature: 0.7
+};
+
+/**
+ * AI聊天服务
+ * 管理多个独立的聊天实例，每个角色有独立的对话上下文
+ */
+export class ChatAIService {
+    private static _instance: ChatAIService;
+    private ai: GoogleGenAI;
+    private chatInstances: Map<number, any> = new Map();
+    private currentRoleId: number | null = null;
+
+    private constructor() {
+        this.ai = new GoogleGenAI({ apiKey: API_CONFIG.apiKey });
+    }
+
+    /**
+     * 获取单例实例
+     */
+    public static get Instance(): ChatAIService {
+        if (!this._instance) {
+            this._instance = new ChatAIService();
+        }
+        return this._instance;
+    }
+
+    /**
+     * 创建或获取指定角色的聊天实例
+     * @param roleId 角色ID
+     */
+    private createOrGetChat(roleId: number): any {
+        if (!this.chatInstances.has(roleId)) {
+            const systemInstruction = RoleConfig.getRoleInstruction(roleId);
+            const chat = this.ai.chats.create({
+                model: API_CONFIG.model,
+                config: {
+                    temperature: API_CONFIG.temperature,
+                    systemInstruction: systemInstruction
+
+                },
+            });
+            this.chatInstances.set(roleId, chat);
+            console.log(`Created new chat instance for role ${roleId}`);
+        }
+        return this.chatInstances.get(roleId);
+    }
+
+    /**
+     * 设置当前活动的角色ID
+     * @param roleId 角色ID
+     */
+    public setCurrentRole(roleId: number): void {
+        this.currentRoleId = roleId;
+        // 预创建聊天实例
+        this.createOrGetChat(roleId);
+    }
+
+    /**
+     * 获取当前角色ID
+     */
+    public getCurrentRoleId(): number | null {
+        return this.currentRoleId;
+    }
+
+    /**
+     * 发送消息到AI
+     * @param roleId 角色ID
+     * @param message 用户消息
+     */
+    public async sendMessage(roleId: number, message: string): Promise<string> {
+        try {
+            const chat = this.createOrGetChat(roleId);
+            const response = await chat.sendMessage({
+                message: message
+            });
+
+            if (response && response.text) {
+                console.log(`Response from role ${roleId}:`, response.text);
+                return response.text;
+            } else {
+                console.warn(`Empty response from role ${roleId}`);
+                return null;
+            }
+        } catch (error) {
+            console.error(`Error sending message to role ${roleId}:`, error);
+            return null;
+        }
+    }
+
+    /**
+     * 清除指定角色的聊天历史
+     * @param roleId 角色ID
+     */
+    public clearChatHistory(roleId: number): void {
+        if (this.chatInstances.has(roleId)) {
+            this.chatInstances.delete(roleId);
+            console.log(`Cleared chat history for role ${roleId}`);
+        }
+    }
+
+    /**
+     * 清除所有聊天历史
+     */
+    public clearAllChatHistory(): void {
+        this.chatInstances.clear();
+        console.log("Cleared all chat histories");
+    }
+
+    /**
+     * 获取当前活跃的聊天实例数量
+     */
+    public getActiveChatCount(): number {
+        return this.chatInstances.size;
+    }
+
+    /**
+     * 兼容旧接口的Post方法
+     * @deprecated 请使用sendMessage方法
+     */
+    public async Post(data: GPTRequest): Promise<string> {
+        const roleId = this.currentRoleId || 10001; // 默认使用第一个角色
+        const message = data.messages[0]?.content || "";
+        return this.sendMessage(roleId, message);
+    }
+}
+
+// 请求数据类型定义
+export interface GPTRequest {
+    model: string;
+    messages: { role: string; content: string }[];
+    temperature: number;
+    id: string;
+}
+
+// 兼容旧名称
+export type GPTResquest = GPTRequest;
+
+// 响应数据类型定义（保留以备后用）
+export interface GPTResult {
+    id: string;
+    object: string;
+    created: number;
+    model: string;
+    usage: {
+        prompt_tokens: number;
+        completion_tokens: number;
+        total_tokens: number;
+    };
+    choices: {
+        message: {
+            role: string;
+            content: string;
+        };
+        finish_reason: string;
+        index: number;
+    }[];
+}