letschatty 0.4.338__py3-none-any.whl → 0.4.350__py3-none-any.whl

letschatty/models/analytics/events/event_types.py

@@ -12,6 +12,51 @@ class EventType(StrEnum):
     AI_AGENT_ASSIGNED_TO_CHAT = "chat.chatty_ai_agent.assigned_to_chat"
     AI_AGENT_REMOVED_FROM_CHAT = "chat.chatty_ai_agent.removed_from_chat"
     AI_AGENT_UPDATED_ON_CHAT = "chat.chatty_ai_agent.updated_on_chat"
+
+    #CHATTY AI AGENT EXECUTION EVENTS - 3-level hierarchy for execution tracking
+    # Pattern: chatty_ai_agent_in_chat.{operation}.{detail}
+    # Note: Execution events are already chat-scoped via CustomerEventData
+
+    # TRIGGER EVENTS - What initiates AI agent processing
+    CHATTY_AI_AGENT_IN_CHAT_TRIGGER_USER_MESSAGE = "chatty_ai_agent_in_chat.trigger.user_message"
+    CHATTY_AI_AGENT_IN_CHAT_TRIGGER_FOLLOW_UP = "chatty_ai_agent_in_chat.trigger.follow_up"
+    CHATTY_AI_AGENT_IN_CHAT_TRIGGER_MANUAL = "chatty_ai_agent_in_chat.trigger.manual"
+    CHATTY_AI_AGENT_IN_CHAT_TRIGGER_RETRY = "chatty_ai_agent_in_chat.trigger.retry"
+
+    # STATE EVENTS - AI agent state changes
+    CHATTY_AI_AGENT_IN_CHAT_STATE_PROCESSING_STARTED = "chatty_ai_agent_in_chat.state.processing_started"
+    CHATTY_AI_AGENT_IN_CHAT_STATE_CALL_STARTED = "chatty_ai_agent_in_chat.state.call_started"
+    CHATTY_AI_AGENT_IN_CHAT_STATE_ESCALATED = "chatty_ai_agent_in_chat.state.escalated"
+    CHATTY_AI_AGENT_IN_CHAT_STATE_UNESCALATED = "chatty_ai_agent_in_chat.state.unescalated"
+
+    # CALL EVENTS - Outbound calls to services
+    CHATTY_AI_AGENT_IN_CHAT_CALL_GET_CHAT_WITH_PROMPT = "chatty_ai_agent_in_chat.call.get_chat_with_prompt"
+    CHATTY_AI_AGENT_IN_CHAT_CALL_TAGGER = "chatty_ai_agent_in_chat.call.tagger"
+    CHATTY_AI_AGENT_IN_CHAT_CALL_DOUBLE_CHECKER = "chatty_ai_agent_in_chat.call.double_checker"
+    CHATTY_AI_AGENT_IN_CHAT_CALL_DEBUGGER = "chatty_ai_agent_in_chat.call.debugger"
+
+    # CALLBACK EVENTS - Responses received from services
+    CHATTY_AI_AGENT_IN_CHAT_CALLBACK_GET_CHAT_WITH_PROMPT = "chatty_ai_agent_in_chat.callback.get_chat_with_prompt"
+    CHATTY_AI_AGENT_IN_CHAT_CALLBACK_TAGGER = "chatty_ai_agent_in_chat.callback.tagger"
+    CHATTY_AI_AGENT_IN_CHAT_CALLBACK_DOUBLE_CHECKER = "chatty_ai_agent_in_chat.callback.double_checker"
+    CHATTY_AI_AGENT_IN_CHAT_CALLBACK_OUTPUT_RECEIVED = "chatty_ai_agent_in_chat.callback.output_received"
+
+    # DECISION EVENTS - AI agent decisions and actions
+    CHATTY_AI_AGENT_IN_CHAT_DECISION_SEND = "chatty_ai_agent_in_chat.decision.send"
+    CHATTY_AI_AGENT_IN_CHAT_DECISION_SUGGEST = "chatty_ai_agent_in_chat.decision.suggest"
+    CHATTY_AI_AGENT_IN_CHAT_DECISION_ESCALATE = "chatty_ai_agent_in_chat.decision.escalate"
+    CHATTY_AI_AGENT_IN_CHAT_DECISION_SKIP = "chatty_ai_agent_in_chat.decision.skip"
+    CHATTY_AI_AGENT_IN_CHAT_DECISION_SENT_TO_API = "chatty_ai_agent_in_chat.decision.sent_to_api"
+    CHATTY_AI_AGENT_IN_CHAT_DECISION_COMPLETED = "chatty_ai_agent_in_chat.decision.completed"
+
+    # ERROR EVENTS - Failures and cancellations
+    CHATTY_AI_AGENT_IN_CHAT_ERROR_CALL_FAILED = "chatty_ai_agent_in_chat.error.call_failed"
+    CHATTY_AI_AGENT_IN_CHAT_ERROR_CALL_CANCELLED = "chatty_ai_agent_in_chat.error.call_cancelled"
+    CHATTY_AI_AGENT_IN_CHAT_ERROR_VALIDATION_FAILED = "chatty_ai_agent_in_chat.error.validation_failed"
+
+    # RATING EVENTS - User feedback
+    CHATTY_AI_AGENT_IN_CHAT_RATING_RECEIVED = "chatty_ai_agent_in_chat.rating.received"
+
     #PRODUCTS
     PRODUCT_ASSIGNED = "chat.product.assigned"
     PRODUCT_REMOVED = "chat.product.removed"
