registry

Class info

Classes

Name	Children	Inherits
Job llmling_agent.models.task A task is a piece of work that can be executed by an agent.		BaseModel Generic
JobRegistrationError llmling_agent.tasks.exceptions Task could not get registered.		JobError
TaskRegistry llmling_agent.tasks.registry Registry for managing tasks.		BaseRegistry

🛈 DocStrings

Task definition and registry for agents.

TaskRegistry

Bases: BaseRegistry[str, Job[Any, Any]]

Registry for managing tasks.

Source code in src/llmling_agent/tasks/registry.py

class TaskRegistry(BaseRegistry[str, Job[Any, Any]]):
    """Registry for managing tasks."""

    @property
    def _error_class(self) -> type[JobRegistrationError]:
        return JobRegistrationError

    def _validate_item(self, item: Any) -> Job[Any, Any]:
        if not isinstance(item, Job):
            msg = f"Expected Job, got {type(item)}"
            raise self._error_class(msg)
        return item

    def register(self, name: str, task: Job[Any, Any], replace: bool = False):
        """Register a task with name.

        Creates a copy of the task with the name set.
        """
        task_copy = task.model_copy(update={"name": name})
        super().register(name, task_copy, replace)

register

register(name: str, task: Job[Any, Any], replace: bool = False)

Register a task with name.

Creates a copy of the task with the name set.

Source code in src/llmling_agent/tasks/registry.py

def register(self, name: str, task: Job[Any, Any], replace: bool = False):
    """Register a task with name.

    Creates a copy of the task with the name set.
    """
    task_copy = task.model_copy(update={"name": name})
    super().register(name, task_copy, replace)