Merge branch 'main' into feature/i18n

Author: mashb1t

api.py

@@ -13,6 +13,7 @@
 from biz.gitlab.webhook_handler import slugify_url
 from biz.queue.worker import handle_merge_request_event, handle_push_event, handle_github_pull_request_event, handle_github_push_event
 from biz.service.review_service import ReviewService
+from biz.utils.config_checker import check_config
 from biz.utils.im import notifier
 from biz.utils.log import logger
 from biz.utils.queue import handle_queue
@@ -22,6 +23,7 @@
 api_app = Flask(__name__)
 
 from biz.utils.i18n import get_translator
+
 _ = get_translator()
 
 PUSH_REVIEW_ENABLED = os.environ.get('PUSH_REVIEW_ENABLED', '0') == '1'
@@ -124,6 +126,7 @@ def handle_webhook():
     else:
         return jsonify({'message': _('Invalid data format')}), 400
 
+
 def handle_github_webhook(event_type, data):
     # 获取GitHub配置
     github_token = os.getenv('GITHUB_ACCESS_TOKEN') or request.headers.get('X-GitHub-Token')
@@ -156,6 +159,7 @@ def handle_github_webhook(event_type, data):
         logger.error(error_message)
         return jsonify(error_message), 400
 
+
 def handle_gitlab_webhook(data):
     object_kind = data.get("object_kind")
 
@@ -207,7 +211,9 @@ def handle_gitlab_webhook(data):
         logger.error(error_message)
         return jsonify(error_message), 400
 
+
 if __name__ == '__main__':
+    check_config()
     # 启动定时任务调度器
     setup_scheduler()
 

biz/llm/client/base.py

@@ -2,11 +2,21 @@
 from typing import List, Dict, Optional
 
 from biz.llm.types import NotGiven, NOT_GIVEN
+from biz.utils.log import logger
 
 
 class BaseClient:
     """ Base class for chat models client. """
 