@@ -42,9 +87,10 @@ class EventType(StrEnum):
     #CONTINUOUS CONVERSATION
     CONTINUOUS_CONVERSATION_CREATED = "chat.continuous_conversation.created"
     CONTINUOUS_CONVERSATION_UPDATED = "chat.continuous_conversation.updated"
-    #CHAT FUNNEL EVENTS
-    CHAT_FUNNEL_STARTED = "chat.funnel.started"
-    CHAT_FUNNEL_STAGE_CHANGED = "chat.funnel.stage_changed"
+    #FUNNEL STAGES
+    # Funnel-level events
+    CHAT_FUNNEL_STARTED = "chat.funnel.started"  # New
+    CHAT_FUNNEL_UPDATED = "chat.funnel.updated"
     CHAT_FUNNEL_COMPLETED = "chat.funnel.completed"
     CHAT_FUNNEL_ABANDONED = "chat.funnel.abandoned"
 
@@ -83,18 +129,13 @@ class EventType(StrEnum):
     FAST_ANSWER_CREATED = "company.fast_answer.created"
     FAST_ANSWER_UPDATED = "company.fast_answer.updated"
     FAST_ANSWER_DELETED = "company.fast_answer.deleted"
-    #FUNNELS
+    #FUNNEL STAGES
     FUNNEL_CREATED = "company.funnel.created"
     FUNNEL_UPDATED = "company.funnel.updated"
     FUNNEL_DELETED = "company.funnel.deleted"
-    #FUNNEL STAGES
     FUNNEL_STAGE_CREATED = "company.funnel_stage.created"
     FUNNEL_STAGE_UPDATED = "company.funnel_stage.updated"
     FUNNEL_STAGE_DELETED = "company.funnel_stage.deleted"
-    #FUNNEL MEMBERS
-    FUNNEL_MEMBER_ADDED = "company.funnel_member.added"
-    FUNNEL_MEMBER_UPDATED = "company.funnel_member.updated"
-    FUNNEL_MEMBER_REMOVED = "company.funnel_member.removed"
     #BUSINESS AREA
     BUSINESS_AREA_CREATED = "company.business_area.created"
     BUSINESS_AREA_UPDATED = "company.business_area.updated"

letschatty/models/chat/chat.py

@@ -19,7 +19,6 @@ from ..utils.custom_exceptions.custom_exceptions import MissingAIAgentForSmartFo
 from .time_left import TimeLeft
 from ..company.conversation_topic import TopicTimelineEntry
 from ..company.form_field import CollectedData
