mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
phernandez
24 lines
675 B
Python
24 lines
675 B
Python
"""Base service class."""
|
|
|
|
from typing import TypeVar, Generic, List, Sequence
|
|
|
|
from basic_memory.repository.repository import Repository
|
|
|
|
T = TypeVar("T", bound=Repository)
|
|
|
|
|
|
class BaseService(Generic[T]):
|
|
"""Base service that takes a repository."""
|
|
|
|
def __init__(self, repository: T):
|
|
"""Initialize service with repository."""
|
|
self.repository = repository
|
|
|
|
async def add(self, model: T) -> T:
|
|
"""Add model to repository."""
|
|
return await self.repository.add(model)
|
|
|
|
async def add_all(self, models: List[T]) -> Sequence[T]:
|
|
"""Add a List of models to repository."""
|
|
return await self.repository.add_all(models)
|