letschatty 0.4.341__py3-none-any.whl → 0.4.342__py3-none-any.whl

letschatty/models/company/assets/chat_assets.py

@@ -10,7 +10,7 @@ from datetime import datetime
 from zoneinfo import ZoneInfo
 from bson import ObjectId
 import json
-from typing import Dict, Any, Optional, List
+from typing import Dict, Any, Optional
 from letschatty.models.utils.types.serializer_type import SerializerType
 from letschatty.models.company.assets.ai_agents_v2.chatty_ai_mode import ChattyAIMode
 
@@ -67,8 +67,7 @@ class AssignedAssetToChat(BaseModel):
 
 
 class SaleAssignedToChat(AssignedAssetToChat):
-    product_id: Optional[StrObjectId] = Field(default=None)
-    product_ids: List[StrObjectId] = Field(default_factory=list)
+    product_id: StrObjectId = Field(frozen=True)
 
 class ContactPointAssignedToChat(AssignedAssetToChat):
     source_id: StrObjectId = Field(frozen=True)

letschatty/models/company/assets/company_assets.py

@@ -5,8 +5,6 @@ class CompanyAssetType(StrEnum):
     USERS = "users"
     BUSINESS_AREAS = "business_areas"
     FUNNELS = "funnels"
-    FUNNEL_STAGES = "funnel_stages"
-    FUNNEL_MEMBERS = "funnel_members"
     PRODUCTS = "products"
     SALES = "sales"
     TAGS = "tags"

letschatty/models/company/assets/sale.py

@@ -5,7 +5,7 @@ from ...utils.types.identifier import StrObjectId
 
 class Sale(CompanyAssetModel):
     chat_id: StrObjectId
-    product_id: Optional[StrObjectId] = Field(default=None)
+    product_id: StrObjectId
     quantity: int
     total_amount: float
     currency: str
@@ -16,7 +16,7 @@ class Sale(CompanyAssetModel):
     creator_id: StrObjectId
 
 class SaleRequest(BaseModel):
-    product_id: Optional[StrObjectId] = Field(default=None)
+    product_id: StrObjectId
     quantity: int
     creator_id: StrObjectId
     total_amount: float
@@ -36,4 +36,4 @@ class SaleRequest(BaseModel):
             "paid_amount": 100,
             "installments": 1,
             "details": {}
-        }
+        }

letschatty/models/data_base/collection_interface.py

@@ -4,6 +4,7 @@ from typing import TYPE_CHECKING, Dict, List, Generic, TypeVar, Type, Optional,
 from bson.objectid import ObjectId
 from pymongo.collection import Collection
 from pymongo.database import Database
+from motor.motor_asyncio import AsyncIOMotorDatabase, AsyncIOMotorCollection
 
 from letschatty.models.chat.chat import Chat
 from ...models.base_models.chatty_asset_model import ChattyAssetModel, CompanyAssetModel, ChattyAssetPreview
@@ -26,58 +27,116 @@ P = TypeVar('P', bound=ChattyAssetPreview)
 class ChattyAssetCollectionInterface(Generic[T, P], ABC):
     def __init__(self, database: str, collection: str, connection: MongoConnection, type: Type[T], preview_type: Optional[Type[P]] = None):
         logger.info(f"Initializing collection {collection} in database {database}")
+        # Sync database and collection (existing)
         self.db: Database = connection.client[database]
         self.collection: Collection = connection.client[database][collection]
+
+        # NEW: Async database and collection
+        # Store connection reference to ensure we use current event loop
+        self._connection = connection
+        self._database_name = database
+        self._collection_name = collection
+        self._async_db: Optional[AsyncIOMotorDatabase] = None
+        self._async_collection: Optional[AsyncIOMotorCollection] = None
+
         self.type = type
         self.preview_type = preview_type
+
+    @property
+    def async_db(self) -> AsyncIOMotorDatabase:
+        """Get async database, ensuring it uses the current event loop"""
+        # Always ensure connection's async client is using current loop (for Lambda compatibility)
+        self._connection._ensure_async_client_loop()
+        # Recreate database reference to ensure it uses the current client
+        self._async_db = self._connection.async_client[self._database_name]
+        return self._async_db
+
+    @property
+    def async_collection(self) -> AsyncIOMotorCollection:
+        """Get async collection, ensuring it uses the current event loop"""
+        # Always ensure connection's async client is using current loop (for Lambda compatibility)
+        self._connection._ensure_async_client_loop()
+        # Recreate collection reference to ensure it uses the current client
+        self._async_collection = self._connection.async_client[self._database_name][self._collection_name]
+        return self._async_collection
     @abstractmethod
     def create_instance(self, data: dict) -> T:
         """Factory method to create instance from data"""
         pass
 
-    def insert(self, asset: T) -> StrObjectId:
+    # All methods are now async-only for better performance
+    async def insert(self, asset: T) -> StrObjectId:
+        """Async insert operation"""
         if not isinstance(asset, self.type):
             raise ValueError(f"Asset must be of type {self.type.__name__}")
         document = asset.model_dump_json(serializer=SerializerType.DATABASE)
         logger.debug(f"Inserting document: {document}")
-        result = self.collection.insert_one(document)
+        result = await self.async_collection.insert_one(document)
         if not result.inserted_id:
             raise Exception("Failed to insert document")
         logger.debug(f"Inserted document with id {result.inserted_id}")
         return result.inserted_id
 
-    def update(self, asset: T) -> StrObjectId:
+    async def update(self, asset: T) -> StrObjectId:
+        """Async update operation"""
         logger.debug(f"Updating document with id {asset.id}")
         if not isinstance(asset, self.type):
             raise ValueError(f"Asset must be of type {self.type.__name__}")
         asset.update_now()
         document = asset.model_dump_json(serializer=SerializerType.DATABASE)
-        document.pop('_id', None)  # Still needed
-        result = self.collection.update_one({"_id": ObjectId(asset.id)}, {"$set": document})
+        document.pop('_id', None)
+        result = await self.async_collection.update_one(
+            {"_id": ObjectId(asset.id)},
+            {"$set": document}
+        )
         if result.matched_count == 0:
             raise NotFoundError(f"No document found with id {asset.id}")
         if result.modified_count == 0:
             logger.debug(f"No changes were made to the document with id {asset.id} probably because the values were the same")
         return asset.id
 
-    def get_by_id(self, doc_id: str) -> T:
-        logger.debug(f"Getting document with id {doc_id} from collection {self.collection.name} and db {self.db.name}")
-        doc = self.collection.find_one({"_id": ObjectId(doc_id)})
-
+    async def get_by_id(self, doc_id: str) -> T:
+        """Get by ID operation"""
+        logger.debug(f"Getting document with id {doc_id} from collection {self.async_collection.name}")
+        doc = await self.async_collection.find_one({"_id": ObjectId(doc_id)})
         if doc:
             return self.create_instance(doc)
         else:
-            raise NotFoundError(f"No document found with id {doc_id} in db collection {self.collection.name} and db {self.db.name}")
+            raise NotFoundError(f"No document found with id {doc_id} in collection")
 
-    def get_docs(self, company_id:Optional[StrObjectId], query = {}, limit = 0) -> List[T]:
-        logger.debug(f"Getting documents from collection {self.collection.name} with company_id {company_id} and query {query}")
+    async def get_docs(self, company_id: Optional[StrObjectId], query={}, limit=0) -> List[T]:
+        """Get multiple documents operation"""
+        logger.debug(f"Getting documents from collection with company_id {company_id} and query {query}")
         if company_id:
-            query = query.copy()  # Create a copy to avoid modifying the original
+            query = query.copy()
             query["company_id"] = company_id
-        docs = list(self.collection.find(filter=query).limit(limit))
-        logger.debug(f"Found {len(docs)} documents in collection {self.collection.name}")
+        cursor = self.async_collection.find(filter=query)
+        if limit:
+            cursor = cursor.limit(limit)
+        docs = await cursor.to_list(length=limit if limit > 0 else None)
+        logger.debug(f"Found {len(docs)} documents")
         return [self.create_instance(doc) for doc in docs]
 
