rag_task

Suggest edits

Bases: BaseTextInputTask[Union[ListArtifact, ErrorArtifact]]

Source Code in griptape/tasks/rag_task.py
@define
class RagTask(BaseTextInputTask[Union[ListArtifact, ErrorArtifact]]):
    rag_engine: RagEngine = field(kw_only=True, default=Factory(lambda: RagEngine()))

    def try_run(self) -> ListArtifact | ErrorArtifact:
        outputs = self.rag_engine.process_query(self.input.to_text()).outputs

        if len(outputs) > 0:
            return ListArtifact(outputs)
        return ErrorArtifact("empty output")
  • rag_engine = field(kw_only=True, default=Factory(lambda: RagEngine())) class-attribute instance-attribute

try_run()

Source Code in griptape/tasks/rag_task.py
def try_run(self) -> ListArtifact | ErrorArtifact:
    outputs = self.rag_engine.process_query(self.input.to_text()).outputs

    if len(outputs) > 0:
        return ListArtifact(outputs)
    return ErrorArtifact("empty output")
← Prev

prompt_task

↑ Up

tasks

Next →

structure_run_task

Could this page be better? Report a problem or suggest an addition!