mjalmoudy94andClaude Sonnet 5.5
dee2f72c46
Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com>
87 lines
3.0 KiB
Python
87 lines
3.0 KiB
Python
"""Wiring for the org-platform workstream (memory, model grants).
|
|
|
|
``build_memory`` is the frozen ``MemoryPort`` builder the engine calls; it now
|
|
returns the real ``MemoryService``. ``build_memory_service`` is the same object,
|
|
typed for the router (list, teach, pin, confirm, delete). ``build_model_grants``
|
|
returns the real ``ModelGrantService``; ``build_model_grant_service`` is its
|
|
typed twin for the admin router.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.application.ports.memory import MemoryPort
|
|
from app.application.ports.model_grants import ModelGrantPort
|
|
from app.application.services.memory_service import MemoryService
|
|
from app.application.services.model_grant_service import ModelGrantService
|
|
from app.infrastructure.repositories.ai_control import (
|
|
AiModelRepository,
|
|
ModelCredentialRepository,
|
|
RateCardRepository,
|
|
)
|
|
from app.infrastructure.repositories.audit import UsageLedgerRepository
|
|
from app.infrastructure.repositories.memory import MemoryRepository
|
|
from app.infrastructure.repositories.org_grants import (
|
|
DefaultModelGrantRepository,
|
|
OrgModelGrantRepository,
|
|
OrgPolicyRepository,
|
|
)
|
|
|
|
|
|
def build_memory(session: AsyncSession) -> MemoryPort:
|
|
"""The real memory service: recall, teach, and one reflection per task.
|
|
|
|
``get_model_provider`` is resolved from ``container``'s namespace rather
|
|
than from the registry directly: that module is what the composition root
|
|
re-exports, so a deployment (or the test suite) that swaps the adapter in
|
|
one place swaps it here too.
|
|
"""
|
|
from app.application.container import get_model_provider
|
|
|
|
return build_memory_service(session, provider=get_model_provider())
|
|
|
|
|
|
def build_memory_service(
|
|
session: AsyncSession,
|
|
*,
|
|
provider=None,
|
|
) -> MemoryService:
|
|
"""The same object, typed for the router (list, teach, pin, confirm, delete)."""
|
|
if provider is None:
|
|
from app.application.container import get_model_provider
|
|
|
|
provider = get_model_provider()
|
|
return MemoryService(
|
|
session=session,
|
|
memories=MemoryRepository(session),
|
|
models=AiModelRepository(session),
|
|
credentials=ModelCredentialRepository(session),
|
|
rate_cards=RateCardRepository(session),
|
|
ledger=UsageLedgerRepository(session),
|
|
provider=provider,
|
|
)
|
|
|
|
|
|
def build_model_grants(session: AsyncSession) -> ModelGrantPort:
|
|
"""The real grant service: an organization sees only what the owner gave it."""
|
|
return build_model_grant_service(session)
|
|
|
|
|
|
def build_model_grant_service(session: AsyncSession) -> ModelGrantService:
|
|
"""The same object, typed for the admin router (grants, defaults, policy)."""
|
|
return ModelGrantService(
|
|
session=session,
|
|
grants=OrgModelGrantRepository(session),
|
|
defaults=DefaultModelGrantRepository(session),
|
|
policies=OrgPolicyRepository(session),
|
|
)
|
|
|
|
|
|
__all__ = [
|
|
"build_memory",
|
|
"build_memory_service",
|
|
"build_model_grant_service",
|
|
"build_model_grants",
|
|
]
|