-from ..company.CRM.funnel import ActiveFunnel
 import json
 import logging
 logger = logging.getLogger(__name__)
@@ -48,7 +47,6 @@ class Chat(CompanyAssetModel):
     chatty_ai_agent: Optional[ChattyAIAgentAssignedToChat] = Field(default=None, description="The id of the chatty ai agent that might or might not be assigned to the chat")
     suggested_messages: List[MessageDraft] = Field(default_factory=list)
     topics_timeline: List[TopicTimelineEntry] = Field(default_factory=list, description="Timeline of conversation topics throughout the conversation")
-    active_funnel: Optional[ActiveFunnel] = Field(default=None, description="Current active funnel for this chat")
     model_config = ConfigDict(extra="ignore")
 
     @property

letschatty/models/chat/chat_with_assets.py

@@ -1,6 +1,6 @@
+
 from .chat import Chat
 from letschatty.models.company.assets import TagPreview, FlowPreview, ContactPoint, Sale, ChattyAIAgentPreview, ProductPreview
-from letschatty.models.company.CRM.funnel import ActiveFunnel
 
 from letschatty.models.utils.types.serializer_type import SerializerType
 from pydantic import BaseModel
@@ -15,8 +15,3 @@ class ChatWithAssets(BaseModel):
     contact_points: List[ContactPoint]
     flows_links_states: List[FlowPreview]
     chatty_ai_agent: Optional[ChattyAIAgentPreview]
-
-    @property
-    def active_funnel(self) -> Optional[ActiveFunnel]:
-        """Convenience property to access the chat's active funnel"""
-        return self.chat.active_funnel

letschatty/models/chat/client.py

@@ -5,6 +5,7 @@ from ..utils.types.identifier import StrObjectId
 from .quality_scoring import QualityScore
 from .highlight import Highlight
 from ..company.assets.chat_assets import AssignedAssetToChat, SaleAssignedToChat, ContactPointAssignedToChat
+from ..company.CRM.funnel import ClientFunnel
 from ..utils.types.serializer_type import SerializerType
 
 class Client(CompanyAssetModel):
@@ -22,6 +23,7 @@ class Client(CompanyAssetModel):
     contact_points: List[ContactPointAssignedToChat] = Field(default=list())
     business_area: Optional[StrObjectId] = Field(default=None, description="It's a business related area, that works as a queue for the chats")
     external_id: Optional[str] = Field(default=None)
+    funnels : List[ClientFunnel] = Field(default=list())
     exclude_fields = {
         SerializerType.FRONTEND: {"products", "tags", "sales", "contact_points", "highlights"}
     }

letschatty/models/chat/continuous_conversation.py

@@ -30,7 +30,7 @@ class ContinuousConversation(ChattyAssetModel):
     template_message_waid: Optional[str] = None
     status: Optional[ContinuousConversationStatus] = Field(default=ContinuousConversationStatus.CREATED)
     active: bool = Field(default=True)
-    expires_at: datetime = Field(default_factory=lambda: datetime.now(ZoneInfo("UTC")) + timedelta(days=10))
+    expires_at: datetime = Field(default=datetime.now(ZoneInfo("UTC")) + timedelta(days=10))
     messages: List[MessageDraft]
     creator_id: StrObjectId
     forced_send: bool = Field(default=False)

letschatty/models/company/CRM/funnel.py

@@ -1,401 +1,69 @@
-from ...base_models import CompanyAssetModel, ChattyAssetPreview, ChattyAssetModel
-from typing import List, Optional, ClassVar, Any
+from ...base_models import CompanyAssetModel
+from typing import List
 from ...utils.types.identifier import StrObjectId
-from pydantic import BaseModel, Field, ConfigDict
+from pydantic import BaseModel, Field
 from enum import StrEnum
 from datetime import datetime
+from typing import Optional
 from zoneinfo import ZoneInfo
 from ...utils.types.executor_types import ExecutorType
-from ..assets.automation import Automation
-
-
-# ============================================================================
-# Enums
-# ============================================================================
 
 class FunnelStatus(StrEnum):
