JiuContinent
aa81fc5321
feat(ai_agent): 完善AI适配器和服务功能 - 添加OpenClaw和Hermes健康检查接口 - 实现OpenClaw工具调用功能 - 重构AI适配器使用常量定义 - 增加AI技能系统支持 - 更新配置文件中的默认模型提供者设置 refactor(scheduler): 使用常量替换硬编码值 - 将硬编码的actor值替换为ActorValue常量 - 将receive_id_type替换为FeishuReceiveIdType枚举 refactor(audit): 统一审计日志常量使用 - 将硬编码的actor、source、risk_level等值替换为对应常量 - 更新审核服务中的状态和操作常量引用 refactor(approvals): 标准化审批模块常量使用 - 将applicant默认值替换为ActorValue.API常量 - 使用ApprovalStatus常量替代硬编码状态值 - 更新审核操作常量引用 ```
60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
from fastapi import APIRouter, Depends
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.core.constants import ActorValue
|
|
from app.core.database import get_db
|
|
from app.core.security import require_api_key
|
|
from app.modules.ai_agent.schemas import (
|
|
AIAskRequest,
|
|
AIAskResponse,
|
|
DraftPolicyRequest,
|
|
InvestmentResearchRequest,
|
|
OpenClawToolInvokeRequest,
|
|
)
|
|
from app.modules.ai_agent.service import AIService
|
|
|
|
router = APIRouter(dependencies=[Depends(require_api_key)])
|
|
|
|
|
|
@router.post("/ask", response_model=AIAskResponse)
|
|
def ask(payload: AIAskRequest, db: Session = Depends(get_db)) -> dict:
|
|
return AIService(db).ask(payload.prompt, payload.context, payload.actor, payload.source)
|
|
|
|
|
|
@router.get("/provider-health")
|
|
def provider_health(actor: str = ActorValue.API, db: Session = Depends(get_db)) -> dict:
|
|
return AIService(db).provider_health(actor=actor)
|
|
|
|
|
|
@router.post("/openclaw/tools/invoke")
|
|
def invoke_openclaw_tool(
|
|
payload: OpenClawToolInvokeRequest,
|
|
db: Session = Depends(get_db),
|
|
) -> dict:
|
|
return AIService(db).invoke_openclaw_tool(
|
|
tool=payload.tool,
|
|
action=payload.action,
|
|
args=payload.args,
|
|
session_key=payload.session_key,
|
|
actor=payload.actor,
|
|
)
|
|
|
|
|
|
@router.post("/draft-policy", response_model=AIAskResponse)
|
|
def draft_policy(payload: DraftPolicyRequest, db: Session = Depends(get_db)) -> dict:
|
|
return AIService(db).draft_policy(
|
|
title=payload.title,
|
|
policy_type=payload.policy_type,
|
|
requirements=payload.requirements,
|
|
actor=payload.actor,
|
|
)
|
|
|
|
|
|
@router.post("/investment-research", response_model=AIAskResponse)
|
|
def investment_research(payload: InvestmentResearchRequest, db: Session = Depends(get_db)) -> dict:
|
|
return AIService(db).draft_investment_research(
|
|
symbol_or_topic=payload.symbol_or_topic,
|
|
risk_preference=payload.risk_preference,
|
|
actor=payload.actor,
|
|
)
|