23 lines
624 B
Python
23 lines
624 B
Python
"""Pipeline task model — thin wrapper, main state is now ReportState."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from app.graph.state import ReportState
|
|
|
|
|
|
def create_report_state(
|
|
requirement: str,
|
|
report_type: str = "行业分析报告",
|
|
extra_data: str = "",
|
|
output_formats: list[str] | None = None,
|
|
client_id: str | None = None,
|
|
) -> ReportState:
|
|
"""Create a ReportState from user input."""
|
|
return ReportState(
|
|
requirement=requirement,
|
|
report_type=report_type,
|
|
extra_data=extra_data,
|
|
output_formats=output_formats or ["docx"],
|
|
client_id=client_id,
|
|
)
|