115 lines
3.3 KiB
Python
115 lines
3.3 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
"""InterleaveThinker workflow helpers for FastVideo."""
|
|
|
|
from fastvideo.workflow.interleave_thinker.config import (
|
|
InterleaveCriticConfig,
|
|
InterleaveImageBackendConfig,
|
|
InterleavePlannerConfig,
|
|
InterleaveRunConfig,
|
|
InterleaveRunStateConfig,
|
|
load_interleave_run_config,
|
|
resolve_interleave_instruction,
|
|
)
|
|
from fastvideo.workflow.interleave_thinker.evaluation import (
|
|
InterleavePromptItem,
|
|
InterleavePromptResult,
|
|
InterleavePromptSetSummary,
|
|
load_interleave_prompt_set,
|
|
prompt_set_summary_to_dict,
|
|
run_interleave_prompt_set,
|
|
run_interleave_prompt_set_config,
|
|
save_prompt_set_summary,
|
|
)
|
|
from fastvideo.workflow.interleave_thinker.generator import (
|
|
FastVideoImageGeneratorBackend,
|
|
ImageGeneratorBackend,
|
|
)
|
|
from fastvideo.workflow.interleave_thinker.orchestrator import (
|
|
AcceptAllCritic,
|
|
CriticProvider,
|
|
InterleaveOrchestrator,
|
|
PlannerProvider,
|
|
SinglePromptPlanner,
|
|
)
|
|
from fastvideo.workflow.interleave_thinker.providers import (
|
|
InterleaveThinkerCriticProvider,
|
|
InterleaveThinkerPlannerProvider,
|
|
)
|
|
from fastvideo.workflow.interleave_thinker.runner import (
|
|
InterleaveRunResult,
|
|
run_interleave_config,
|
|
)
|
|
from fastvideo.workflow.interleave_thinker.schema import (
|
|
CriticDecision,
|
|
CriticInput,
|
|
GeneratedImage,
|
|
InterleaveAttempt,
|
|
InterleaveEditRequest,
|
|
InterleaveEditResponse,
|
|
InterleaveTrace,
|
|
PlannedInterleaveStep,
|
|
PlannerInput,
|
|
)
|
|
from fastvideo.workflow.interleave_thinker.trace import (
|
|
save_trace,
|
|
trace_to_dict,
|
|
)
|
|
from fastvideo.workflow.interleave_thinker.trace_eval import (
|
|
InterleaveTraceEvaluationSummary,
|
|
InterleaveTraceMetrics,
|
|
discover_interleave_trace_paths,
|
|
evaluate_interleave_traces,
|
|
interleave_trace_evaluation_to_dict,
|
|
load_interleave_trace_metrics,
|
|
write_interleave_trace_evaluation,
|
|
write_interleave_trace_html_report,
|
|
)
|
|
|
|
__all__ = [
|
|
"AcceptAllCritic",
|
|
"CriticDecision",
|
|
"CriticInput",
|
|
"CriticProvider",
|
|
"FastVideoImageGeneratorBackend",
|
|
"GeneratedImage",
|
|
"ImageGeneratorBackend",
|
|
"InterleaveAttempt",
|
|
"InterleaveCriticConfig",
|
|
"InterleaveEditRequest",
|
|
"InterleaveEditResponse",
|
|
"InterleaveImageBackendConfig",
|
|
"InterleaveOrchestrator",
|
|
"InterleavePlannerConfig",
|
|
"InterleavePromptItem",
|
|
"InterleavePromptResult",
|
|
"InterleavePromptSetSummary",
|
|
"InterleaveRunConfig",
|
|
"InterleaveRunResult",
|
|
"InterleaveRunStateConfig",
|
|
"InterleaveThinkerCriticProvider",
|
|
"InterleaveThinkerPlannerProvider",
|
|
"InterleaveTrace",
|
|
"InterleaveTraceEvaluationSummary",
|
|
"InterleaveTraceMetrics",
|
|
"PlannedInterleaveStep",
|
|
"PlannerInput",
|
|
"PlannerProvider",
|
|
"SinglePromptPlanner",
|
|
"discover_interleave_trace_paths",
|
|
"evaluate_interleave_traces",
|
|
"interleave_trace_evaluation_to_dict",
|
|
"load_interleave_prompt_set",
|
|
"load_interleave_run_config",
|
|
"load_interleave_trace_metrics",
|
|
"prompt_set_summary_to_dict",
|
|
"resolve_interleave_instruction",
|
|
"run_interleave_prompt_set",
|
|
"run_interleave_prompt_set_config",
|
|
"run_interleave_config",
|
|
"save_prompt_set_summary",
|
|
"save_trace",
|
|
"trace_to_dict",
|
|
"write_interleave_trace_evaluation",
|
|
"write_interleave_trace_html_report",
|
|
]
|