Class Paper2FigureRequest

fastapi_app/schemas.py:94–189

Paper2Figure 的请求参数定义。 注意： - 为了兼容 dataflow_agent 内部对 state.request 的访问， 这里额外提供 language 字段，并实现一个简单的 get 方法， 使其既能通过属性访问（.language），也能通过 dict 风格访问（.get）。

class Paper2FigureRequest(BaseModel):
    """
    Paper2Figure 的请求参数定义。

    注意：
    - 为了兼容 dataflow_agent 内部对 state.request 的访问，
      这里额外提供 language 字段，并实现一个简单的 get 方法，
      使其既能通过属性访问（.language），也能通过 dict 风格访问（.get）。
    """

    # ---------------------- 基础 LLM 设置 ----------------------
    language: str = "en"
    # 工作流内部有角色会访问 state.request.language

    chat_api_url: str = settings.DEFAULT_LLM_API_URL
    # 与大模型交互使用的 API URL

    # ---------------------- 图类型 & 难度设置 ----------------------
    figure_complex: str = "easy"
    # 绘图难度：仅在 graph_type == "model_arch" 时生效，前端透传 easy/mid/hard

    resolution: str = "2K"
    # 图像分辨率：2K 或 4K，影响生成图像的质量和生成时间

    chat_api_key: str = "fill the key"
    # chat_api_url 对应的 API KEY；用于访问后端 LLM 服务

    api_key: str = ""
    # 如果使用第三方外部 API（如 OpenAI），在此填写外部 API Key；为空则使用内部服务

    image_api_url: str = ""
    image_api_key: str = ""

    model: str = settings.PAPER2FIGURE_TEXT_MODEL
    # 用于执行理解、抽象、描述生成的文本模型名称

    gen_fig_model: str = settings.PAPER2FIGURE_IMAGE_MODEL
    # 用于生成插图 / 构图草图的图像模型名称

    bg_rm_model: str = f"{get_project_root()}/models/RMBG-2.0"

    # 新增模型参数
    vlm_model: str = settings.PAPER2FIGURE_VLM_MODEL
    tec_vlm_desc_model: str = settings.PAPER2FIGURE_REF_IMG_DESC_MODEL
    chart_model: str = settings.PAPER2FIGURE_CHART_MODEL
    fig_desc_model: str = settings.PAPER2FIGURE_DESC_MODEL
    technical_model: str = settings.PAPER2FIGURE_TECHNICAL_MODEL
    tech_route_template: str = ""
    tech_route_palette: str = ""

    # ---------------------- 输入类型设置 ----------------------
    input_type: Literal["PDF", "TEXT", "FIGURE"] = "PDF"
    # 指定输入内容的形式：
    # - "PDF": 输入为 PDF 文件路径
    # - "TEXT": 输入为纯文本内容
    # - "FIGURE": 输入为图片文件路径（如 JPG/PNG），用于图像解析或转图

    input_content: str = ""