Skip to content

response_rag_stage

ResponseRagStage

Bases: BaseRagStage

Source code in griptape/engines/rag/stages/response_rag_stage.py 
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
@define(kw_only=True)
class ResponseRagStage(BaseRagStage):
    response_modules: list[BaseResponseRagModule] = field()

    @property
    def modules(self) -> list[BaseRagModule]:
        ms = []

        ms.extend(self.response_modules)

        return ms

    def run(self, context: RagContext) -> RagContext:
        logging.info("ResponseRagStage: running %s retrieval modules in parallel", len(self.response_modules))

        with self.create_futures_executor() as futures_executor:
            results = utils.execute_futures_list(
                [futures_executor.submit(with_contextvars(r.run), context) for r in self.response_modules]
            )

        context.outputs = results

        return context

modules property

response_modules = field() class-attribute instance-attribute

run(context)

Source code in griptape/engines/rag/stages/response_rag_stage.py 
32
33
34
35
36
37
38
39
40
41
42
def run(self, context: RagContext) -> RagContext:
    logging.info("ResponseRagStage: running %s retrieval modules in parallel", len(self.response_modules))

    with self.create_futures_executor() as futures_executor:
        results = utils.execute_futures_list(
            [futures_executor.submit(with_contextvars(r.run), context) for r in self.response_modules]
        )

    context.outputs = results

    return context