letschatty 0.4.280__py3-none-any.whl → 0.4.343__py3-none-any.whl

letschatty/models/data_base/mongo_connection.py

@@ -1,14 +1,8 @@
 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__(
@@ -16,103 +10,26 @@ class MongoConnection(metaclass=SingletonMeta):
         username: Optional[str] = None,
         password: Optional[str] = None,
         uri_base: Optional[str] = None,
-        instance: Optional[str] = None,
-        verify_on_init: bool = True
+        instance: Optional[str] = None
     ):
         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")
-
-        # 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)
+            
+        uri = f"{self.uri_base}://{self.username}:{self.password}@{self.instance}.mongodb.net"
         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:
-            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
+            self.client.admin.command('ping')
         except Exception as e:
-            logger.error(f"Failed to verify MongoDB connection: {e}")
-            raise ConnectionError(f"Failed to verify MongoDB connection: {e}")
+            self.client.close()
+            raise ConnectionError(f"Failed to connect to MongoDB: {str(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,5 +1,6 @@
 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):
@@ -8,5 +9,6 @@ class ChattyContentDocument(ChattyContentMedia):
     @model_validator(mode='before')
     def validate_filename(cls, data: dict, info: ValidationInfo):
         if not data.get("filename") and data.get("url"):
-            data["filename"] = data["url"].split("/")[-1]
-        return data
+            parsed = urlparse(data["url"])
+            data["filename"] = unquote(parsed.path.split("/")[-1])
+        return data

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

@@ -1,5 +1,6 @@
 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")
@@ -11,9 +12,9 @@ class ChattyContentMedia(BaseModel):
     def validate_url(cls, v):
         if not v:
             raise ValueError("URL is required")
-        HttpUrl(v)
-        return v
+        encoded = quote(str(v), safe=":/?&=%#")
+        HttpUrl(encoded)
+        return encoded
     
     def get_body_or_caption(self) -> str:
         return self.caption
-

letschatty/models/utils/custom_exceptions/custom_exceptions.py

@@ -5,7 +5,6 @@ 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):
@@ -60,6 +59,30 @@ class WhatsAppPayloadValidationError(Exception):
     def __init__(self, message="WhatsApp payload validation error", status_code=400, **context_data):
         super().__init__(message, status_code=status_code, **context_data)
 
+
+class ChatWithActiveContinuousConversation(CustomException):
+    """
+    Raised when an AI agent is triggered on a chat that has an active Continuous Conversation.
+    """
+    def __init__(self, message="Chat has an active continuous conversation", status_code=400, **context_data):
+        super().__init__(message, status_code=status_code, **context_data)
+
+
+class ChattyAIModeOff(CustomException):
+    """
+    Raised when an AI agent is in OFF mode and cannot be triggered.
+    """
+    def __init__(self, message="Chatty AI agent is OFF", status_code=400, **context_data):
+        super().__init__(message, status_code=status_code, **context_data)
+
+
+class MissingAIAgentInChat(CustomException):
+    """
+    Raised when a chat has no AI agent assigned but one is required for the operation.
+    """
+    def __init__(self, message="AI agent not assigned to chat", status_code=404, **context_data):
+        super().__init__(message, status_code=status_code, **context_data)
+
 class UnsuportedChannel(CustomException):
     def __init__(self, message="Channel not supported", status_code=400, **context_data):
         super().__init__(message, status_code=status_code, **context_data)
@@ -206,16 +229,4 @@ 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)
-
-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_service_v2.py

@@ -1,5 +1,6 @@
 from typing import List
 from letschatty.models.chat.chat import Chat, FlowStateAssignedToChat
+from letschatty.models.chat.flow_link_state import StateTrigger
 from letschatty.models.company.assets.ai_agents_v2.follow_up_strategy import FollowUpStrategy
 from letschatty.models.company.assets.ai_agents_v2.ai_agents_decision_output import SmartFollowUpDecision, SmartFollowUpDecisionAction
 from letschatty.services.chat.chat_service import ChatService
@@ -85,5 +86,14 @@ class SmartFollowUpService:
             SmartFollowUpService.validate_next_call_time_or_default(decision, smart_follow_up_state.consecutive_count + 1, follow_up_strategy)
             smart_follow_up_state.next_call = decision.next_call_time_value
             ChatService.update_workflow_link(chat=chat, workflow_id=smart_follow_up_state.flow_id, workflow_link=smart_follow_up_state, execution_context=execution_context)
+        elif decision.action == SmartFollowUpDecisionAction.POSTPONE_DELTA_TIME:
+            logger.debug(f"Postponing smart follow up for chat {chat.id} by delta time {decision.next_call_time_value}")
+            SmartFollowUpService.validate_next_call_time_or_default(decision, smart_follow_up_state.consecutive_count + 1, follow_up_strategy)
+            smart_follow_up_state.next_call = decision.next_call_time_value
+            ChatService.update_workflow_link(chat=chat, workflow_id=smart_follow_up_state.flow_id, workflow_link=smart_follow_up_state, execution_context=execution_context)
+        elif decision.action == SmartFollowUpDecisionAction.POSTPONE_TILL_UPDATE:
+            logger.debug(f"Postponing smart follow up for chat {chat.id} till update")
+            smart_follow_up_state.trigger = StateTrigger.CHAT_UPDATE
+            ChatService.update_workflow_link(chat=chat, workflow_id=smart_follow_up_state.flow_id, workflow_link=smart_follow_up_state, execution_context=execution_context)
         else:
             raise ValueError(f"Invalid action: {decision.action}")

