idZddlZddlZddlmZmZddlmZmZm Z ddl m Z m Z m Z mZeGddZeGd d ZGd d ZeZy) zY Dialogue State Management Manages conversation state, cooldowns, and call persistence. N) dataclassfield)OptionalDictList)AI_SPEAK_COOLDOWN_SECMAX_CLARIFICATION_COUNTSTALE_CALL_TIMEOUT_SEClog_callceZdZUdZdZeed<dZeed<dZ e ed<dZ eed<dZ eed <d Z eeed <efd ed efdZdZdZdZdZefde d efdZdefdZdZdZd efdZd efdZeded dfdZy ) DialogueStatez State machine for managing dialogue flow. Tracks AI responses, user turns, clarification loops, and conversation status. last_ai_speak_tsF awaiting_userr clarify_countconversation_complete call_endedN locked_intent cooldown_secreturncJtj|jz |k\S)z2Check if AI is allowed to speak (cooldown expired))timer)selfrs 0/home/sas/my/fyp/ringai/ringai/dialogue/state.py can_speakzDialogueState.can_speak#syy{T222lBBcDtj|_d|_y)zRecord that AI just spokeTNrrrrs rrecord_ai_speakzDialogueState.record_ai_speak's $ !rc\tj|jz dkDrd|_yy)z!Record that user started speakingg?FNrr s rrecord_user_inputzDialogueState.record_user_input,s( 99;.. . 4!&D  5rc.|xjdz c_y)zIncrement clarification counterNrr s rincrement_clarifyzDialogueState.increment_clarify1s arcd|_y)zReset clarification counterrNr&r s r reset_clarifyzDialogueState.reset_clarify5s r max_countc |j|kDS)z7Check if should escalate due to too many clarificationsr&)rr*s rshould_escalatezDialogueState.should_escalate9s!!I--rintentc2|r|j||_yyy)z&Lock conversation to a specific intentN)r)rr-s r lock_intentzDialogueState.lock_intent=s d((0!'D 16rc d|_d|_y)z,Mark conversation as complete (graceful end)TFN)rrr s rend_conversationzDialogueState.end_conversationBs%)""rcd|_y)zMark call as ended (hangup)TN)rr s rend_callzDialogueState.end_callGs rc:|j xr |j S)z%Check if conversation is still active)rrr s r is_activezDialogueState.is_activeKs??"E4+E+E'EErc|j|j|j|j|j|j dS)zExport state as dictionaryrrrrrrr7r s rto_dictzDialogueState.to_dictOsB!% 5 5!//!//%)%?%?//!//   rdatac ||jdd|jdd|jdd|jdd|jdd|jd  S) zCreate state from dictionaryrrrFrrrrrr7)get)clsr9s r from_dictzDialogueState.from_dictZse!XX&8#>((?E:((?A6"&((+BE"Jxx e4((?3   r) __name__ __module__ __qualname____doc__rfloat__annotations__rboolrintrrrrstrr rr!r#r'r)r r,r/r1r3r5dictr8 classmethodr=rrrrs "e!M4M3"'4'J#'M8C='.CCeCC" '  0G..4.(#( # F4F      T  o    rrc>eZdZUdZeed<eeZe eed<eeZ e e ed<ee Z e ed<eejZejed<eej"Zeed<d Zefd ed efd Zd efdZd efdZdedefdZdefdZy) CallStatezj Persistent state for a single call. Survives WebSocket reconnections within timeout period. call_id)default_factorycommitted_texts llm_historydialogue_statellm_lock last_activityc6tj|_y)zUpdate last activity timestampNrrRr s rupdate_activityzCallState.update_activityvs!YY[r timeout_secrcJtj|jz |kDS)z(Check if call has been inactive too longrT)rrVs ris_stalezCallState.is_stalezsyy{T///+==rtextc:|jj|y)z'Add transcribed text to committed textsN)rNappend)rrYs radd_transcriptzCallState.add_transcript~s ##D)rcTdj|jjS)zGet complete transcript )joinrNstripr s rget_final_transcriptzCallState.get_final_transcripts xx,,-3355rrolecontentcf|jj||tjdy)zAdd message to LLM history)rbrctsN)rOr[r)rrbrcs radd_to_historyzCallState.add_to_historys* ))+!  r max_lengthcpt|j|kDr|j| d|jddyy)zTrim history to maximum lengthN)lenrO)rrgs r trim_historyzCallState.trim_historys9 t : -"&"2"2J;<"@D  Q  .rN)r>r?r@rArFrCrlistrNrrOrGrrPasyncioLockrQrrRrBrUr rDrXr\rarfrErjrIrrrKrKgs L!&t!E>t>*3*6c6 3  AsArrKceZdZdZdZdedeefdZdedefdZ defdZ e fde de efd Zdefd Zdefd Zdedefd Zy )CallStateManagerzk Manager for all active call states. Handles creation, retrieval, and cleanup of call states. ci|_yN_statesr s r__init__zCallStateManager.__init__s -/ rrLrc8|jj|S)zGet call state if it exists)rsr;rrLs rr;zCallStateManager.gets||((rc||jvr0t||j|<tjd|ntjd||j|}|j |S)z)Get existing call state or create new one)rLzcall=%s new state createdzcall=%s state restored)rsrKr inforUrrLstates r get_or_createzCallStateManager.get_or_creates^ $,, &$-g$>DLL ! MM5w ? MM2G < W%  rch||jvr$|j|=tjd|yy)zRemove call statezcall=%s state removedN)rsr rxrvs rremovezCallStateManager.removes. dll " W% MM17 ; #rrVcg}tj}|jjD])\}}||jz |kDs|j |+|S)z Find and return list of stale call IDs. Does NOT remove them - caller should finalize before removing. )rrsitemsrRr[)rrVstalenowrLrzs r cleanup_stalezCallStateManager.cleanup_stalesZ iik"ll002 &NGUU(((;6 W% & rc|jj|}|r1|jjt j d|yy)z!Mark call as ended (from webhook)zcall=%s marked as endedN)rsr;rPr3r rxrys r mark_endedzCallStateManager.mark_endeds?   )   ) ) + MM3W = rc,t|jSrq)rirsr s r__len__zCallStateManager.__len__s4<<  rc||jvSrqrrrvs r __contains__zCallStateManager.__contains__s$,,&&rN)r>r?r@rArtrFrrKr;r{r}r rBrrrrErrDrrIrrroros 0)3)8I#6) S Y #>!!'C'D'rro)rArlr dataclassesrrtypingrrrconfigr r r r rrKrocall_state_managerrIrrrsu  ('' O O  O d )A)A )AX:':'|&'r