dependencies.py

from collections.abc import Callable
from typing import Any

from fastapi import BackgroundTasks

from agent_memory_server.config import settings


class DocketBackgroundTasks(BackgroundTasks):
    """A BackgroundTasks implementation that uses Docket."""

    async def add_task(
        self, func: Callable[..., Any], *args: Any, **kwargs: Any
    ) -> None:
        """Run tasks either directly or through Docket"""
        from docket import Docket

        if settings.use_docket:
            async with Docket(
                name=settings.docket_name,
                url=settings.redis_url,
            ) as docket:
                # Schedule task through Docket
                await docket.add(func)(*args, **kwargs)
        else:
            await func(*args, **kwargs)


def get_background_tasks() -> DocketBackgroundTasks:
    """
    Dependency function that returns a DocketBackgroundTasks instance.

    This is used by API endpoints to inject a consistent background tasks object.
    """
    return DocketBackgroundTasks()