+    async def delete(self, doc_id: str, deletion_type: DeletionType = DeletionType.LOGICAL) -> StrObjectId:
+        """Delete operation"""
+        logger.debug(f"Deleting document with id {doc_id} - deletion type: {deletion_type}")
+        if deletion_type == DeletionType.LOGICAL:
+            result = await self.async_collection.update_one(
+                {"_id": ObjectId(doc_id)},
+                {"$set": {"deleted_at": datetime.now(ZoneInfo("UTC")), "updated_at": datetime.now(ZoneInfo("UTC"))}}
+            )
+            if result.modified_count == 0:
+                raise NotFoundError(f"No document found with id {doc_id}")
+            return doc_id
+        elif deletion_type == DeletionType.PHYSICAL:
+            result = await self.async_collection.delete_one({"_id": ObjectId(doc_id)})
+            if result.deleted_count == 0:
+                raise NotFoundError(f"No document found with id {doc_id}")
+            return doc_id
+        else:
+            raise ValueError(f"Invalid deletion type: {deletion_type}")
+
+    # Additional methods - keeping these sync as they're less critical
     def get_preview_docs(self, projection = {}, all=True) -> List[P]:
         """We get the previews of all the documents in the collection for all companies"""
         if not self.preview_type:
@@ -98,18 +157,31 @@ class ChattyAssetCollectionInterface(Generic[T, P], ABC):
         docs = self.collection.find(query)
         return [self.create_instance(doc) for doc in docs]
 
-    def delete(self, doc_id: str, deletion_type : DeletionType = DeletionType.LOGICAL) -> StrObjectId:
-        logger.debug(f"Deleting document with id {doc_id} - deletion type: {deletion_type}")
-        if deletion_type == DeletionType.LOGICAL:
-            result = self.collection.update_one({"_id": ObjectId(doc_id)}, {"$set": {"deleted_at": datetime.now(ZoneInfo("UTC")), "updated_at": datetime.now(ZoneInfo("UTC"))}})
-            if result.modified_count == 0:
-                raise NotFoundError(f"No document found with id {doc_id}")
-            return doc_id
-        elif deletion_type == DeletionType.PHYSICAL:
-            result = self.collection.delete_one({"_id": ObjectId(doc_id)})
-            if result.deleted_count == 0:
-                raise NotFoundError(f"No document found with id {doc_id}")
-            return doc_id
-        else:
-            raise ValueError(f"Invalid deletion type: {deletion_type}")
+    async def get_by_ids(self, ids: List[StrObjectId]) -> List[T]:
+            """
+            Get multiple assets by their IDs in a single query.
+
+            Args:
+                ids: List of asset IDs
+
+            Returns:
+                List of assets objects
+            """
+            if not ids:
+                return []
+
+            # Convert string IDs to ObjectIds
+            object_ids = [ObjectId(id) for id in ids]
+
+            # Query for all filter criteria with matching IDs
+            query = {
+                "_id": {"$in": object_ids},
+                "deleted_at": None
+            }
+
+            # Use the sync collection directly (inherited from ChattyAssetCollectionInterface)
+            docs = await self.async_collection.find(query).to_list(length=None)
+
+            # Create FilterCriteria instances
+            return [self.create_instance(doc) for doc in docs]
 

letschatty/models/data_base/mongo_connection.py

@@ -1,8 +1,14 @@
 from ..base_models.singleton import SingletonMeta
 from pymongo import MongoClient
+from motor.motor_asyncio import AsyncIOMotorClient
 from typing import Optional
+from urllib.parse import quote_plus
 import os
 import atexit
+import asyncio
+import logging
+
+logger = logging.getLogger(__name__)
 
 class MongoConnection(metaclass=SingletonMeta):
     def __init__(
@@ -10,26 +16,103 @@ class MongoConnection(metaclass=SingletonMeta):
         username: Optional[str] = None,
         password: Optional[str] = None,
         uri_base: Optional[str] = None,
-        instance: Optional[str] = None
+        instance: Optional[str] = None,
+        verify_on_init: bool = True
     ):
         self.username = username or os.getenv('MONGO_USERNAME')
         self.password = password or os.getenv('MONGO_PASSWORD')
         self.uri_base = uri_base or os.getenv('MONGO_URI_BASE')
         self.instance = instance or os.getenv('MONGO_INSTANCE_COMPONENT')
-        
+
         if not all([self.username, self.password, self.uri_base, self.instance]):
             raise ValueError("Missing required MongoDB connection parameters")
