Class Pack

autopack/pack.py:12–146

class Pack(BaseModel):
    class Config:
        arbitrary_types_allowed = True

    ## Required

    # The name of the tool that will be provided to the LLM
    name: ClassVar[str]
    # The description of the tool which is passed to the LLM
    description: ClassVar[str]

    ## Optional

    # Any pip packages this Pack depends on.
    dependencies: ClassVar[Optional[list[str]]] = None
    # A list of Pack IDs needed for this tool to function effectively (e.g. `write_file` depends on `read_file`)
    depends_on: ClassVar[Optional[list[str]]] = None
    # Enhances tool selection by grouping this tool with other tools of the same category
    categories: ClassVar[Optional[list[str]]] = None
    # If this tool has side effects that cannot be undone (e.g. sending an email)
    reversible: ClassVar[bool] = True
    # A Pydantic BaseModel describing the Pack's run arguments
    args_schema: ClassVar[Optional[type[BaseModel]]] = None

    llm: Optional[Callable[[str], str]] = Field(
        None, description="A callable function to call an LLM (string in string out)"
    )
    allm: Union[None, Callable[[str], str], Coroutine[Any, Any, str]] = Field(
        None, description="An asynchronous callable function to call an LLM (string in string out)"
    )
    config: PackConfig = Field(default_factory=PackConfig.global_config)

    def __init__(self, **data):
        super().__init__(**data)
        if not self.config.filesystem_manager:
            self.config.init_filesystem_manager()

        if self.llm and not callable(self.llm):
            raise TypeError(f"LLM object {self.llm} must be callable")

        if self.allm and not iscoroutinefunction(self.allm):
            raise TypeError(f"Async LLM object {self.llm} must be async and callable")

        if self.name is None:
            raise TypeError(f"Class {self.__class__.__name__} must define 'name' as a class variable")
        if self.description is None:
            raise TypeError(f"Class {self.__class__.__name__} must define 'description' as a class variable")

    def run(self, *args, **kwargs) -> str:
        """Execute the _run function of the subclass, verifying the arguments. (Will eventually do callbacks or some
        such)

        Args: **kwargs (dict): The arguments to pass to _run. Each key should be the name of an argument,
        and the value should be the value of the argument.

        Returns: The response from the _run function of the subclass
        """
        try: