hankin
423baff73b
- Docker bridge 网络隔离（8000 端口封死） - Gunicorn 4 Worker 多进程 - Alembic 数据库迁移基线 - 日志轮转 20m×3 - JWT 密钥 + DB 密码 + CORS 收紧 - 3-2-1 备份链路（NAS + R740-B 冷备） - 连接池 pool_pre_ping + pool_recycle=3600
143 lines
4.6 KiB
Python
143 lines
4.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
Dify BaaS API 客户端
|
|
取代原有的 OllamaClient，所有 AI 调用统一走 Dify 平台 API。
|
|
|
|
Dify 部署地址: http://192.168.1.88
|
|
文档参考: https://docs.dify.ai/guides/application-publishing/developing-with-apis
|
|
"""
|
|
|
|
import logging
|
|
|
|
import httpx
|
|
|
|
from app.core.config import settings
|
|
|
|
logger = logging.getLogger("dify_client")
|
|
|
|
|
|
class DifyClient:
|
|
"""
|
|
Dify 平台 API 客户端（异步）。
|
|
|
|
每个 Dify 应用有独立的 API Key:
|
|
- 日志分析 App → DIFY_LOG_APP_API_KEY
|
|
- 月度报告 App → DIFY_REPORT_APP_API_KEY
|
|
调用时传入对应 key 即可。
|
|
"""
|
|
|
|
def __init__(self, base_url: str = "http://192.168.1.88/v1"):
|
|
self.base_url = base_url.rstrip("/")
|
|
|
|
async def call_text_generator(
|
|
self,
|
|
api_key: str,
|
|
inputs: dict,
|
|
query: str = "",
|
|
) -> str:
|
|
"""
|
|
调用 Dify 文本生成（completion）类应用。
|
|
|
|
:param api_key: Dify App API Key (app-xxx 格式)
|
|
:param inputs: 传入变量字典，键名需与 Dify 后台配置的变量名一致
|
|
:param query: 可选的用户查询文本
|
|
:return: Dify 返回的 answer 文本，失败时返回空字符串
|
|
"""
|
|
url = f"{self.base_url}/completion-messages"
|
|
headers = {
|
|
"Authorization": f"Bearer {api_key}",
|
|
"Content-Type": "application/json",
|
|
}
|
|
payload = {
|
|
"inputs": inputs,
|
|
"query": query,
|
|
"response_mode": "blocking",
|
|
"user": "crm-backend",
|
|
}
|
|
|
|
try:
|
|
async with httpx.AsyncClient(timeout=60.0) as client:
|
|
response = await client.post(url, headers=headers, json=payload)
|
|
|
|
if response.status_code != 200:
|
|
logger.error(
|
|
"Dify API 非 200 响应: status=%d body=%s",
|
|
response.status_code,
|
|
response.text[:500],
|
|
)
|
|
return ""
|
|
|
|
data = response.json()
|
|
answer = data.get("answer", "")
|
|
logger.info(
|
|
"Dify 调用成功: %d chars (key=...%s)",
|
|
len(answer),
|
|
api_key[-6:],
|
|
)
|
|
return answer
|
|
|
|
except httpx.TimeoutException:
|
|
logger.error("Dify API 超时 (60s): url=%s key=...%s", url, api_key[-6:])
|
|
return ""
|
|
|
|
except Exception as e:
|
|
logger.error("Dify API 异常: %s (key=...%s)", e, api_key[-6:], exc_info=True)
|
|
return ""
|
|
|
|
async def call_workflow(
|
|
self,
|
|
api_key: str,
|
|
inputs: dict,
|
|
user: str = "crm-backend",
|
|
) -> dict | str:
|
|
"""
|
|
调用 Dify 工作流（workflow）类应用。
|
|
|
|
:param api_key: Dify App API Key (app-xxx 格式)
|
|
:param inputs: 传入变量字典，键名需与 Dify 后台配置的变量名一致
|
|
:param user: 用户标识
|
|
:return: Dify 工作流返回的 outputs 字典，失败时返回空字符串
|
|
"""
|
|
url = f"{self.base_url}/workflows/run"
|
|
headers = {
|
|
"Authorization": f"Bearer {api_key}",
|
|
"Content-Type": "application/json",
|
|
}
|
|
payload = {
|
|
"inputs": inputs,
|
|
"response_mode": "blocking",
|
|
"user": user,
|
|
}
|
|
|
|
try:
|
|
async with httpx.AsyncClient(timeout=60.0) as client:
|
|
response = await client.post(url, headers=headers, json=payload)
|
|
|
|
if response.status_code != 200:
|
|
logger.error(
|
|
"Dify Workflow 非 200 响应: status=%d body=%s",
|
|
response.status_code,
|
|
response.text[:500],
|
|
)
|
|
return ""
|
|
|
|
data = response.json()
|
|
outputs = data.get("data", {}).get("outputs", {})
|
|
logger.info(
|
|
"Dify Workflow 调用成功: outputs_keys=%s (key=...%s)",
|
|
list(outputs.keys()) if isinstance(outputs, dict) else "N/A",
|
|
api_key[-6:],
|
|
)
|
|
return outputs
|
|
|
|
except httpx.TimeoutException:
|
|
logger.error("Dify Workflow 超时 (60s): url=%s key=...%s", url, api_key[-6:])
|
|
return ""
|
|
|
|
except Exception as e:
|
|
logger.error("Dify Workflow 异常: %s (key=...%s)", e, api_key[-6:], exc_info=True)
|
|
return ""
|
|
|
|
# 全局单例，使用 settings 中配置的 Dify 地址
|
|
dify_client = DifyClient(base_url=settings.DIFY_BASE_URL)
|