| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172 |
- # -*- coding: utf-8 -*-
- from fastapi import FastAPI, Form, HTTPException, Request,status,APIRouter,Query,Path, Depends
- from tools.loglog import logger,log_err_e
- from core.respone_format import *
- from gpt.get_article2 import GetArticle
- from pydantic import BaseModel, ValidationError, conint,Field,conint
- from typing import List, Optional,Literal
- import asyncio
- router = APIRouter()
- async def get_article_dependency():
- article_generator = GetArticle()
- try:
- yield article_generator
- finally:
- await article_generator.cleanup()
- class Word(BaseModel):
- meaning_id:int = Field(..., description="单词的词义id")
- word_id:int = Field(..., description="单词id")
- spell: str = Field(..., description="单词的拼写")
- meaning: str = Field(..., description="单词的意思")
- class ArticleRequest(BaseModel):
- core_words: List[Word] = Field(..., description="单词列表")
- take_count: int = 2
- demo_name: Optional[str] = "无"
- reading_level: conint(ge=1, le=30) = Field(default=10, description="阅读水平，默认值为10;[8,16,24]小学初中高中")
- article_length:int = Field(default=None,description="需要生成的文章长度，可以不传，不传自己根据reading_level判断")
- exercise_id:int = Field(default=0,description="学案ID，用于日志快速定位")
- @router.post("/article/reading-comprehension")
- async def post_article(
- json_data:ArticleRequest,
- request:Request,
- get_article: GetArticle = Depends(get_article_dependency)
- ):
- """
- 异步处理文章生成请求
- """
- json_data = json_data.model_dump()
- real_ip = request.headers.get("X-Real-IP","0.0.0.0")
- core_words = json_data["core_words"]
- take_count = json_data["take_count"]
- demo_name = json_data["demo_name"]
- reading_level = json_data["reading_level"]
- article_length = json_data["article_length"]
- exercise_id = json_data["exercise_id"]
- try:
-
- r = await get_article.submit_task(
- real_ip=real_ip,
- core_words=core_words,
- take_count=take_count,
- demo_name=demo_name,
- reading_level=reading_level,
- article_length=article_length,
- exercise_id=exercise_id
- )
- return r if not isinstance(r,str) else resp_500(message=r)
- except Exception as e:
- log_err_e(e, msg="文章2接口错误/article/reading-comprehension；")
- return resp_500(message=f"{type(e).__name__},{e}")
|