-            
-        uri = f"{self.uri_base}://{self.username}:{self.password}@{self.instance}.mongodb.net"
+
+        # URL-encode username and password to handle special characters per RFC 3986
+        encoded_username = quote_plus(self.username)
+        encoded_password = quote_plus(self.password)
+
+        uri = f"{self.uri_base}://{encoded_username}:{encoded_password}@{self.instance}.mongodb.net"
+
+        # Sync client (existing)
         self.client = MongoClient(uri)
+
+        # NEW: Async client for async operations
+        # Don't pass io_loop - Motor will automatically use the current event loop
+        # This is important for Lambda where the event loop changes between invocations
+        self.async_client = AsyncIOMotorClient(uri)
+
+        # Verify connections if requested
+        if verify_on_init:
+            try:
+                # Try to get running loop
+                loop = asyncio.get_running_loop()
+                # If we get here, there's a running loop
+                logger.warning(
+                    "Event loop is already running. Skipping connection verification in __init__. "
+                    "Call verify_connection_async() from async context to verify connection."
+                )
+                self._connection_verified = False
+            except RuntimeError:
+                # No running loop, safe to use run_until_complete
+                try:
+                    # Test sync client
+                    self.client.admin.command('ping')
+
+                    # Test async client in sync context
+                    loop = asyncio.new_event_loop()
+                    asyncio.set_event_loop(loop)
+                    loop.run_until_complete(self.async_client.admin.command('ping'))
+                    self._connection_verified = True
+                    loop.close()
+                except Exception as e:
+                    self.client.close()
+                    self.async_client.close()
+                    raise ConnectionError(f"Failed to connect to MongoDB: {str(e)}")
+        else:
+            self._connection_verified = False
+
+        atexit.register(self.close)
+
+    def _ensure_async_client_loop(self):
+        """Ensure async client is using the current event loop (for Lambda compatibility)"""
         try:
-            self.client.admin.command('ping')
+            current_loop = asyncio.get_running_loop()
+            # Check if client's loop is closed or different
+            client_loop = getattr(self.async_client, '_io_loop', None)
+            if client_loop is not None:
+                try:
+                    # Try to check if the loop is closed
+                    if client_loop.is_closed():
+                        # Recreate client with current loop
+                        logger.warning("Async client's event loop is closed, recreating client")
+                        old_client = self.async_client
+                        uri = f"{self.uri_base}://{quote_plus(self.username)}:{quote_plus(self.password)}@{self.instance}.mongodb.net"
+                        self.async_client = AsyncIOMotorClient(uri)
+                        try:
+                            old_client.close()
+                        except:
+                            pass
+                except AttributeError:
+                    # _io_loop might not exist in newer Motor versions
+                    pass
+        except RuntimeError:
+            # No running loop, which is fine - Motor will handle it
+            pass
+
+    async def verify_connection_async(self) -> bool:
+        """Verify MongoDB connection asynchronously. Safe to call from async context."""
+        try:
+            # Ensure we're using the current event loop
+            self._ensure_async_client_loop()
+            await self.async_client.admin.command('ping')
+            self._connection_verified = True
+            return True
         except Exception as e:
-            self.client.close()
-            raise ConnectionError(f"Failed to connect to MongoDB: {str(e)}")
+            logger.error(f"Failed to verify MongoDB connection: {e}")
+            raise ConnectionError(f"Failed to verify MongoDB connection: {e}")
 
-        atexit.register(self.close)
-    
     def close(self) -> None:
         if hasattr(self, 'client'):
             self.client.close()
+        if hasattr(self, 'async_client'):
+            self.async_client.close()

letschatty/models/messages/chatty_messages/schema/chatty_content/content_document.py

@@ -1,6 +1,5 @@
 from pydantic import BaseModel, Field, model_validator, ValidationInfo
 from typing import Optional
-from urllib.parse import urlparse, unquote
 from .content_media import ChattyContentMedia
 
 class ChattyContentDocument(ChattyContentMedia):
@@ -9,6 +8,5 @@ class ChattyContentDocument(ChattyContentMedia):
     @model_validator(mode='before')
     def validate_filename(cls, data: dict, info: ValidationInfo):
         if not data.get("filename") and data.get("url"):
-            parsed = urlparse(data["url"])
-            data["filename"] = unquote(parsed.path.split("/")[-1])
-        return data
+            data["filename"] = data["url"].split("/")[-1]
+        return data

