chore: initial baseline with P0-safety .gitignore

2026-06-14 16:49:18 +08:00
commit 63262292d7
510 changed files with 146008 additions and 0 deletions
@@ -0,0 +1,61 @@
+# =============================================================================
+# RAGFlow 配置加载器
+# =============================================================================
+# 说明：从数据库 system_configs 表加载 RAGFlow 配置，创建客户端实例
+# 配置项：integration_ragflow_api_url + integration_ragflow_api_key
+
+import logging
+from typing import Optional
+
+from sqlalchemy import select
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.models.system_config import SystemConfig
+
+from .client import RagflowClient
+from .exceptions import RagflowConfigError
+
+logger = logging.getLogger(__name__)
+
+# 默认 RAGFlow API 地址（生产环境）
+DEFAULT_RAGFLOW_BASE_URL = "http://10.80.0.85:9380"
+
+
+async def _get_config(db: AsyncSession, key: str) -> str:
+    """从数据库读取单个配置值。"""
+    result = await db.execute(
+        select(SystemConfig.config_value).where(SystemConfig.config_key == key)
+    )
+    row = result.scalar()
+    return row if row else ""
+
+
+async def get_ragflow_client(db: AsyncSession) -> RagflowClient:
+    """从数据库配置创建 RAGFlow 客户端实例。
+
+    读取 system_configs 表中的：
+    - integration_ragflow_api_url: RAGFlow API 地址
+    - integration_ragflow_api_key: RAGFlow API Key
+
+    Args:
+        db: 数据库会话
+
+    Returns:
+        RagflowClient: 客户端实例
+
+    Raises:
+        RagflowConfigError: 配置缺失
+    """
+    api_url = await _get_config(db, "integration_ragflow_api_url")
+    api_key = await _get_config(db, "integration_ragflow_api_key")
+
+    # 如果数据库没有配置，使用默认地址
+    if not api_url:
+        api_url = DEFAULT_RAGFLOW_BASE_URL
+
+    if not api_key:
+        raise RagflowConfigError(
+            "RAGFlow API Key 未配置，请在管理后台 → 集成管理 → RAGFlow 中设置"
+        )
+
+    return RagflowClient(api_key=api_key, base_url=api_url)