Self discover

SelfDiscoverPack

Bases: BaseLlamaPack

Self-Discover Pack.

Source code in llama-index-packs/llama-index-packs-self-discover/llama_index/packs/self_discover/base.py

class SelfDiscoverPack(BaseLlamaPack):
    """Self-Discover Pack."""

    def __init__(
        self,
        llm: Optional[Any] = None,
        verbose: bool = True,
    ) -> None:
        """Init params."""
        self.llm = llm or OpenAI(model="gpt-3.5-turbo")
        self.reasoning_modules = _REASONING_MODULES
        self.verbose = verbose

    def get_modules(self) -> Dict[str, Any]:
        """Get modules."""
        return {"llm": self.llm, "reasoning_modules": self.reasoning_modules}

    def run(self, task):
        """Runs the configured pipeline for a specified task and reasoning modules."""
        configurator = PipelineConfigurator(
            task, self.reasoning_modules, self.verbose, self.llm
        )
        pipeline = configurator.configure()
        return pipeline.run(task=task, reasoning_modules=self.reasoning_modules)

get_modules

get_modules() -> Dict[str, Any]

Get modules.

Source code in llama-index-packs/llama-index-packs-self-discover/llama_index/packs/self_discover/base.py

def get_modules(self) -> Dict[str, Any]:
    """Get modules."""
    return {"llm": self.llm, "reasoning_modules": self.reasoning_modules}

run

run(task)

Runs the configured pipeline for a specified task and reasoning modules.

Source code in llama-index-packs/llama-index-packs-self-discover/llama_index/packs/self_discover/base.py

def run(self, task):
    """Runs the configured pipeline for a specified task and reasoning modules."""
    configurator = PipelineConfigurator(
        task, self.reasoning_modules, self.verbose, self.llm
    )
    pipeline = configurator.configure()
    return pipeline.run(task=task, reasoning_modules=self.reasoning_modules)