Class Paper2FigureRequest

dataflow_agent/state.py:285–321

@dataclass
class Paper2FigureRequest(MainRequest):
    gen_fig_model: str = "gemini-2.5-flash-image-preview"
    # gen_fig_model: str = "gemini-3-pro-image-preview"
    sam2_model: str = "models/facebook/sam2.1-hiera-tiny"
    bg_rm_model: str = "models/RMBG-2.0"

    # 新增：用于 wf_pdf2ppt_qwenvl.py 的 VLM 模型
    vlm_model: str = "qwen-vl-ocr-2025-11-20"
    
    # 新增：用于 wf_paper2expfigure.py 的图表相关模型
    chart_model: str = "gpt-4o"
    
    # 新增：用于 wf_paper2figure_image_only.py 的描述生成模型
    fig_desc_model: str = "gpt-5.1"
    
    # 新增：用于 wf_paper2technical.py 的技术路线生成模型
    technical_model: str = "gpt-5.4"
    # 技术路线图模板/配色（可选）
    tech_route_template: str = ""
    tech_route_palette: str = ""

    input_type: str = "PDF"
    input_content: str = ""
    #  科研绘图复杂度    
    figure_complex: str = "hard"
    style: str = "kartoon"

    # PPT的页面数量 
    page_count: int = 10
    # 是否编辑完毕，也就是是否需要重新生成完整的 PPT
    all_edited_down: bool = False

    # pdf2ppt是否使用AI编辑
    use_ai_edit: bool = False

    # paper2ppt的参考图路径：
    ref_img: str = ''

@dataclass
class Paper2FigureState(MainState):