-    """Status of a chat within a funnel"""
     IN_PROGRESS = "in_progress"
     COMPLETED = "completed"
     ABANDONED = "abandoned"
 
-
-class FunnelMemberRole(StrEnum):
-    """Role of a user within a funnel"""
-    ADMIN = "admin"      # Full control over funnel settings, stages, and members
-    EDITOR = "editor"    # Can move chats between stages, view all chats
-    VIEWER = "viewer"    # Read-only access to funnel and chats
-
-
-# ============================================================================
-# Embedded Models (BaseModel)
-# ============================================================================
-
 class StageTransition(BaseModel):
-    """
-    Record of a chat transitioning between stages within a funnel.
-    Tracks the transition details and time spent in the previous stage.
-    """
-    from_stage_id: Optional[StrObjectId] = Field(
-        default=None, 
-        description="The stage the chat came from (None if entering funnel)"
-    )
-    to_stage_id: StrObjectId = Field(description="The stage the chat moved to")
-    transitioned_at: datetime = Field(default_factory=lambda: datetime.now(ZoneInfo("UTC")))
-    executor_type: ExecutorType = Field(description="Type of executor that triggered the transition")
-    executor_id: StrObjectId = Field(description="ID of the executor that triggered the transition")
-    time_in_previous_stage_seconds: Optional[int] = Field(
-        default=None, 
-        description="Time spent in the previous stage before this transition"
-    )
-    from_stage_order: Optional[int] = Field(
-        default=None,
-        description="Order of the from_stage at transition time (for detecting regressions)"
-    )
-    to_stage_order: Optional[int] = Field(
-        default=None,
-        description="Order of the to_stage at transition time"
-    )
-
-    @property
-    def is_regression(self) -> bool:
-        """Check if this transition moved backwards in the funnel"""
-        if self.from_stage_order is None or self.to_stage_order is None:
-            return False
-        return self.to_stage_order < self.from_stage_order
-
-    @property
-    def is_entry(self) -> bool:
-        """Check if this is an entry transition (no previous stage)"""
-        return self.from_stage_id is None
-
-
-class ActiveFunnel(BaseModel):
-    """
-    Lightweight funnel state embedded in Chat for fast filtering and display.
-    
-    The full history (stage_transitions, time metrics) is stored in the
-    separate chat_funnels collection via ChatFunnel.
-    """
-    chat_funnel_id: StrObjectId = Field(
-        description="Reference to the full ChatFunnel document in chat_funnels collection"
-    )
-    funnel_id: StrObjectId = Field(description="The funnel the chat is in")
-    funnel_name: str = Field(description="Denormalized funnel name for display")
-    current_stage_id: StrObjectId = Field(description="Current stage ID")
-    current_stage_name: str = Field(description="Denormalized stage name for display")
-    entered_current_stage_at: datetime = Field(
-        default_factory=lambda: datetime.now(ZoneInfo("UTC")),
-        description="When the chat entered the current stage"
-    )
-
-    @property
-    def time_in_current_stage_seconds(self) -> int:
-        """Calculate time spent in the current stage"""
-        return int((datetime.now(tz=ZoneInfo("UTC")) - self.entered_current_stage_at).total_seconds())
-
-    def update_stage(self, stage_id: StrObjectId, stage_name: str) -> None:
-        """Update the current stage (called when transitioning)"""
-        self.current_stage_id = stage_id
-        self.current_stage_name = stage_name
-        self.entered_current_stage_at = datetime.now(tz=ZoneInfo("UTC"))
-
-
-# ============================================================================
-# Preview Classes - For efficient listing
-# ============================================================================
-
-class FunnelPreview(ChattyAssetPreview):
-    """Preview of a Funnel for listing without full data"""
-    is_active: bool = Field(default=True)
-    
-    @classmethod
-    def get_projection(cls) -> dict[str, Any]:
-        base = super().get_projection()
-        base["is_active"] = 1
-        return base
-
-    @classmethod
-    def from_asset(cls, asset: 'Funnel') -> 'FunnelPreview':
-        return cls(
-            _id=asset.id,
-            name=asset.name,
-            company_id=asset.company_id,
-            created_at=asset.created_at,
-            deleted_at=asset.deleted_at,
-            is_active=asset.is_active
-        )
-
-
-class FunnelStagePreview(ChattyAssetPreview):
-    """Preview of a FunnelStage for listing"""
+    from_stage_id: Optional[StrObjectId] = None
+    to_stage_id: StrObjectId
+    transitioned_at: datetime = Field(default=datetime.now(ZoneInfo("UTC")))
+    executor_type: ExecutorType
+    executor_id: StrObjectId
+    time_in_previous_stage_seconds: Optional[int] = None
+
+class ClientFunnel(CompanyAssetModel):
     funnel_id: StrObjectId
