Newer
Older
navi-1 / navi / api / deps.py
@Eugene Sukhodolskiy Eugene Sukhodolskiy on 9 Apr 1 KB Add long-term user memory system
"""FastAPI dependency injection — provides shared singletons to route handlers."""

from typing import Annotated

from fastapi import Depends

from navi.config import settings
from navi.core import (
    Agent,
    BackendRegistry,
    ProfileRegistry,
    SessionStore,
    SqliteSessionStore,
    ToolRegistry,
    build_default_registries,
)
from navi.memory import MemoryStore
from navi.workers import Worker, build_default_workers


_memory_store: MemoryStore = MemoryStore(settings.db_path)
_registries: tuple[ToolRegistry, ProfileRegistry, BackendRegistry] | None = None


def get_memory_store() -> MemoryStore:
    return _memory_store


def get_registries() -> tuple[ToolRegistry, ProfileRegistry, BackendRegistry]:
    global _registries
    if _registries is None:
        _registries = build_default_registries(memory_store=_memory_store)
    return _registries


def get_tool_registry() -> ToolRegistry:
    return get_registries()[0]


def get_profile_registry() -> ProfileRegistry:
    return get_registries()[1]


def get_backend_registry() -> BackendRegistry:
    return get_registries()[2]


_session_store = SqliteSessionStore(settings.db_path)
_workers: list[Worker] = build_default_workers()


def get_session_store() -> SessionStore:
    return _session_store


def get_workers() -> list[Worker]:
    return _workers


def get_agent(
    session_store: Annotated[SessionStore, Depends(get_session_store)],
    profile_registry: Annotated[ProfileRegistry, Depends(get_profile_registry)],
    tool_registry: Annotated[ToolRegistry, Depends(get_tool_registry)],
    backend_registry: Annotated[BackendRegistry, Depends(get_backend_registry)],
) -> Agent:
    return Agent(session_store, profile_registry, tool_registry, backend_registry,
                 workers=get_workers(), memory_store=_memory_store)