yj1B ddlmZddlmZddlmZddlmZddlm Z m Z m Z m Z m Z ddlmZddlmZddlmZdd lmZmZdd lmZmZmZmZdd lmZdd lmZm Z dd l!m"Z"ddl#m$Z$ddl%m&Z&ee'Z(eGddZ)dZ*edZ+de,de,fdZ-ee*ej.j/deeej0e*ej.j1de+dede,de2e,efdzfdZ3dddedej4d e5de$ej6fd!Z7 d(d"e2e,efdzde e8ej6fd#Z9dede,dej6fd$Z:dede,dej;dej6fd%ZdS))) dataclass) getLogger)Any)NOT_NONE)Selectdeleteexistsfuncselect)IntegrityError) AsyncSession)make_transient_to_detached)modelsschemas)cacheget_cache_namespacesafe_cache_deletesafe_cache_set)settings)ConflictExceptionResourceNotFoundException) apply_filter)GetOrCreateResult)get_external_vector_storecPeZdZUdZejed<eed<eed<eed<eed<dS)WorkspaceDeletionResultz8Result of a workspace deletion including cascade counts. workspace peers_deletedsessions_deletedmessages_deletedconclusions_deletedN)__name__ __module__ __qualname____doc__r Workspace__annotations__int:/DATA/AppData/hermes/projects/honcho/src/crud/workspace.pyrrsWBB    r*rzv2:workspace:{workspace_name}z:lock:v2workspace_namereturnc\tdzt|zS)z!Generate cache key for workspace.:r,)rWORKSPACE_CACHE_KEY_TEMPLATEformatr0s r+workspace_cache_keyr3*s4    & - -^ - L L Mr*s)keyttlprefix condition)r5r6r7dbNc K|ttjtjj|kd{V}|dS|j|j|j|j|j |j dS)zPFetch a workspace from the database and return as a plain dict for safe caching.Nidname h_metadatainternal_metadata configuration created_at) scalarr rr&wherer=r<r>r?r@rA)r9r,objs r+_fetch_workspacerE3s v  &&v'7'<'NOO      C {tfn 2*n   r*F_retryrrGc K|jstdt||jd{V}|ntd|jt jdi|}t|||dd{V}t|dSt j|j|j |j d } | 4d{V||dddd{Vn#1d{VswxYwYtd |jt|j |j|j|j|j|j |jd  fd }t|d| S#t($r5|rt+d|jdt-||dd{VcYSwxYw)aO Get an existing workspace or create a new one if it doesn't exist. Args: db: Database session workspace: Workspace creation schema Returns: GetOrCreateResult containing the workspace and whether it was created Raises: ConflictException: If we fail to get or create the workspace zWorkspace name must be providedNzFound existing workspace: %sFload)createdT exclude_none)r=r>r@z"Workspace created successfully: %sr;cZKttjjd{VdS)N)expire)rrCACHEDEFAULT_TTL_SECONDS) _cache_data _cache_keysr+_warm_workspace_cachez6get_or_create_workspace.._warm_workspace_cachesT ~9         r*)rK on_commitz#Unable to create or get workspace: rFr))r= ValueErrorrEloggerdebugrr&rmergermetadatar@ model_dump begin_nestedaddr3r<r>r?rAr rget_or_create_workspace) r9rrGdatarDexisting_workspacehoncho_workspacerTrRrSs @@r+r^r^Qs( ><:;;;""in55 5 5 5 5 5 5D  3Y^DDD&&&&"3'''#%88Ce8#<#<<<<<<< !3UCCCC' ^%-88d8KK I??$$ % % % % % % % % FF# $ $ $ % % % % % % % % % % % % % % % % % % % % % % % % % % % 99>JJJ)88 "%$)*5!1!C-;*5         ! d6K     III  #FinFF -R4HHHHHHHHHHHH Is7!F';D#F'# D--F'0D-1A5F''r@copyr[commit post_commitrWrXr3rr<) r9r,r ws_resultra needs_update base_config merged_config cache_keys r+update_workspacerts,. '-2   I*3);L &  '9+= = =&/&8# *'5;AACC   %00d0CC   )] : :-:  *L  iikk##%%%%%%%%% >OOO ))++    ! !!!!!!!!$N33I I & &&&&&&&& LL<>N>QRRR r*c *Kt|ttt jj|kt jjdkd{V}|rtd|ddS)z Verify that a workspace has no active sessions. Args: db: Database session workspace_name: Name of the workspace Raises: ConflictException: If active sessions exist in the workspace TNzCannot delete workspace 'z7': active session(s) remain. Delete all sessions first.) boolrBr r rCrSessionr, is_activer)r9r,has_active_sessionss r+check_no_active_sessionsrzs!%ii N1^CN,4           ! !  o o o o     r*c Ktd|ttjtjj|k}||d{V}|}|)td|ttj|j|j |j |j }t|tt!jtjjtjj|kd{Vpd}t|tt!jtjjtjj|kd{Vpd}t|tt!jtjjtjj|kd{Vpd}t|tt!jtjjtjj|kd{Vpd} |t1tjt!jtjjdd|kd{V|t1tjtjj|kd{Vttjjtjj|k} |t1tjtjj| d{V|ttjtjj|kd{V} | !} |t1tj"tj"j|kd{V|t1tjtjj|kd{V|t1tjtjj|kd{V|t1tjtjj|kd{V|t1tj#tj#j|kd{V|t1tj$tj$j|kd{V|t1tjtjj|kd{V|t1tjtjj|kd{V||d{V|%d{VtM} | r| 'd|} | (|d{Vt)d ||n3#tT$r&}td ||Yd}~nd}~wwxYw| D]}| 'd ||j+|j,} | (|d{Vt)d ||j+|j,h#tT$r&}td ||Yd}~d}~wwxYwt[|}|d}t]j/|d{Vt)d|nD#tT$r7t0d||1d{VwxYwte||||| S)a Delete a workspace. Args: db: Database session workspace_name: Name of the workspace Returns: WorkspaceDeletionResult containing a snapshot of the deleted workspace and cascade counts for deleted resources zDeleting workspace %sNzWorkspace %s not found)r=r>r@rArr/messagez8Deleted message embeddings namespace %s for workspace %sz4Failed to delete message embeddings namespace %s: %sdocumentz2Deleted document namespace %s for collection %s/%sz*Failed to delete document namespace %s: %s*zWorkspace %s deletedzFailed to delete workspace %s)rrrr r!)3rWwarningr rr&rCr=executescalar_one_or_nonerrr>r@rAr(rBr countPeerr<r,rwMessageDocumentrActiveQueueSession split_part work_unit_key QueueItem message_idin_ CollectionscalarsallMessageEmbeddingWebhookEndpoint SessionPeerrmrget_vector_namespacedelete_namespacerX Exceptionobserverobservedr3r delete_match exceptionrollbackr)r9r,reresultraworkspace_snapshot peers_countsessions_countmessages_countconclusions_countmessage_ids_subquerycollections_result collectionsexternal_vector_storemessage_namespacee collection doc_namespacersworkspace_patterns r+delete_workspacer*s  NN*N;;; &" # # ) )&*:*?>*Q R RD::d## # # # # # #F0022/@@@')))!*  "#.&4#. ii 4:fkn-- . . 4 4 *n<           Kii 4:fn/00 1 1 7 7-?           Nii 4:fn/00 1 1 7 7-?           Nii 4:fo011 2 2 8 8..@           6Djj 6, - - 3 3 9 GaPP!"           jj 6# $ $ * * />A            &fn&788>> N )^ ;  jj 6# $ $ * * +//0DEE           $&:: 6$ % % + +!0NB  $ $        )00226688 jj 6* + + 1 1'6.H           jj 6? # # ) )..@           jj 6$ % % + +!0NB           jj 6> " " ( (-?           jj 6) * * 0 0&5G           jj 6% & & , ,"1^C           jj 6> " " ( (-?           jj 6;   % %fk&@N&R S S         ii()))))))))iikk!: ; ; !'  5 J J>!!  +<<=NOOOOOOOOO N%"    J% *   5 J J"'' !! /@@OOOOOOOOOLLL%"+"+  !NND%(77 (OOO !2333333333 +^<<<<  +    kkmm  #$!''-    sdT d8&7`d8 a(a d8 a)d88Ab;:d8; c+c&!d8&c++A d88Ae9)N)? dataclassesrloggingrtypingrcashewsr sqlalchemyrrr r r sqlalchemy.excr sqlalchemy.ext.asyncior sqlalchemy.ormrsrcrrsrc.cache.clientrrrr src.configrsrc.exceptionsrrsrc.utils.filterrsrc.utils.typesrsrc.vector_storerr"rWrr1WORKSPACE_LOCK_PREFIXstrr3rPrQlockedDEFAULT_LOCK_TTL_SECONDSdictrErjrvr&r^tuplerfrhWorkspaceUpdatertrzrr)r*r+rs(!!!!!!;;;;;;;;;;;;;;))))))//////555555  GGGGGGGG))))))------666666 8     ?..00:::$ > -000     $ > 2555  &) #s(^d  . GIGIGIGI&GI  GI v'( GIGIGIGIV&*   #s(^d "  E&" #$