-    color: str
-    order: int
-    is_exit_stage: bool = Field(default=False)
-
-    @classmethod
-    def get_projection(cls) -> dict[str, Any]:
-        base = super().get_projection()
-        base.update({
-            "funnel_id": 1,
-            "color": 1,
-            "order": 1,
-            "is_exit_stage": 1
-        })
-        return base
-
-    @classmethod
-    def from_asset(cls, asset: 'FunnelStage') -> 'FunnelStagePreview':
-        return cls(
-            _id=asset.id,
-            name=asset.name,
-            company_id=asset.company_id,
-            created_at=asset.created_at,
-            deleted_at=asset.deleted_at,
-            funnel_id=asset.funnel_id,
-            color=asset.color,
-            order=asset.order,
-            is_exit_stage=asset.is_exit_stage
-        )
-
-
-# ============================================================================
-# Company Assets (CompanyAssetModel) - Stored in separate collections
-# ============================================================================
-
-class Funnel(CompanyAssetModel):
-    """
-    A funnel represents a pipeline/process for managing chats (e.g., sales funnel, support pipeline).
-    Companies can create multiple funnels to organize their chat workflows.
-    """
-    name: str = Field(description="Name of the funnel")
-    description: Optional[str] = Field(default=None, description="Description of the funnel's purpose")
-    created_by: StrObjectId = Field(description="User ID who created the funnel")
-    is_active: bool = Field(default=True, description="Whether the funnel is active or archived")
-
-    preview_class: ClassVar[type[FunnelPreview]] = FunnelPreview
-
-    model_config = ConfigDict(
-        validate_by_name=True,
-        validate_by_alias=True
-    )
-
-
-class FunnelStage(CompanyAssetModel):
-    """
-    A stage within a funnel. Stages are ordered and can have automations
-    that execute when a chat enters the stage.
-    """
-    funnel_id: StrObjectId = Field(frozen=True, description="The funnel this stage belongs to")
-    name: str = Field(description="Name of the stage")
-    description: Optional[str] = Field(default=None, description="Description of the stage")
-    color: str = Field(default="#808080", description="Hex color for the stage (e.g., '#FFAA00')")
-    order: int = Field(ge=0, description="Position of the stage in the funnel (0-indexed)")
-    inflexion_conversion_point: bool = Field(
-        default=False, 
-        description="If true, prioritized for chat assignment and sends notifications in automatic mode"
-    )
-    is_exit_stage: bool = Field(
-        default=False,
-        description="If true, moving to this stage completes the funnel for the chat"
-    )
-    automations: Automation = Field(
-        default_factory=Automation,
-        description="Automations to execute when a chat enters this stage"
-    )
-
-    preview_class: ClassVar[type[FunnelStagePreview]] = FunnelStagePreview
-
-    model_config = ConfigDict(
-        validate_by_name=True,
-        validate_by_alias=True
-    )
-
-
-class FunnelMember(CompanyAssetModel):
-    """
-    A user's membership in a funnel with their assigned role.
-    Determines what actions the user can perform within the funnel.
-    """
-    name: str = Field(default="Funnel Member", description="Name for asset compatibility")
-    funnel_id: StrObjectId = Field(frozen=True, description="The funnel this membership belongs to")
-    user_id: StrObjectId = Field(frozen=True, description="The user who is a member")
-    role: FunnelMemberRole = Field(description="The user's role within the funnel")
-
-    model_config = ConfigDict(
-        validate_by_name=True,
-        validate_by_alias=True
-    )
-
-    @property
-    def can_edit(self) -> bool:
-        """Check if the member can edit (move chats, etc.)"""
-        return self.role in (FunnelMemberRole.ADMIN, FunnelMemberRole.EDITOR)
-
-    @property
-    def can_admin(self) -> bool:
-        """Check if the member can administer the funnel"""
-        return self.role == FunnelMemberRole.ADMIN
-
-
-# ============================================================================
-# ChatFunnel - Stored in separate chat_funnels collection
-# This is a ChattyAssetModel (not CompanyAssetModel) because it's a chat history
-# record rather than a company-level asset.
-# ============================================================================
-
-class ChatFunnel(ChattyAssetModel):
-    """
-    Full record of a chat's journey through a funnel.
-    Stored in a separate 'chat_funnels' collection, NOT embedded in chat.
-    
-    This extends ChattyAssetModel (not CompanyAssetModel) because it's a 
-    chat-level record tracking funnel history, not a company-owned asset.
-    
-    A new ChatFunnel is created each time a chat enters a funnel.
-    If a chat completes a funnel and enters it again, a NEW ChatFunnel is created.
-    
-    The Chat document stores a lightweight ActiveFunnel for fast
-    filtering and display, with a reference to this full record.
-    """
-    company_id: StrObjectId = Field(frozen=True, description="Company for multi-tenant isolation")
-    chat_id: StrObjectId = Field(frozen=True, description="The chat this funnel record belongs to")
-    funnel_id: StrObjectId = Field(frozen=True, description="The funnel the chat entered")
-    status: FunnelStatus = Field(default=FunnelStatus.IN_PROGRESS)
-    started_at: datetime = Field(default_factory=lambda: datetime.now(ZoneInfo("UTC")))
-    completed_at: Optional[datetime] = Field(default=None)
-    abandoned_at: Optional[datetime] = Field(default=None)
-    current_stage_id: Optional[StrObjectId] = Field(
-        default=None, 
-        description="Current stage ID (None if funnel is completed/abandoned)"
-    )
-    entered_current_stage_at: Optional[datetime] = Field(
-        default=None,
-        description="When the chat entered the current stage"
-    )
-    stage_transitions: List[StageTransition] = Field(
-        default_factory=list,
-        description="Complete history of all stage transitions"
-    )
-
-    model_config = ConfigDict(
-        validate_by_name=True,
-        validate_by_alias=True
-    )
+    status: FunnelStatus
+    started_at: datetime = Field(default=datetime.now(ZoneInfo("UTC")))
+    completed_at: Optional[datetime] = None
+    abandoned_at: Optional[datetime] = None
+    current_stage_id: Optional[StrObjectId] = None
+    entered_current_stage_at: datetime
+    stage_transitions: List[StageTransition] = Field(default_factory=list)
 
     @property
     def is_completed(self) -> bool:
