yj ddlZddlmZmZmZmZmZmZmZddl m Z ddl m Z ddl mZddlmZmZddlmZddlmZdd lmZmZdd lmZdd lmZmZdd lmZej e!Z"ed dgZ#e#$dej%eddeeefdedej&dedefdZ'e#$de ej%eedgeddefdej(dzdefd Z)e#*d!ej%eed"#gededd$efd"e+dej,defd%Z-e#.d!d&eed"#g'edefd"e+defd(Z/e#$d)e0ej1eed"#gededd*fd"e+d+ej2fd,Z3e#4d-ej5eed"#gededd.edd/edd0efd"e+d1e+dzd2e+dzd3e+dzdef d4Z6e#$d5d6eed"#g'ededd7fd"e+d8ej7fd9Z8dS):N) APIRouterBodyDepends HTTPExceptionPathQueryResponse)Page) apaginate) AsyncSession)crudschemas)settings)db)enqueue_deletion enqueue_dream)AuthenticationException) JWTParams require_auth)searchz /workspaces workspaces)prefixtags)response_model.zWorkspace creation parameters) descriptionresponse workspace jwt_paramsrcjK|jr'|js|j|jkrtdn"|jstd|j|_t j||d{V}|jd{V|d{V|jrdnd|_ |j S)z Get a Workspace by ID. If workspace_id is provided as a query parameter, it uses that (must match JWT workspace_id). Otherwise, it uses the workspace_id from the JWT. zUnauthorized access to resourcez0Workspace ID not found in query parameter or JWT)rN) nameadwrr get_or_create_workspacecommit post_commitcreated status_coderesource)rrrrresults >/DATA/AppData/hermes/projects/honcho/src/routers/workspaces.pyr&r&s ~ &} M!?!?)*KLL L| )B $ /iHHH H H H H H HF ")++     "(.933cH ?z/listT)admin)r dependenciesz8Filtering and pagination options for the workspaces listoptionscKd}|rt|dr|j}|ikrd}t|tj|d{Vd{VS)z4Get all Workspaces, paginated with optional filters.Nfilters)r3)hasattrr3r r get_all_workspaces)r1r filter_params r-r5r5:sL 77I..  2  L %l;;;;;;;;;      r.z/{workspace_id} workspace_idworkspace_namezUpdated workspace parameterscDKtj|||d{V}|S)z/Update Workspace metadata and/or configuration.)r9rN)r update_workspace)r7rrhoncho_workspaces r-r;r;RsM"2 <9 r.)r*r0cKtj||d{Vtj||d{Vt|d||d{V|jd{VddiS)aw Delete a Workspace. This accepts the deletion request and processes it in the background, permanently deleting all peers, messages, conclusions, and other resources associated with the workspace. Returns 409 Conflict if the workspace contains active sessions. Delete all sessions first, then delete the workspace. This action cannot be undone. r8Nr) db_sessionmessagezWorkspace deletion accepted)r get_workspacecheck_no_active_sessionsrr')r7rs r-delete_workspacerCfs(  R = = ========  '< H H HHHHHHHH <lr R R RRRRRRRR ")++ 4 55r.z/{workspace_id}/searchzMessage search parametersbodycfK|jpi}||d<t|j||jd{VS)z{ Search messages in a Workspace using optional filters. Use `limit` to control the number of results returned. r7)r3limitN)r3rqueryrF)r7rDr3s r-search_workspacerHsI l bG*GN G4:FFF F F F F F FFr.z/{workspace_id}/queue/statusz!Optional observer ID to filter byzOptional sender ID to filter byz Optional session ID to filter by observer_id sender_id session_idcK tj|||||d{VS#t$rN}tdt |t dt ||d}~wwxYw)a{ Get the processing queue status for a Workspace, optionally scoped to an observer, sender, and/or session. Only tracks user-facing task types (representation, summary, dream). Internal infrastructure tasks (reconciler, webhook, deletion) are excluded. Note: completed counts reflect items since the last periodic queue cleanup, not lifetime totals. )r9 session_nameobserverobservedNzInvalid request parameters: r*detail)r get_queue_status ValueErrorloggerwarningstrr)r7rIrJrKres r-rSrSs2 C* '#             CCC>c!ff>>???CFF;;;BCs# A;A A66A;z/{workspace_id}/schedule_dreamzDream scheduling parametersrequestc (Ktjjstdd|j}|j|jn|j}|j}t|||||jddd{Vt d|j ||||jdS) a Manually schedule a dream task for a specific collection. This endpoint bypasses all automatic dream conditions (document threshold, minimum hours between dreams) and schedules the dream task for a future execution. Currently this endpoint only supports scheduling immediate dreams. In the future, users may pass a cron-style expression to schedule dreams at specific times. rPz2Dreams are not enabled in the system configurationrQNmanual immediate)rNrO dream_typerMtrigger_reason delay_reasonz7Manually scheduled dream: %s for %s/%s/%s (session: %s)) rDREAMENABLEDrrNrOr^rrKrUinfovalue)r7rZrNrOr^s r-schedule_dreamres* > ! G    H#*#3#?wWEUH#J '              KKA r.)9loggingfastapirrrrrrr fastapi_paginationr !fastapi_pagination.ext.sqlalchemyr sqlalchemy.ext.asyncior srcr r src.configrsrc.dependenciesrsrc.deriver.enqueuerrsrc.exceptionsr src.securityrrsrc.utils.searchr getLogger__name__rUrouterpost WorkspaceWorkspaceCreater& WorkspaceGetr5putrWWorkspaceUpdater;deleterClistMessageMessageSearchOptionsrHget QueueStatusrSScheduleDreamRequestrer.r-rsRRRRRRRRRRRRRRRRRR######777777//////????????22222200000000######  8 $ $       R 122*. 8***$GLLNN33 &    32B )*',,T222334  ,04 T,,,   !D (    &$',,nEEEFFG  S )- 7***   &       ',,nEEEFFG S 666666  66(',,nEEEFFG  S )- 4*** G G G  & G G G  G "&',,nEEEFFG  S #e ="E$4UVVV"U <CCCtC Tz C d C CCC  CB$',,nEEEFFG  S ,0D 6---111  )111  111r.