JiuContinent
71ca804764
feat: 添加公司AI管理平台基础架构 添加了完整的FastAPI后端项目结构，包括： - 环境配置文件(.env.example)和项目说明文档(README.md、AGENTS.md) - Dockerfile用于容器化部署 - 核心基础设施：配置管理、数据库连接、调度器、安全认证 - 模块化设计：AI代理、审批流程、审计日志、业务台账等功能模块 - 支持多数据库连接(主库和遗留系统只读库) - AI适配器支持OpenClaw、Hermes、OpenAI兼容接口 - 飞书集成、报表生成、风险监控等企业级功能 - 完整的依赖管理和测试指南 ```
39 lines
1.3 KiB
Python
39 lines
1.3 KiB
Python
from fastapi import APIRouter, Depends
|
|
from sqlalchemy.orm import Session
|
|
|
|
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,
|
|
)
|
|
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.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,
|
|
)
|