-        """Check if the chat has completed the funnel"""
-        return self.status == FunnelStatus.COMPLETED and self.completed_at is not None
+        return self.completed_at is not None
 
     @property
     def is_abandoned(self) -> bool:
-        """Check if the chat has abandoned the funnel"""
-        return self.status == FunnelStatus.ABANDONED and self.abandoned_at is not None
-
-    @property
-    def is_active(self) -> bool:
-        """Check if the chat is still active in the funnel"""
-        return self.status == FunnelStatus.IN_PROGRESS
+        return self.abandoned_at is not None
 
     @property
     def time_in_funnel_seconds(self) -> int:
-        """Calculate total time spent in the funnel"""
         end_time = self.completed_at or self.abandoned_at or datetime.now(tz=ZoneInfo("UTC"))
         return int((end_time - self.started_at).total_seconds())
 
     @property
     def time_in_current_stage_seconds(self) -> Optional[int]:
-        """Calculate time spent in the current stage"""
-        if not self.entered_current_stage_at or not self.current_stage_id:
+        if not self.entered_current_stage_at:
             return None
         return int((datetime.now(tz=ZoneInfo("UTC")) - self.entered_current_stage_at).total_seconds())
 
     @property
     def unique_stages_visited(self) -> int:
-        """Count of unique stages the chat has visited"""
         return len(set(t.to_stage_id for t in self.stage_transitions))
 