letschatty/services/chat/chat_service.py

@@ -28,9 +28,10 @@ from ...models.chat.scheduled_messages import ScheduledMessageStatus
 from ...models.utils.types.identifier import StrObjectId
 from ...models.utils.custom_exceptions.custom_exceptions import AssetAlreadyAssigned, MessageNotFoundError, NotFoundError, MessageAlreadyInChat, MetaErrorNotification, ChatAlreadyAssigned, AlreadyCompleted, ErrorToMantainSafety
 from ..factories.messages.central_notification_factory import CentralNotificationFactory
+from ..factories.messages.chatty_message_factory import from_message_draft
 from ...models.messages.chatty_messages.base.message_draft import ChattyContentAudio, MessageDraft
 from ...models.messages.chatty_messages.schema.chatty_content.content_central import CentralNotificationStatus
-from ...models.messages.chatty_messages.schema import ChattyContext
+from ...models.messages.chatty_messages.schema import ChattyContext, ChattyContentText
 from ...models.utils.types.message_types import MessageType
 from .conversation_topics_service import ConversationTopicsService
 import logging
@@ -211,6 +212,43 @@ 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,
+        reason: 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 reason:
+            body = f"{body}. Motivo: {reason}"
+        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)
+        )
+        if message:
+            outgoing_context = ChattyContext(chain_of_thought_id=execution_context.chain_of_thought_id)
+            outgoing_message = from_message_draft(
+                MessageDraft(
+                    type=MessageType.TEXT,
+                    content=ChattyContentText(body=message),
+                    context=outgoing_context,
+                    subtype=MessageSubtype.NONE,
+                    is_incoming_message=False
+                ),
+                sent_by=execution_context.executor.id
+            )
+            ChatService.add_message(chat=chat, message=outgoing_message)
+
     @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:
         """
@@ -266,36 +304,46 @@ 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 : Product) -> SaleAssignedToChat:
+    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:
         """
         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
+            product_id=product.id if product else None,
+            product_ids=assigned_product_ids
         )
         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 {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)
+        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)
         return assigned_asset
 
     @staticmethod
-    def update_sale(chat : Chat, execution_context: ExecutionContext, sale : Sale, product : Product) -> Sale:
+    def update_sale(chat: Chat, execution_context: ExecutionContext, sale: Sale, product_label: str) -> 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.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)
+        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)
         return sale
 
     @staticmethod
-    def delete_sale(chat : Chat, execution_context: ExecutionContext, sale_id : StrObjectId, product : Product) -> SaleAssignedToChat:
+    def delete_sale(chat: Chat, execution_context: ExecutionContext, sale_id: StrObjectId, product_label: str) -> SaleAssignedToChat:
         """
         Logically remove a sale from the chat.
         """
@@ -303,8 +351,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.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)
+            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)
             return assigned_asset_to_remove
         except StopIteration:
             raise NotFoundError(message=f"Sale with id {sale_id} not found in chat {chat.id}")
@@ -663,7 +711,10 @@ class ChatService:
         if client_data.external_id is not None:
             chat.client.external_id = client_data.external_id
         if client_data.lead_form_data is not None:
-            chat.client.lead_form_data = client_data.lead_form_data
+            # Merge with existing lead_form_data instead of replacing
+            if chat.client.lead_form_data is None:
+                chat.client.lead_form_data = {}
+            chat.client.lead_form_data.update(client_data.lead_form_data)
         execution_context.set_event_time(datetime.now(tz=ZoneInfo("UTC")))
         ChatService.add_central_notification_from_text(chat=chat, body=f"La info del cliente fue actualizada por {execution_context.executor.name}", subtype=MessageSubtype.CLIENT_INFO_UPDATED)
         return chat
@@ -846,6 +897,13 @@ class ChatService:
             chat.client.email = collected_data.email
             updated_fields.append("email")
 
+        if collected_data.phone:
+            if chat.client.lead_form_data is None:
+                chat.client.lead_form_data = {}
+            if chat.client.lead_form_data.get("phone") != collected_data.phone:
+                chat.client.lead_form_data["phone"] = collected_data.phone
+                updated_fields.append("phone")
+
         if collected_data.document_id and chat.client.document_id != collected_data.document_id:
             chat.client.document_id = collected_data.document_id
             updated_fields.append("document_id")
@@ -863,11 +921,18 @@ 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"Collected customer data: {', '.join(updated_fields)}",
+                body=f"Datos del cliente recopilados: {', '.join(display_fields)}",
                 subtype=MessageSubtype.CLIENT_INFO_UPDATED,
                 context=ChattyContext(chain_of_thought_id=execution_context.chain_of_thought_id)
             )
 
-        return chat.client.lead_form_data
+        return chat.client.lead_form_data

letschatty/services/chatty_assets/__init__.py

@@ -1,14 +1,2 @@
 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
-)