mirror of
https://github.com/run-llama/template-workflow-data-extraction.git
synced 2026-07-01 21:34:19 -04:00
Adrian Lyjak
23 lines
671 B
Python
23 lines
671 B
Python
import logging
|
|
|
|
from llama_index.core.workflow import StartEvent, StopEvent, Workflow, step
|
|
from llama_index.core.workflow.retry_policy import ConstantDelayRetryPolicy
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class FileEvent(StartEvent):
|
|
file_path: str
|
|
|
|
|
|
class ProcessFileWorkflow(Workflow):
|
|
"""
|
|
Given a file path, this workflow will process a single file through the custom extraction logic.
|
|
"""
|
|
|
|
@step(retry_policy=ConstantDelayRetryPolicy(maximum_attempts=3, delay=10))
|
|
async def run_file(self, event: FileEvent) -> StopEvent:
|
|
logger.info(f"Processing file {event.file_path}")
|
|
# TODO: process the file
|
|
return StopEvent()
|