Agent Zero
7f61dfb25b
Check Cross-Plugin Imports / check (push) Has been cancelled
Migration 0137: Drop AI chat tables (ai_chat_sessions, ai_chat_messages, ai_chat_attachments, ai_conversations, ai_messages)
Backend:
- Remove AIChatSession, AIChatMessage, AIChatAttachment models from ai_assistant/models.py
- Remove AIConversation, AIMessage from app/models/__init__.py
- Remove session/message/stream/attachment routes from ai_assistant/routes.py
- Add new streaming route POST /ai/conversations/{conversation_id}/stream using comm tables
- Add new messages route GET /ai/conversations/{conversation_id}/messages using comm tables
- Add stream_chat_comm, get_comm_messages, save_comm_message to services.py
- Update external_api.py to use CommConversation/CommMessage instead of AIChatSession/AIChatMessage
- Update unified_search ai_chat_provider to search comm_messages with conversation_type=ai
- Remove ai_copilot router from main.py and routes/__init__.py
- Remove ai_conversation from entity_permissions.py and owner_transfer_service.py
- Update ai_assistant/plugin.py get_entity_models to remove AIChatSession
- Guard ai_copilot_service.py imports with try/except
Frontend:
- Remove AIAssistant.tsx, AIAssistantStandalone.tsx, SessionList.tsx, ChatWindow.tsx
- Remove AI Assistant routes from routes/index.tsx
- Update api/ai.ts: streamChat uses /ai/conversations/{id}/stream, fetchMessages uses /ai/conversations/{id}/messages
- Update Communication.tsx: use convId for AI streaming, remove aiSessionId, use fetchAiMessages for AI conversations
- Update AiChatPanel.tsx: create comm conversation instead of AI session, use new fetchMessages
- Update AISidebar.tsx: remove ChatWindow import, show placeholder
tsc clean, build successful, backend import OK
96 lines
2.7 KiB
Python
96 lines
2.7 KiB
Python
"""Service for bulk-transferring ownership of records between users."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
import uuid
|
|
|
|
from sqlalchemy import text
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.core.audit import log_audit
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# Mapping of entity_type -> database table name
|
|
ENTITY_TABLES: dict[str, str] = {
|
|
"contacts": "contacts",
|
|
"addresses": "addresses",
|
|
"attachments": "attachments",
|
|
"bank_accounts": "bank_accounts",
|
|
"workflows": "workflows",
|
|
"sequences": "sequences",
|
|
"saved_filters": "saved_filters",
|
|
"saved_views": "saved_views",
|
|
"webhooks": "webhooks",
|
|
"notifications": "notifications",
|
|
}
|
|
|
|
|
|
async def transfer_ownership(
|
|
db: AsyncSession,
|
|
tenant_id: uuid.UUID,
|
|
from_user_id: uuid.UUID,
|
|
to_user_id: uuid.UUID,
|
|
entity_types: list[str] | None = None,
|
|
) -> dict[str, int]:
|
|
"""Bulk-transfer all records from one user to another for the given entity types.
|
|
|
|
Args:
|
|
db: Database session.
|
|
tenant_id: Tenant scope.
|
|
from_user_id: Current owner whose records will be transferred.
|
|
to_user_id: New owner for the records.
|
|
entity_types: List of entity types to transfer. If None, all known types.
|
|
|
|
Returns:
|
|
Dict mapping entity_type -> number of records transferred.
|
|
"""
|
|
if entity_types is None:
|
|
entity_types = list(ENTITY_TABLES.keys())
|
|
|
|
results: dict[str, int] = {}
|
|
|
|
for entity_type in entity_types:
|
|
table = ENTITY_TABLES.get(entity_type)
|
|
if table is None:
|
|
logger.warning("Unknown entity_type=%s, skipping", entity_type)
|
|
continue
|
|
|
|
# Build and execute the UPDATE
|
|
stmt = text(
|
|
f"UPDATE {table} SET owner_id = :to_user_id "
|
|
f"WHERE owner_id = :from_user_id AND tenant_id = :tenant_id"
|
|
)
|
|
stmt = stmt.bindparams(
|
|
to_user_id=str(to_user_id),
|
|
from_user_id=str(from_user_id),
|
|
tenant_id=str(tenant_id),
|
|
)
|
|
result = await db.execute(stmt)
|
|
count = result.rowcount
|
|
results[entity_type] = count if count is not None else 0
|
|
|
|
if count and count > 0:
|
|
logger.info(
|
|
"Transferred %d %s from user %s to user %s (tenant %s)",
|
|
count, entity_type, from_user_id, to_user_id, tenant_id,
|
|
)
|
|
|
|
# Log the transfer in audit log
|
|
await log_audit(
|
|
db,
|
|
tenant_id,
|
|
to_user_id,
|
|
"transfer_ownership",
|
|
"ownership",
|
|
changes={
|
|
"from_user_id": str(from_user_id),
|
|
"to_user_id": str(to_user_id),
|
|
"entity_types": entity_types,
|
|
"results": results,
|
|
},
|
|
)
|
|
|
|
return results
|