-    @property
-    def total_transitions(self) -> int:
-        """Total number of stage transitions"""
-        return len(self.stage_transitions)
-
-    @property
-    def regression_count(self) -> int:
-        """Count of times the chat moved backwards in the funnel"""
-        return sum(1 for t in self.stage_transitions if t.is_regression)
-
-    @property
-    def last_transition(self) -> Optional[StageTransition]:
-        """Get the most recent stage transition"""
-        return self.stage_transitions[-1] if self.stage_transitions else None
-
-    def record_transition(
-        self,
-        to_stage_id: StrObjectId,
-        executor_type: ExecutorType,
-        executor_id: StrObjectId,
-        from_stage_order: Optional[int] = None,
-        to_stage_order: Optional[int] = None
-    ) -> StageTransition:
-        """
-        Record a transition to a new stage.
-        Returns the created StageTransition.
-        
-        Args:
-            to_stage_id: The stage to transition to
-            executor_type: Who/what triggered the transition
-            executor_id: ID of the executor
-            from_stage_order: Order of current stage (for regression detection)
-            to_stage_order: Order of target stage (for regression detection)
-        """
-        time_in_previous = self.time_in_current_stage_seconds
-        
-        transition = StageTransition(
-            from_stage_id=self.current_stage_id,
-            to_stage_id=to_stage_id,
-            executor_type=executor_type,
-            executor_id=executor_id,
-            time_in_previous_stage_seconds=time_in_previous,
-            from_stage_order=from_stage_order,
-            to_stage_order=to_stage_order
-        )
-        
-        self.stage_transitions.append(transition)
-        self.current_stage_id = to_stage_id
-        self.entered_current_stage_at = transition.transitioned_at
-        
-        return transition
-
-    def complete(self) -> None:
-        """Mark the funnel as completed"""
-        self.status = FunnelStatus.COMPLETED
-        self.completed_at = datetime.now(tz=ZoneInfo("UTC"))
+class FunnelStage(CompanyAssetModel):
+    name: str
+    description: str
+    index: int
+    inflexion_conversion_point: bool = Field(default=False, description="If true, it'll be prioritized for chat assignment and will send notifications in automatic mode to all agents.")
+    workflow_ids: List[StrObjectId] = Field(default_factory=list)
 
-    def abandon(self) -> None:
-        """Mark the funnel as abandoned"""
-        self.status = FunnelStatus.ABANDONED
-        self.abandoned_at = datetime.now(tz=ZoneInfo("UTC"))
 
-    def to_active_funnel(self, funnel_name: str, stage_name: str) -> ActiveFunnel:
-        """
-        Create an ActiveFunnel for embedding in the Chat document.
-        Call this after creating/updating the ChatFunnel.
-        
-        Args:
-            funnel_name: Name of the funnel (for denormalization)
-            stage_name: Name of the current stage (for denormalization)
-        """
-        if not self.current_stage_id or not self.entered_current_stage_at:
-            raise ValueError("Cannot create ActiveFunnel without current stage")
-        
-        return ActiveFunnel(
-            chat_funnel_id=self.id,
-            funnel_id=self.funnel_id,
-            funnel_name=funnel_name,
-            current_stage_id=self.current_stage_id,
-            current_stage_name=stage_name,
-            entered_current_stage_at=self.entered_current_stage_at
-        )
+class Funnel(CompanyAssetModel):
+    name: str
+    description: str
+    stages: List[FunnelStage]
+    assignment_priority: int = Field(ge=0, le=10, description="Priority for chat assignment, between funnels")

letschatty/models/company/__init__.py

@@ -1,4 +1,3 @@
 from .assets import *
 from .empresa import EmpresaModel