letschatty/models/messages/chatty_messages/schema/chatty_content/content_media.py

@@ -1,6 +1,5 @@
 from pydantic import BaseModel, Field, field_validator, HttpUrl
 from typing import Optional
-from urllib.parse import quote
 class ChattyContentMedia(BaseModel):
     id: Optional[str] = Field(description="Unique identifier for the image. Also known as media_id", default="")
     url: str = Field(description="URL of the media from S3")
@@ -12,9 +11,9 @@ class ChattyContentMedia(BaseModel):
     def validate_url(cls, v):
         if not v:
             raise ValueError("URL is required")
-        encoded = quote(str(v), safe=":/?&=%#")
-        HttpUrl(encoded)
-        return encoded
+        HttpUrl(v)
+        return v
     
     def get_body_or_caption(self) -> str:
         return self.caption
+

letschatty/models/utils/custom_exceptions/custom_exceptions.py

@@ -5,6 +5,7 @@ import json
 from datetime import timedelta
 
 from letschatty.models.utils.definitions import Area
+from pydantic_core.core_schema import custom_error_schema
 logger = logging.getLogger("logger")
 
 class Context(BaseModel):
@@ -229,4 +230,16 @@ class OpenAIError(CustomException):
 
 class NewerAvailableMessageToBeProcessedByAiAgent(CustomException):
     def __init__(self, message="Duplicated incoming message call for ai agent", status_code=400, **context_data):
-        super().__init__(message, status_code=status_code, **context_data)
+        super().__init__(message, status_code=status_code, **context_data)
+
+class MissingAIAgentInChat(CustomException):
+    def __init__(self, message="Missing AI agent in chat", status_code=400, **context_data):
+        super().__init__(message, status_code=status_code, **context_data)
+
+class ChattyAIModeOff(CustomException):
+    def __init__(self, message="Chatty AI agent is in OFF mode", status_code=400, **context_data):
+        super().__init__(message, status_code=status_code, **context_data)
+
+class ChatWithActiveContinuousConversation(CustomException):
+    def __init__(self, message="Chat has active continuous conversation", status_code=400, **context_data):
+        super().__init__(message, status_code=status_code, **context_data)

letschatty/services/ai_agents/smart_follow_up_context_builder_v2.py

@@ -14,13 +14,16 @@ class SmartFollowUpContextBuilder(ContextBuilder):
 
     @staticmethod
     def check_minimum_time_since_last_message(chat: Chat, follow_up_strategy: FollowUpStrategy,smart_follow_up_state: FlowStateAssignedToChat) -> bool:
-        expected_interval_minutes = follow_up_strategy.get_interval_for_followup(smart_follow_up_state.consecutive_count)
+        # consecutive_count is 0-indexed (0 = no follow-ups sent yet), but get_interval_for_followup expects 1-indexed
+        # So we add 1 to get the interval for the follow-up we're about to send
+        next_followup_number = smart_follow_up_state.consecutive_count + 1
+        expected_interval_minutes = follow_up_strategy.get_interval_for_followup(next_followup_number)
         last_message_timestamp = chat.last_message_timestamp
         if last_message_timestamp is None:
             raise HumanInterventionRequired("There's no last message in the chat, can't validate the minimum time since last message for the smart follow up")
         time_since_last_message = datetime.now(ZoneInfo('UTC')) - last_message_timestamp
         if time_since_last_message.total_seconds() < expected_interval_minutes * 60:
-            raise PostponeFollowUp(time_delta= timedelta(seconds=expected_interval_minutes * 60 - time_since_last_message.total_seconds()), message=f"Se pospuso el Smart Follow Up porque no ha pasado el tiempo mínimo esperado de {expected_interval_minutes/60} horas para el seguimiento #{smart_follow_up_state.consecutive_count}")
+            raise PostponeFollowUp(time_delta= timedelta(seconds=expected_interval_minutes * 60 - time_since_last_message.total_seconds()), message=f"Se pospuso el Smart Follow Up porque no ha pasado el tiempo mínimo esperado de {expected_interval_minutes/60} horas para el seguimiento #{next_followup_number}")
         return True
 
 

letschatty/services/chat/chat_service.py

