Skip to content

extraction_task

ExtractionTask

Bases: BaseTextInputTask[ListArtifact]

Source code in griptape/tasks/extraction_task.py 
14
15
16
17
18
19
20
@define
class ExtractionTask(BaseTextInputTask[ListArtifact]):
    extraction_engine: BaseExtractionEngine = field(kw_only=True)
    args: dict = field(kw_only=True, factory=dict)

    def try_run(self) -> ListArtifact:
        return self.extraction_engine.extract_artifacts(ListArtifact([self.input]), rulesets=self.rulesets, **self.args)

args = field(kw_only=True, factory=dict) class-attribute instance-attribute

extraction_engine = field(kw_only=True) class-attribute instance-attribute

try_run()

Source code in griptape/tasks/extraction_task.py 
19
20
def try_run(self) -> ListArtifact:
    return self.extraction_engine.extract_artifacts(ListArtifact([self.input]), rulesets=self.rulesets, **self.args)