yjlddlmZddlmZmZddlmZmZddlmZm Z m Z GddeZ Gdde Z Gd d e Z Gd d e ZGd de ZGdde ZGdde ZGdde ZdedeeefdeeeffdZdddddddededededzdedzd edzd!edzd"edzdeeeffd#Zd$ed%d&edeeeffd'Z d0ddd(d)eeefd*e d+ed,d-edzd.eedzdedzdeeeffd/ZdS)1)datetime)AnyLiteral) BaseModel ConfigDict) DreamTypeReconcilerTypeResolvedConfigurationc*eZdZdZedZdS) BasePayloadz Base payload with common fields.forbid)extraN)__name__ __module__ __qualname____doc__r model_config?/DATA/AppData/hermes/projects/honcho/src/utils/queue_payload.pyr r s&**:H---LLLrr cveZdZUdZdZeded<eed<eed<eeed<eed<e ed<e ed <d S) RepresentationPayloadz!Payload for representation tasks.representation task_type session_namecontent observersobserved created_at configurationN) rrrrrr__annotations__strlistrr rrrrrst+++;Iw'(;;; LLLCyMMM((((((rrc*eZdZUdZeeed<dS)RepresentationPayloadsz,Payload for a batch of representation tasks.payloadsN)rrrrr#rr!rrrr%r%s*66())))))rr%c`eZdZUdZdZeded<eed<eed<e ed<dZ edzed<dS) SummaryPayloadzPayload for summary tasks.summaryrrmessage_seq_in_sessionr Nmessage_public_id) rrrrrrr!r"intr r+rrrr(r(!sf$$$-Iwy!---(((($(sTz(((((rr(cReZdZUdZdZeded<eed<eee fed<dS)WebhookPayloadz#Payload for webhook delivery tasks.webhookr event_typedataN) rrrrrrr!r"dictrrrrr.r.,sJ--$-Iwy!---OOO sCx.rr.ceZdZUdZdZeded<eed<eed<eed<dZ edzed<dZ edzed <dZ edzed <dZ e dzed <dZe dzed <dS) DreamPayloadzPayload for dream tasks.dreamr dream_typeobserverrNrtrigger_reason delay_reason&documents_since_last_dream_at_scheduledocument_threshold)rrrrrrr!rr"rr8r9r:r,r;rrrr4r44s""")Iww)))MMMMMM#L#*### "&NC$J%%%#L#*###9=*C$J===%)d )))))rr4cNeZdZUdZdZeded<eded<eed<dS)DeletionPayloadzPayload for deletion tasks.deletionrsession observation workspace deletion_type resource_idN)rrrrrrr!r"rrrr=r=GsK%%%/Iwz"///@AAAArr=c8eZdZUdZdZeded<eed<dS)ReconcilerPayloadzHPayload for reconciler tasks (vector sync, queue cleanup, self-healing). reconcilerrreconciler_typeN)rrrrrrr!r rrrrFrFOs9RR'3Iw|$333######rrFr0r1returncNt||ddS)zCreate a webhook payload.r0r1jsonTmode exclude_none)r. model_dumprKs rcreate_webhook_payloadrQVs4 Zd ; ; ; F F $ G  rN)rr8r9r:r;r6r7rrr8r9r:r;c Zt||||||||ddS)zCreate a dream payload.r6r7rrr8r9r:r;rLTrM)r4rPrSs rcreate_dream_payloadrT`sF !%!/U-   jf4j00 1rrCr?rDcNt||ddS)zCreate a deletion payload.rCrDrLTrM)r=rPrVs rcreate_deletion_payloadrWxs4 #   jf4j001r)rrmessager r)rr)r*rcj|d}|d}|d}t|tstdt|tstdt|tstd |dkr|d} |d } t| tstd t| t std |t |d krtd|tdt| || |||} ny|dkrs|td|d} | 8t| tr| stdt|||| } | dd} n2#t$r%}tdt||d }~wwxYw| S)a Create a processed payload from a message for queue processing. Note: workspace_name and message_id are no longer included in the returned payload as they are now stored in dedicated columns on the queue table. The caller is responsible for extracting and passing these values separately. Args: message: The original message dictionary task_type: Type of task ('representation' or 'summary') message_seq_in_session: Required for summary tasks, must be None for representation observers: List of observer peer names (required for representation tasks) observed: Name of the observed peer (*always* the peer who sent the message) (required for representation tasks) Returns: Processed payload dictionary ready for queue processing (without workspace_name and message_id) Raises: ValueError: If the payload doesn't match the expected schema workspace_namer message_idzWorkspace name must be a stringzSession name must be a stringzMessage ID must be an integerrrrz Message content must be a stringz$created_at must be a datetime objectNrz.observers is required for representation tasksz-observed is required for representation tasks)rrrrrr r)z4message_seq_in_session is required for summary tasksr+z8message_public_id must be a non-empty string if provided)rr*r r+rLTrMz Failed to create valid payload: ) get isinstancer" TypeErrorr,rlen ValueErrorrstripr(rP Exception)rXr rr*rrrZrr[rrvalidated_payloadr+payloades rcreate_payloadrfsx<[[!122N;;~..L\**J nc * *;9::: lC ( (97888 j# & &978880M ( ( (kk),,G \22Jgs++ D BCCCj(33 H FGGG C NNa$7$7 !QRRR !PQQQ 5)%#!+ !!!  ) # #%- !WXXX ' ,? @ @  ,0#66->O>U>U>W>W-!N!/)'=+"3 !!! $..F.NN MMMDCFFDDEE1LM Ns-EH H0 H++H0)N)rtypingrrpydanticrr src.schemasrr r r rr%r(r.r4r=rFr"r2rQr,rTrWr#rfrrrrjs********HHHHHHHHHH.....)... ) ) ) ) )K ) ) )*****[*** )))))[)))[*****;***&k$$$$$ $$$ sCx. #s(^ $!%#9=%)11111 1 * 1 $J 1*1-0$J1d 1 #s(^111101@A11 #s(^1111*. ^ #'^^^ #s(^^(^23^ $J ^ Cy4 ^Dj^ #s(^^^^^^^r