重新划分目录 (#214)

* 🦄 refactor(web): 修改zhikongtaiweb到web * 🦄 refactor: 重写前端路由守护尚未写完 * 🦄 refactor: 标准化路由 * update:前端重写，保留后端 * update:添加前端代码 * update:pip转成poetry启动 * update:增加mem0ai包依赖 * update:文档增加mem0ai的描述 * feat: play online mp3 (#181) Co-authored-by: 欣南科技 <huangrongzhuang@xin-nan.com> * 修改前端代码 * update:调整项目目录 * update:优化 * update:配置文件去除8002端口 * update:增加开发说明 * update:更新开发协议 --------- Co-authored-by: kalicyh <34980061+kaliCYH@users.noreply.github.com> Co-authored-by: hrz <1710360675@qq.com> Co-authored-by: freshlife001 <talent@mises.site> Co-authored-by: CGD <3030332422@qq.com>
2026-07-31 03:33:56 +08:00 · 2025-03-05 23:13:24 +08:00
parent 8b151d07c2
commit 0e43748fdc
289 changed files with 18127 additions and 85418 deletions
@@ -0,0 +1,81 @@
+from config.logger import setup_logging
+import google.generativeai as genai
+from core.providers.llm.base import LLMProviderBase
+
+TAG = __name__
+logger = setup_logging()
+
+class LLMProvider(LLMProviderBase):
+    def __init__(self, config):
+        """初始化Gemini LLM Provider"""
+        self.model_name = config.get("model_name", "gemini-1.5-pro") 
+        self.api_key = config.get("api_key")
+        
+        if not self.api_key or "你" in self.api_key:
+            logger.bind(tag=TAG).error("你还没配置Gemini LLM的密钥，请在配置文件中配置密钥，否则无法正常工作")
+            return
+            
+        try:
+            # 初始化Gemini客户端
+            genai.configure(api_key=self.api_key)
+            self.model = genai.GenerativeModel(self.model_name)
+            
+            # 设置生成参数
+            self.generation_config = {
+                "temperature": 0.7,
+                "top_p": 0.9,
+                "top_k": 40,
+                "max_output_tokens": 2048,
+            }
+            self.chat = None
+        except Exception as e:
+            logger.bind(tag=TAG).error(f"Gemini初始化失败: {e}")
+            self.model = None
+
+    def response(self, session_id, dialogue):
+        """生成Gemini对话响应"""
+        if not self.model:
+            yield "【Gemini服务未正确初始化】"
+            return
+
+        try:
+            # 处理对话历史
+            chat_history = []
+            for msg in dialogue[:-1]:  # 历史对话
+                role = "model" if msg["role"] == "assistant" else "user"
+                content = msg["content"].strip()
+                if content:
+                    chat_history.append({
+                        "role": role,
+                        "parts": [content]
+                    })
+
+            # 获取当前消息
+            current_msg = dialogue[-1]["content"]
+
+            # 创建新的聊天会话
+            chat = self.model.start_chat(history=chat_history)
+            
+            # 发送消息并获取流式响应
+            response = chat.send_message(
+                current_msg,
+                stream=True,
+                generation_config=self.generation_config
+            )
+
+            # 处理流式响应
+            for chunk in response:
+                if hasattr(chunk, 'text') and chunk.text:
+                    yield chunk.text
+
+        except Exception as e:
+            error_msg = str(e)
+            logger.bind(tag=TAG).error(f"Gemini响应生成错误: {error_msg}")
+            
+            # 针对不同错误返回友好提示
+            if "Rate limit" in error_msg:
+                yield "【Gemini服务请求太频繁,请稍后再试】"
+            elif "Invalid API key" in error_msg:
+                yield "【Gemini API key无效】"
+            else:
+                yield f"【Gemini服务响应异常: {error_msg}】"