155 lines
4.8 KiB
Python
155 lines
4.8 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
文字内容API路由
|
||
"""
|
||
|
||
import logging
|
||
from fastapi import APIRouter, Depends, HTTPException
|
||
from typing import List, Dict, Any
|
||
|
||
from core.config import ConfigManager
|
||
from core.ai import AIAgent
|
||
from utils.file_io import OutputManager
|
||
from api.services.tweet import TweetService
|
||
from api.models.tweet import (
|
||
TopicRequest, TopicResponse,
|
||
ContentRequest, ContentResponse,
|
||
JudgeRequest, JudgeResponse,
|
||
PipelineRequest, PipelineResponse
|
||
)
|
||
|
||
# 从依赖注入模块导入依赖
|
||
from api.dependencies import get_config, get_ai_agent, get_output_manager
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
# 创建路由
|
||
router = APIRouter()
|
||
|
||
# 依赖注入函数
|
||
def get_tweet_service(
|
||
config_manager: ConfigManager = Depends(get_config),
|
||
ai_agent: AIAgent = Depends(get_ai_agent),
|
||
output_manager: OutputManager = Depends(get_output_manager)
|
||
) -> TweetService:
|
||
"""获取文字内容服务"""
|
||
return TweetService(ai_agent, config_manager, output_manager)
|
||
|
||
|
||
@router.post("/topics", response_model=TopicResponse, summary="生成选题")
|
||
async def generate_topics(
|
||
request: TopicRequest,
|
||
tweet_service: TweetService = Depends(get_tweet_service)
|
||
):
|
||
"""
|
||
生成选题
|
||
|
||
- **date**: 选题日期，格式为YYYY-MM-DD
|
||
- **num_topics**: 要生成的选题数量
|
||
- **style**: 内容风格，如'旅游攻略'、'亲子游'等
|
||
- **target_audience**: 目标受众，如'年轻人'、'家庭'等
|
||
"""
|
||
try:
|
||
request_id, topics = await tweet_service.generate_topics(
|
||
date=request.date,
|
||
num_topics=request.num_topics,
|
||
style=request.style,
|
||
target_audience=request.target_audience
|
||
)
|
||
|
||
return TopicResponse(
|
||
request_id=request_id,
|
||
topics=topics
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"生成选题失败: {e}", exc_info=True)
|
||
raise HTTPException(status_code=500, detail=f"生成选题失败: {str(e)}")
|
||
|
||
|
||
@router.post("/content", response_model=ContentResponse, summary="生成内容")
|
||
async def generate_content(
|
||
request: ContentRequest,
|
||
tweet_service: TweetService = Depends(get_tweet_service)
|
||
):
|
||
"""
|
||
为选题生成内容
|
||
|
||
- **topic**: 选题信息
|
||
"""
|
||
try:
|
||
request_id, topic_index, content = await tweet_service.generate_content(
|
||
topic=request.topic
|
||
)
|
||
|
||
return ContentResponse(
|
||
request_id=request_id,
|
||
topic_index=topic_index,
|
||
content=content
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"生成内容失败: {e}", exc_info=True)
|
||
raise HTTPException(status_code=500, detail=f"生成内容失败: {str(e)}")
|
||
|
||
|
||
@router.post("/judge", response_model=JudgeResponse, summary="审核内容")
|
||
async def judge_content(
|
||
request: JudgeRequest,
|
||
tweet_service: TweetService = Depends(get_tweet_service)
|
||
):
|
||
"""
|
||
审核内容
|
||
|
||
- **topic**: 选题信息
|
||
- **content**: 要审核的内容
|
||
"""
|
||
try:
|
||
request_id, topic_index, judged_content, judge_success = await tweet_service.judge_content(
|
||
topic=request.topic,
|
||
content=request.content
|
||
)
|
||
|
||
return JudgeResponse(
|
||
request_id=request_id,
|
||
topic_index=topic_index,
|
||
judged_content=judged_content,
|
||
judge_success=judge_success
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"审核内容失败: {e}", exc_info=True)
|
||
raise HTTPException(status_code=500, detail=f"审核内容失败: {str(e)}")
|
||
|
||
|
||
@router.post("/pipeline", response_model=PipelineResponse, summary="运行完整流水线")
|
||
async def run_pipeline(
|
||
request: PipelineRequest,
|
||
tweet_service: TweetService = Depends(get_tweet_service)
|
||
):
|
||
"""
|
||
运行完整流水线，包括生成选题、生成内容和审核内容
|
||
|
||
- **date**: 选题日期，格式为YYYY-MM-DD
|
||
- **num_topics**: 要生成的选题数量
|
||
- **style**: 内容风格，如'旅游攻略'、'亲子游'等
|
||
- **target_audience**: 目标受众，如'年轻人'、'家庭'等
|
||
- **skip_judge**: 是否跳过内容审核步骤
|
||
"""
|
||
try:
|
||
request_id, topics, contents, judged_contents = await tweet_service.run_pipeline(
|
||
date=request.date,
|
||
num_topics=request.num_topics,
|
||
style=request.style,
|
||
target_audience=request.target_audience,
|
||
skip_judge=request.skip_judge
|
||
)
|
||
|
||
return PipelineResponse(
|
||
request_id=request_id,
|
||
topics=topics,
|
||
contents=contents,
|
||
judged_contents=judged_contents
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"运行流水线失败: {e}", exc_info=True)
|
||
raise HTTPException(status_code=500, detail=f"运行流水线失败: {str(e)}") |