Simon
96 lines
3.5 KiB
Python
96 lines
3.5 KiB
Python
# =============================================================================
|
||
# 企微IT智能服务台 — 阶段5 自动化 外部客户端工厂
|
||
# =============================================================================
|
||
# 说明：集中解析各外部系统的配置来源：
|
||
# - 优先使用 settings 中的 AUTOMATION_* 环境变量（架构约定）
|
||
# - 缺失时回退到既有 system_configs 表配置
|
||
# （huorong/lianruan/ragflow 在 app/integrations/*/config.py 中已有 getter）
|
||
#
|
||
# 为什么有工厂：自动化引擎既能用新加的 AUTOMATION_* 配置，也能复用阶段1-4
|
||
# 已落地的集成配置，避免重复维护两套配置源。
|
||
# =============================================================================
|
||
|
||
from __future__ import annotations
|
||
|
||
import logging
|
||
from typing import Any, Callable, Optional
|
||
|
||
from sqlalchemy.ext.asyncio import AsyncSession
|
||
|
||
from app.config import settings
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
async def build_huorong_client(
|
||
db: AsyncSession, audit: Optional[Callable[..., Any]] = None
|
||
):
|
||
"""构建火绒客户端：settings 优先，否则 system_configs。"""
|
||
from app.integrations.huorong.client import HuorongClient
|
||
|
||
if (
|
||
settings.automation_huorong_base_url
|
||
and settings.automation_huorong_access_key_id
|
||
and settings.automation_huorong_access_key_secret
|
||
):
|
||
return HuorongClient(
|
||
access_key_id=settings.automation_huorong_access_key_id,
|
||
access_key_secret=settings.automation_huorong_access_key_secret,
|
||
base_url=settings.automation_huorong_base_url,
|
||
)
|
||
from app.integrations.huorong.config import get_huorong_client
|
||
|
||
return await get_huorong_client(db)
|
||
|
||
|
||
async def build_lianruan_client(
|
||
db: AsyncSession, audit: Optional[Callable[..., Any]] = None
|
||
):
|
||
"""构建联软客户端：settings 优先，否则 system_configs。"""
|
||
from app.integrations.lianruan.client import LianruanClient
|
||
|
||
if (
|
||
settings.automation_lianruan_base_url
|
||
and settings.automation_lianruan_api_account
|
||
and settings.automation_lianruan_api_password
|
||
):
|
||
return LianruanClient(
|
||
base_url=settings.automation_lianruan_base_url,
|
||
api_account=settings.automation_lianruan_api_account,
|
||
api_password=settings.automation_lianruan_api_password,
|
||
validate_key=settings.automation_lianruan_validate_key,
|
||
)
|
||
from app.integrations.lianruan.config import get_lianruan_client
|
||
|
||
return await get_lianruan_client(db)
|
||
|
||
|
||
async def build_ragflow_client(
|
||
db: AsyncSession, audit: Optional[Callable[..., Any]] = None
|
||
):
|
||
"""构建 RAGFlow 客户端：settings 优先，否则 system_configs。"""
|
||
from app.integrations.ragflow.client import RagflowClient
|
||
|
||
if settings.automation_ragflow_base_url and settings.automation_ragflow_api_key:
|
||
return RagflowClient(
|
||
api_key=settings.automation_ragflow_api_key,
|
||
base_url=settings.automation_ragflow_base_url,
|
||
)
|
||
from app.integrations.ragflow.config import get_ragflow_client
|
||
|
||
return await get_ragflow_client(db)
|
||
|
||
|
||
async def build_dify_client(audit: Optional[Callable[..., Any]] = None):
|
||
"""构建 Dify 客户端（仅 settings）。"""
|
||
from app.integrations.dify import get_dify_client
|
||
|
||
return await get_dify_client(audit=audit)
|
||
|
||
|
||
async def build_ehr_client(audit: Optional[Callable[..., Any]] = None):
|
||
"""构建 EHR 客户端（仅 settings）。"""
|
||
from app.integrations.ehr import get_ehr_client
|
||
|
||
return await get_ehr_client(audit=audit)
|