+    def ping(self) -> bool:
+        """Ping the model to check connectivity."""
+        try:
+            result = self.completions(messages=[{"role": "user", "content": '请仅返回 "ok"。'}])
+            return result and result == 'ok'
+        except Exception:
+            logger.error("尝试连接LLM失败， {e}")
+            return False
+
     @abstractmethod
     def completions(self,
                     messages: List[Dict[str, str]],

biz/llm/factory.py

@@ -24,7 +24,6 @@ def getClient(provider: str = None) -> BaseClient:
 
         provider_func = chat_model_providers.get(provider)
         if provider_func:
-            logger.info(_("Successfully loaded LLM provider: {provider}").format(provider=provider))
             return provider_func()
         else:
             raise Exception(_('Unknown chat model provider: {provider}'.format(provider=provider)))

biz/utils/config_checker.py

@@ -0,0 +1,77 @@
+import os
+
+from dotenv import load_dotenv
+
+from biz.llm.factory import Factory
+from biz.utils.log import logger
+
+# 指定环境变量文件路径
+ENV_FILE_PATH = "conf/.env"
+load_dotenv(ENV_FILE_PATH)
+
+from i18n import get_translator
+
+_ = get_translator()
+
+REQUIRED_ENV_VARS = [
+    "LLM_PROVIDER",
+]
+
+# 允许的 LLM 供应商
+LLM_PROVIDERS = {"zhipuai", "openai", "deepseek", "ollama"}
+
+# 每种供应商必须配置的键
+LLM_REQUIRED_KEYS = {
+    "zhipuai": ["ZHIPUAI_API_KEY", "ZHIPUAI_API_MODEL"],
+    "openai": ["OPENAI_API_KEY", "OPENAI_API_MODEL"],
+    "deepseek": ["DEEPSEEK_API_KEY", "DEEPSEEK_API_MODEL"],
+    "ollama": ["OLLAMA_API_BASE_URL", "OLLAMA_API_MODEL"],
+}
+
+
+def check_env_vars():
+    """检查环境变量"""
+    missing_vars = [var for var in REQUIRED_ENV_VARS if var not in os.environ]
+    if missing_vars:
+        logger.warning(_("缺少环境变量: {}").format(", ".join(missing_vars)))
+    else:
+        logger.info(_("所有必要的环境变量均已设置。"))
+
+
+def check_llm_provider():
+    """检查 LLM 供应商的配置"""
+    llm_provider = os.getenv("LLM_PROVIDER")
+
+    if not llm_provider:
+        logger.error(_("LLM_PROVIDER 未设置！"))
+        return
+
+    if llm_provider not in LLM_PROVIDERS:
+        logger.error(_("LLM_PROVIDER 值错误，应为 {} 之一。").format(LLM_PROVIDERS))
+        return
+
+    required_keys = LLM_REQUIRED_KEYS.get(llm_provider, [])
+    missing_keys = [key for key in required_keys if not os.getenv(key)]
+
+    if missing_keys:
+        logger.error(_("当前 LLM 供应商为 {}，但缺少必要的环境变量: {}").format(llm_provider, ', '.join(missing_keys)))
+    else:
+        logger.info(_("LLM 供应商 {} 的配置项已设置。").format(llm_provider))
+
+
+def check_llm_connectivity():
+    client = Factory().getClient()
+    logger.info(_("正在检查 LLM 供应商的连接..."))
+    if client.ping():
+        logger.info(_("LLM 可以连接成功。"))
+    else:
+        logger.error(_("LLM连接可能有问题，请检查配置项。"))
+
+
+def check_config():
+    """主检查入口"""
+    logger.info(_("开始检查配置项..."))
+    check_env_vars()
+    check_llm_provider()
+    check_llm_connectivity()
+    logger.info(_("配置项检查完成。"))

locales/en_US/LC_MESSAGES/messages.mo

@@ -7,7 +7,7 @@ msgid ""
 msgstr ""
 "Project-Id-Version: PACKAGE VERSION\n"
 "Report-Msgid-Bugs-To: \n"
-"POT-Creation-Date: 2025-03-28 17:40+0100\n"
+"POT-Creation-Date: 2025-03-29 15:24+0100\n"
 "PO-Revision-Date: 2025-03-18 15:05+0100\n"
 "Last-Translator: Automatically generated\n"
 "Language-Team: none\n"
@@ -16,67 +16,67 @@ msgstr ""
 "Content-Type: text/plain; charset=UTF-8\n"
 "Content-Transfer-Encoding: 8bit\n"
 
-#: api.py:55 api.py:56
+#: api.py:57 api.py:58
 msgid "No data to process."
 msgstr "No data to process."
 
-#: api.py:66
+#: api.py:68
 msgid "代码提交日报"
 msgstr "Code Submission Daily Report"
 
-#: api.py:71
+#: api.py:73
 msgid "Failed to generate daily report: {}"
 msgstr "Failed to generate daily report: {}"
 
-#: api.py:99
+#: api.py:101
 msgid "Scheduler started successfully."
 msgstr "Scheduler started successfully."
 
-#: api.py:104
+#: api.py:106
 msgid "Error setting up scheduler: {}"
 msgstr "Error setting up scheduler: {}"
 
-#: api.py:115
+#: api.py:117
 msgid "Invalid JSON"
 msgstr "Invalid JSON"
 
-#: api.py:125
+#: api.py:127
 msgid "Invalid data format"
 msgstr "Invalid data format"
 
-#: api.py:131 api.py:167 api.py:170
+#: api.py:134 api.py:171 api.py:174
 msgid "Missing GitLab URL"
 msgstr "Missing GitLab URL"
 
-#: api.py:137 api.py:186
+#: api.py:140 api.py:190
 msgid "Received event: {}"
 msgstr "Received event: {}"
 
-#: api.py:138 api.py:187
+#: api.py:141 api.py:191
 msgid "Payload: {}"
 msgstr "Payload: {}"
 
-#: api.py:144 api.py:150
+#: api.py:147 api.py:153
 msgid "GitHub request received(event_type={}), will process asynchronously."
 msgstr "GitHub request received(event_type={}), will process asynchronously."
 
-#: api.py:154
+#: api.py:157
 msgid ""
 "Only pull_request and push events are supported for GitHub webhook, but "
 "received:: {}."
 msgstr ""
 "Only pull_request and push events are supported for GitHub webhook, but "
 "received:: {}."
 
-#: api.py:181
+#: api.py:185
 msgid "Missing GitLab access token"
 msgstr "Missing GitLab access token"
 
-#: api.py:194 api.py:201
+#: api.py:198 api.py:205
 msgid "Request received(object_kind={}), will process asynchronously."
 msgstr "Request received(object_kind={}), will process asynchronously."
 
-#: api.py:205
+#: api.py:209
 msgid ""
 "Only merge_request and push events are supported (both Webhook and System "
 "Hook), but received: {}."
@@ -325,12 +325,7 @@ msgstr ""
 msgid "调用DeepSeek API时出错: {e}"
 msgstr "Error occurred while calling DeepSeek API: {e}"
 
-#: biz/llm/factory.py:27
-#, python-brace-format
-msgid "Successfully loaded LLM provider: {provider}"
-msgstr "Successfully loaded LLM provider: {provider}"
-
-#: biz/llm/factory.py:30
+#: biz/llm/factory.py:29
 #, python-brace-format
 msgid "Unknown chat model provider: {provider}"
 msgstr "Unknown chat model provider: {provider}"
@@ -429,6 +424,50 @@ msgstr "Code is empty"
 msgid "总分[:：]\\s*\\**(\\d+)分?"
 msgstr "Total [sS]core[:：]\\s*\\**(\\d+) points"
 
+#: biz/utils/config_checker.py:36
+msgid "缺少环境变量: {}"
+msgstr "Missing environment variables: {}"
+
+#: biz/utils/config_checker.py:38
+msgid "所有必要的环境变量均已设置。"
+msgstr "All necessary environment variables are set."
+
+#: biz/utils/config_checker.py:46
+msgid "LLM_PROVIDER 未设置！"
+msgstr "LLM_PROVIDER not set!"
+
+#: biz/utils/config_checker.py:50
+msgid "LLM_PROVIDER 值错误，应为 {} 之一。"
+msgstr "Wrong LLM_PROVIDER value, should be one of {}."
+
+#: biz/utils/config_checker.py:57
+msgid "当前 LLM 供应商为 {}，但缺少必要的环境变量: {}"
+msgstr "The current LLM provider is {}, but the required environment variables are missing: {}"
+
+#: biz/utils/config_checker.py:59
+msgid "LLM 供应商 {} 的配置项已设置。"
+msgstr "Configuration item for LLM vendor {} has been set."
+
+#: biz/utils/config_checker.py:64
+msgid "正在检查 LLM 供应商的连接..."
+msgstr "Checking LLM supplier connectivity..."
+
+#: biz/utils/config_checker.py:66
+msgid "LLM 可以连接成功。"
+msgstr "LLM can connect successfully."
+
+#: biz/utils/config_checker.py:68
+msgid "LLM连接可能有问题，请检查配置项。"
+msgstr "There may be a problem with the LLM connection. Please check the configuration items."
+
+#: biz/utils/config_checker.py:73
+msgid "开始检查配置项..."
+msgstr "Starting to check configuration items..."
+
+#: biz/utils/config_checker.py:77
+msgid "配置项检查完成。"
+msgstr "The configuration item check is complete."
+
 #: biz/utils/im/dingtalk.py:30
 msgid "未提供项目名称，且未设置默认的钉钉 Webhook URL。"
 msgstr "Project name not provided, and no default DingTalk Webhook URL set."