feat(agent): migrate to native CrewAI tool loop and async resume enqueue
This commit is contained in:
@@ -1,13 +1,16 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import re
|
||||
from typing import TYPE_CHECKING, Protocol
|
||||
from typing import TYPE_CHECKING, Protocol, cast
|
||||
from uuid import UUID
|
||||
|
||||
from fastapi import HTTPException
|
||||
from sqlalchemy.exc import SQLAlchemyError
|
||||
|
||||
from core.auth.models import CurrentUser
|
||||
from core.agent.infrastructure.persistence.user_context_cache import (
|
||||
create_user_context_cache,
|
||||
)
|
||||
from core.db.base_service import BaseService
|
||||
from core.logging import get_logger
|
||||
from v1.users.repository import UserRepository
|
||||
@@ -31,6 +34,10 @@ class AuthByEmailGateway(Protocol):
|
||||
async def get_user_by_email(self, email: str) -> "UserByEmailResponse": ...
|
||||
|
||||
|
||||
class UserContextInvalidator(Protocol):
|
||||
async def invalidate_user(self, *, user_id: UUID) -> int: ...
|
||||
|
||||
|
||||
class AuthLookupAdapter:
|
||||
def __init__(self, gateway: AuthByEmailGateway) -> None:
|
||||
self._gateway = gateway
|
||||
@@ -55,6 +62,7 @@ class UserService(BaseService):
|
||||
_repository: UserRepository
|
||||
_session: AsyncSession
|
||||
_auth_gateway: AuthLookupGateway | None
|
||||
_user_context_cache: UserContextInvalidator
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
@@ -62,11 +70,16 @@ class UserService(BaseService):
|
||||
session: AsyncSession,
|
||||
current_user: CurrentUser | None,
|
||||
auth_gateway: AuthLookupGateway | None = None,
|
||||
user_context_cache: UserContextInvalidator | None = None,
|
||||
) -> None:
|
||||
super().__init__(current_user=current_user)
|
||||
self._repository = repository
|
||||
self._session = session
|
||||
self._auth_gateway = auth_gateway
|
||||
self._user_context_cache = cast(
|
||||
UserContextInvalidator,
|
||||
user_context_cache or create_user_context_cache(),
|
||||
)
|
||||
|
||||
async def get_me(self) -> UserResponse:
|
||||
user_id = self.require_user_id()
|
||||
@@ -109,6 +122,15 @@ class UserService(BaseService):
|
||||
if user is None:
|
||||
raise HTTPException(status_code=404, detail="User not found")
|
||||
|
||||
try:
|
||||
await self._user_context_cache.invalidate_user(user_id=user_id)
|
||||
except Exception as exc:
|
||||
logger.warning(
|
||||
"Failed to invalidate user context cache after profile update",
|
||||
user_id=str(user_id),
|
||||
error=str(exc),
|
||||
)
|
||||
|
||||
return UserResponse(
|
||||
id=str(user.id),
|
||||
username=user.username,
|
||||
|
||||
Reference in New Issue
Block a user