Skip to content

Csv extraction task

CsvExtractionTask

Bases: ExtractionTask

Source code in griptape/tasks/csv_extraction_task.py 
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
@define
class CsvExtractionTask(ExtractionTask):
    _extraction_engine: CsvExtractionEngine = field(default=None, kw_only=True, alias="extraction_engine")

    @property
    def extraction_engine(self) -> CsvExtractionEngine:
        if self._extraction_engine is None:
            if self.structure is not None:
                self._extraction_engine = CsvExtractionEngine(
                    prompt_driver=self.structure.config.global_drivers.prompt_driver
                )
            else:
                raise ValueError("Extraction Engine is not set.")
        return self._extraction_engine

    @extraction_engine.setter
    def extraction_engine(self, value: CsvExtractionEngine) -> None:
        self._extraction_engine = value

extraction_engine: CsvExtractionEngine property writable