@@ -211,25 +211,6 @@ class ChatService:
         ChatService.add_central_notification_from_text(chat=chat, body=f"Agente de IA {chatty_ai_agent.name} actualizado en el chat {chat.id} por {execution_context.executor.name}", subtype=MessageSubtype.CHATTY_AI_AGENT_UPDATED)
         return chat.chatty_ai_agent
 
-    @staticmethod
-    def escalate_chatty_ai_agent(chat: Chat, execution_context: ExecutionContext, message: Optional[str] = None) -> None:
-        """
-        Mark the chat's AI agent as requiring human intervention and add a central notification.
-        """
-        if chat.chatty_ai_agent and not chat.chatty_ai_agent.requires_human_intervention:
-            chat.chatty_ai_agent.requires_human_intervention = True
-            execution_context.set_event_time(datetime.now(tz=ZoneInfo("UTC")))
-        body = "El chat fue escalado a un agente humano"
-        if message:
-            body = f"{body}: {message}"
-        ChatService.add_central_notification_from_text(
-            chat=chat,
-            body=body,
-            subtype=MessageSubtype.CHATTY_AI_AGENT_NOTIFICATION,
-            content_status=CentralNotificationStatus.WARNING,
-            context=ChattyContext(chain_of_thought_id=execution_context.chain_of_thought_id)
-        )
-
     @staticmethod
     def add_workflow_link(chat : Chat, link : LinkItem, flow:FlowPreview, execution_context: ExecutionContext, description: str, last_incoming_message_id: Optional[str] = None, next_call: Optional[datetime] = None) -> FlowStateAssignedToChat:
         """
@@ -285,46 +266,36 @@ class ChatService:
         return next((state for state in chat.flow_states if state.is_smart_follow_up), None)
 
     @staticmethod
-    def create_sale(
-        chat: Chat,
-        execution_context: ExecutionContext,
-        sale: Sale,
-        product: Optional[Product],
-        product_ids: Optional[List[StrObjectId]] = None,
-        product_label: Optional[str] = None
-    ) -> SaleAssignedToChat:
+    def create_sale(chat : Chat, execution_context: ExecutionContext, sale : Sale, product : Product) -> SaleAssignedToChat:
         """
         Add a sale to the chat.
         """
         if next((sale for sale in chat.client.sales if sale.asset_id == sale.id), None) is not None:
             raise AssetAlreadyAssigned(f"Sale with id {sale.id} already assigned to chat {chat.id}")
-        label = product_label or (product.name if product else "multiples productos")
-        assigned_product_ids = product_ids or ([product.id] if product else [])
         assigned_asset = SaleAssignedToChat(
             asset_type=ChatAssetType.SALE,
             asset_id=sale.id,
             assigned_at=sale.created_at,
             assigned_by=execution_context.executor.id,
-            product_id=product.id if product else None,
-            product_ids=assigned_product_ids
+            product_id=product.id
         )
         execution_context.set_event_time(assigned_asset.assigned_at)
         bisect.insort(chat.client.sales, assigned_asset)
-        ChatService.create_highlight(chat=chat, execution_context=execution_context, highlight_data=HighlightRequestData(title=f"🛍️ Venta de {label}", description=f"Venta de {label} creada por {execution_context.executor.name}", starred=False, subtype=MessageSubtype.SALE_ADDED))
-        ChatService.add_central_notification_from_text(chat=chat, body=f"Venta de {label} agregada al chat {chat.id} por {execution_context.executor.name}", subtype=MessageSubtype.SALE_ADDED)
+        ChatService.create_highlight(chat=chat, execution_context=execution_context, highlight_data=HighlightRequestData(title=f"🛍️ Venta de {product.name}", description=f"Venta de {product.name} creada por {execution_context.executor.name}", starred=False, subtype=MessageSubtype.SALE_ADDED))
+        ChatService.add_central_notification_from_text(chat=chat, body=f"Venta de {product.name} agregada al chat {chat.id} por {execution_context.executor.name}", subtype=MessageSubtype.SALE_ADDED)
         return assigned_asset
 
     @staticmethod
-    def update_sale(chat: Chat, execution_context: ExecutionContext, sale: Sale, product_label: str) -> Sale:
+    def update_sale(chat : Chat, execution_context: ExecutionContext, sale : Sale, product : Product) -> Sale:
         """
         Update a sale for the chat.
         """
-        ChatService.create_highlight(chat=chat, execution_context=execution_context, highlight_data=HighlightRequestData(title=f"🛍️ Venta actualizada de {product_label}", description=f"Venta de {product_label} actualizada por {execution_context.executor.name}", starred=False, subtype=MessageSubtype.SALE_UPDATED))
-        ChatService.add_central_notification_from_text(chat=chat, body=f"Venta de {product_label} actualizada en el chat {chat.id} por {execution_context.executor.name}", subtype=MessageSubtype.SALE_UPDATED)
+        ChatService.create_highlight(chat=chat, execution_context=execution_context, highlight_data=HighlightRequestData(title=f"🛍️ Venta actualizada de {product.name}", description=f"Venta de {product.name} actualizada por {execution_context.executor.name}", starred=False, subtype=MessageSubtype.SALE_UPDATED))
+        ChatService.add_central_notification_from_text(chat=chat, body=f"Venta de {product.name} actualizada en el chat {chat.id} por {execution_context.executor.name}", subtype=MessageSubtype.SALE_UPDATED)
         return sale
 
     @staticmethod
-    def delete_sale(chat: Chat, execution_context: ExecutionContext, sale_id: StrObjectId, product_label: str) -> SaleAssignedToChat:
+    def delete_sale(chat : Chat, execution_context: ExecutionContext, sale_id : StrObjectId, product : Product) -> SaleAssignedToChat:
         """
         Logically remove a sale from the chat.
         """
@@ -332,8 +303,8 @@ class ChatService:
             assigned_asset_to_remove = next(sale for sale in chat.client.sales if sale.asset_id == sale_id)
             chat.client.sales.remove(assigned_asset_to_remove)
             execution_context.set_event_time(datetime.now(tz=ZoneInfo("UTC")))
-            ChatService.create_highlight(chat=chat, execution_context=execution_context, highlight_data=HighlightRequestData(title=f"🛍️ Venta eliminada de {product_label}", description=f"Venta de {product_label} eliminada por {execution_context.executor.name}", starred=False, subtype=MessageSubtype.SALE_DELETED))
-            ChatService.add_central_notification_from_text(chat=chat, body=f"Venta de {product_label} eliminada del chat {chat.id} por {execution_context.executor.name}", subtype=MessageSubtype.SALE_DELETED)
+            ChatService.create_highlight(chat=chat, execution_context=execution_context, highlight_data=HighlightRequestData(title=f"🛍️ Venta eliminada de {product.name}", description=f"Venta de {product.name} eliminada por {execution_context.executor.name}", starred=False, subtype=MessageSubtype.SALE_DELETED))
+            ChatService.add_central_notification_from_text(chat=chat, body=f"Venta de {product.name} eliminada del chat {chat.id} por {execution_context.executor.name}", subtype=MessageSubtype.SALE_DELETED)
             return assigned_asset_to_remove
         except StopIteration:
             raise NotFoundError(message=f"Sale with id {sale_id} not found in chat {chat.id}")
@@ -902,16 +873,9 @@ class ChatService:
             execution_context.set_event_time(datetime.now(tz=ZoneInfo("UTC")))
             logger.info(f"Updated collected data for chat {chat.id}: {', '.join(updated_fields)}")
 
-            field_label_map = {
-                "name": "nombre",
-                "email": "email",
-                "phone": "telefono",
-                "document_id": "dni",
-            }
-            display_fields = [field_label_map.get(field, field) for field in updated_fields]
             ChatService.add_central_notification_from_text(
                 chat=chat,
-                body=f"Datos del cliente recopilados: {', '.join(display_fields)}",
+                body=f"Collected customer data: {', '.join(updated_fields)}",
                 subtype=MessageSubtype.CLIENT_INFO_UPDATED,
                 context=ChattyContext(chain_of_thought_id=execution_context.chain_of_thought_id)
             )

letschatty/services/chatty_assets/__init__.py

@@ -1,2 +1,14 @@
 from .base_container import ChattyAssetBaseContainer
 from .base_container_with_collection import ChattyAssetContainerWithCollection
+from .assets_collections import AssetsCollections
+from .services import (
+    ProductService,
+    TagService,
+    UserService,
+    ChatService,
+    SourceService,
+    FlowService,
+    SaleService,
+    ContactPointService,
+    AiAgentService
+)