-from .form_field import FormField, FormFieldPreview, CollectedData, SystemFormFields
-from .CRM.funnel import Funnel, FunnelPreview, ChatFunnel, StageTransition, FunnelStatus
+from .form_field import FormField, FormFieldPreview, CollectedData, SystemFormFields

letschatty/models/company/assets/ai_agents_v2/ai_agents_decision_output.py

@@ -55,7 +55,7 @@ class SmartFollowUpDecision(BaseModel):
                             SmartFollowUpDecisionAction.POSTPONE_DELTA_TIME]:
             # Postpone actions don't require messages
             if self.messages is not None and len(self.messages) > 0:
-                raise ValueError("Messages are not allowed when action is postpone/postponed")
+                raise ValueError("Messages are not allowed when action is postpone")
         else:
             raise ValueError(f"Invalid action: {self.action}")
 

letschatty/models/company/assets/ai_agents_v2/chain_of_thought_in_chat.py

@@ -84,12 +84,13 @@ class ChainOfThoughtInChatPreview(ChattyAssetPreview):
     # trigger_id removed - trigger info is in: incoming_message_ids for user_message,
     # triggered_by_user_id for manual_trigger, smart_follow_up_id for follow_up
     chain_of_thought : Optional[str] = Field(default=None, description="The chain of thought")
+    confidence: Optional[int] = Field(default=None, ge=0, le=100, description="Confidence 0-100")
     status: Optional[ChainOfThoughtInChatStatus] = Field(default=None, description="The status of the chain of thought")
     name: str = Field(description="A title for the chain of thought")
 
     @classmethod
     def get_projection(cls) -> dict[str, Any]:
-        return super().get_projection() | {"chat_id": 1, "trigger": 1, "chain_of_thought": 1, "name": 1, "status": 1}
+        return super().get_projection() | {"chat_id": 1, "trigger": 1, "chain_of_thought": 1, "confidence": 1, "name": 1, "status": 1}
 
     @classmethod
     def from_dict(cls, data: dict) -> 'ChainOfThoughtInChatPreview':
@@ -103,6 +104,7 @@ class ChainOfThoughtInChatPreview(ChattyAssetPreview):
             chat_id=asset.chat_id,
             trigger=asset.trigger,
             chain_of_thought=asset.chain_of_thought,
+            confidence=asset.confidence,
             status=asset.status,
             company_id=asset.company_id,
             created_at=asset.created_at,
@@ -121,6 +123,7 @@ class ChainOfThoughtInChat(CompanyAssetModel):
     chatty_ai_agent_id : StrObjectId = Field(description="The chatty ai agent id")
     chatty_ai_agent : Dict[str, Any] = Field(description="The chatty ai agent at the moment of triggering the chain of thought", exclude=True,default_factory=dict)
     chain_of_thought : Optional[str] = Field(default=None, description="The chain of thought")
+    confidence: Optional[int] = Field(default=None, ge=0, le=100, description="Confidence 0-100")
     name: str = Field(description="A title for the chain of thought", alias="title")
     preview_class: ClassVar[Type[ChattyAssetPreview]] = ChainOfThoughtInChatPreview
 
@@ -128,5 +131,4 @@ class ChainOfThoughtInChat(CompanyAssetModel):
         if self.chain_of_thought is None:
             self.chain_of_thought = cot
         else:
-            self.chain_of_thought += f"\n{cot}"
-
+            self.chain_of_thought += f"\n{cot}"

letschatty/models/company/assets/ai_agents_v2/chatty_ai_agent_in_chat.py

@@ -92,6 +92,10 @@ class ChattyAIAgentInChat(CompanyAssetModel):
         default=None,
         description="If the trigger is a user message, this will be the id of the incoming message"
     )
+    last_reset_message_id: Optional[str] = Field(
+        default=None,
+        description="Last reset control trigger message id handled for this chat"
+    )
 
     # Assignment metadata
     assigned_at: datetime = Field(default_factory=lambda: datetime.now(ZoneInfo("UTC")))