yjU.dZddlZddlZddlmZmZmZmZddlm Z ddl Z ddl m Z m Z mZmZmZmZmZmZddlmZmZddlmZmZmZmZmZdZd Zd Zd ed efd Z ddde!e"efde#d dfdZ$d ed efdZ%ee!e"efee%fZ&Gdde Z'Gdde'Z(Gdde'Z)Gdde'Z*Gdde'Z+Gdde Z,Gd d!e,Z-Gd"d#e,Z.Gd$d%e,Z/Gd&d'e,Z0Gd(d)e Z1Gd*d+e Z2Gd,d-e Z3Gd.d/e Z4Gd0d1e Z5Gd2d3e5Z6Gd4d5e5Z7Gd6d7e5Z8Gd8d9e5Z9Gd:d;e Z:Gd<d=e Z;Gd>d?e Z<Gd@dAe<Z=GdBdCe<Z>GdDdEe<Z?GdFdGe<Z@GdHdIe ZAGdJdKe<ZBGdLdMe ZCGdNdOe<ZDGdPdQe ZEGdRdSe ZFGdTdUe ZGGdVdWe ZHGdXdYe ZIGdZd[e ZJGd\d]e ZKGd^d_e ZLGd`dae ZMGdbdce ZNGdddee ZOGdfdge ZPGdhdie ZQGdjdke ZRGdldmeRZSGdndoeRZTdS)pzPydantic schemas for API request/response validation. These schemas are consumed by the FastAPI routers and define the public API contract. N) AnnotatedAnySelfcast)urlparse) AliasChoices BaseModelBeforeValidator ConfigDictField PrivateAttrfield_validatormodel_validator)ReasoningLevelsettings) DreamTypeMessageConfigurationSessionConfigurationSessionPeerConfigWorkspaceConfigurationz^[a-zA-Z0-9_-]+$dvreturnct|tr|ddSt|trEt ttt f|}d|DSt|tr,t tt |}d|DS|S)zCRecursively strip NUL bytes from strings in nested data structures.cNi|]"\}}t|t|#S_sanitize_value).0kvals 7/DATA/AppData/hermes/projects/honcho/src/schemas/api.py z#_sanitize_value..1s.QQQVQ""OC$8$8QQQc,g|]}t|Srr r"items r% z#_sanitize_value..4s 666$%%666r') isinstancestrreplacedictrritemslist)rdlsts r%r!r!+s!S%yy$$$!TR c3h # #QQqwwyyQQQQ!T749a  66#6666 Hr'_current_depthdatar6c|tkrtdt|dkr0t|tkrtdtd|D]P}t |t r9ttt ttf||dzQdS)zHValidate metadata dict doesn't exceed key count or nesting depth limits.z*Metadata nesting exceeds maximum depth of r4zMetadata exceeds maximum of z top-level keysr5N) _METADATA_MAX_DEPTH ValueErrorlen_METADATA_MAX_KEYSvaluesr,r/_check_metadata_limitsrr-r)r7r6rs r%r>r>8s +++ N9L N N   s4yy+=== N+= N N N   [[]] a    "T#s(^Q''8J    r'ct|ts|Sttttf|}t |t |S)zJValidate and sanitize a metadata dict: enforce limits and strip NUL bytes.)r,r/rr-rr>r!)rr7s r%_validate_metadatar@MsL a   S#X " "D4   4  r'ceZdZdS) WorkspaceBaseN__name__ __module__ __qualname__rr'r%rBrB]Dr'rBceZdZUeeedddefed<iZe ed<ee Z e ed<e d Z d S) WorkspaceCreateidr4alias min_length max_lengthpatternnamemetadatadefault_factory configurationTpopulate_by_nameN)rDrErFrr-r RESOURCE_NAME_PATTERN__annotations__rR_SanitizedMetadatarrUr model_configrr'r%rIrIas   DQ3@UVVV X $&H %%%,1E.---M):t444LLLr'rIc4eZdZUdZeeefdzed<dS) WorkspaceGetNfiltersrDrErFr^r/r-rrYrr'r%r]r]n/%)GT#s(^d ")))))r'r]c8eZdZUdZedzed<dZedzed<dS)WorkspaceUpdateNrRrU)rDrErFrRrZrYrUrrr'r%rbrbrs>*.H 4'...37M)D077777r'rbceZdZUedZeed<eedZeee fed<eeZ eee fed<e j ed <e d d Z d S) WorkspacerJserialization_aliasrQrRrTrf h_metadatarSrU created_atTfrom_attributesrWN)rDrErFr rQr-rYr/rhrrUdatetimer r[rr'r%rdrdws$///D#///!&*"""JS#X%*E$$?$?$?M4S>???!!!!:tLLLr'rdceZdZdS)PeerBaseNrCrr'r%rnrnrGr'rnceZdZUeeedddefed<dZe dzed<dZ e ee fdzed<e d ZdS) PeerCreaterJr4rKrLrQNrRrUTrV)rDrErFrr-r rXrYrRrZrUr/rr r[rr'r%rprps   DQ3@UVVV X +/H 4'...+/M4S>D(///:t444LLLr'rpc4eZdZUdZeeefdzed<dS)PeerGetNr^r_rr'r%rrrrr`r'rrcHeZdZUdZedzed<dZeee fdzed<dS) PeerUpdateNrRrU) rDrErFrRrZrYrUr/r-rrr'r%rtrtsE*.H 4'...+/M4S>D(/////r'rtceZdZUedZeed<edZeed<ejed<ee dZ e ee fed <ee Z e ee fed <e d d ZdS)PeerrJrerQ workspace_idworkspace_namerirRrgrhrSrUTrjN)rDrErFr rQr-rYrxrlr/rhrrUr r[rr'r%rvrvs$///D#///%NCCCNCCCC!!!!!&*"""JS#X%*E$$?$?$?M4S>???:tLLLr'rvcBeZdZUeddZedzed<eddZedzed<eddZedzed<edd d d Z e dzed <edddd Z e dzed<eddZ edzed<edd d dZe dzed<dS)PeerRepresentationGetNz.sJ-7tS,A,AK VR(((tr')r,r1rrclsrs r%sanitize_peer_cardzPeerCardSet.sanitize_peer_cardsL a    cA.. r'N) rDrErFr rr1r-rYr classmethodrrrr'r%rrs 52PQQQItCyQQQ_[x00033[10r'rceZdZdS) MessageBaseNrCrr'r%rrrGr'rc~eZdZUeeedejfed<edZ eed<dZ e dzed<dZ e dzed <dZejdzed <eg Zeeed <edd ededefdZedeefdZed defdZdS) MessageCreaterrNrOcontentpeer_id)rM peer_nameNrRrUrir_encoded_messageafterrrrc.|ddSNrrr.rs r%sanitize_contentzMessageCreate.sanitize_contentyy$$$r'c|jS)N)r)selfs r%encoded_messagezMessageCreate.encoded_messages $$r'cptjd}||j}||_|S)N o200k_base)tiktoken get_encodingencoderr)rencodingrs r%validate_and_set_token_countz*MessageCreate.validate_and_set_token_counts2(66"//$,77 / r')rDrErFrr-r rMAX_MESSAGE_SIZErYrrRrZrUrrirlr rr1rrrrpropertyrrrrrr'r%rrsg sEEQ8;TUUUU VVVVU+++Is+++*.H 4'...15M'$.555+/J!D(///"-+b"9"9"9d3i999_YW---%%%%%[.-%%c%%%X%_'"""d#"r'rc4eZdZUdZeeefdzed<dS) MessageGetNr^r_rr'r%rrr`r'rc$eZdZUdZedzed<dS) MessageUpdateNrR)rDrErFrRrZrYrr'r%rrs(*.H 4'.....r'rceZdZUedZeed<eed<edZeed<edZeed<ee d Z e ee fed <e j ed <ed Z eed<eed<eddZdS)MessagerJre public_idrrrr} session_namerRrgrhrirwrx token_countTrjN)rDrErFr rr-rYrrr/rhrrlrxrr r[rr'r%rrsUt444Is444 LLLUy999Is999,???L#???!&*"""JS#X!!!!%NCCCNCCCC:tLLLr'rcFeZdZUdZedddZeeed<dS)MessageBatchCreatezFF$)E#!$L$L$LHd=!LLLLLr'rceZdZUdZeddZeed<dZe dzed<dZ e dzed<dZ e j dzed <ed ZdS) MessageUploadCreatez-Schema for message creation from file uploads.z#ID of the peer creating the messager{rNrRrUriTrV)rDrErFrr rr-rYrRrZrUrrirlr r[rr'r%rr.s775*OPPPGSPPP*.H 4'...15M'$.555+/J!D(///:t444LLLr'rceZdZdS) SessionBaseNrCrr'r%rr>rGr'rceZdZUeeedddefed<dZe dzed<edd Z e ee fdzed <dZ edzed <ed ZdS) SessionCreaterJr4rKrLrQNrRpeers)rrM peer_namesrUTrV)rDrErFrr-r rXrYrRrZrr/rrUrr r[rr'r%rrBs   DQ3@UVVV X +/H 4'...6;eDPW6X6X6XJS++,t3XXX15M'$.555:t444LLLr'rc4eZdZUdZeeefdzed<dS) SessionGetNr^r_rr'r%rrNr`r'rc8eZdZUdZedzed<dZedzed<dS) SessionUpdateNrRrU)rDrErFrRrZrYrUrrr'r%rrRs>*.H 4'...15M'$.55555r'rceZdZUedZeed<eed<edZeed<ee dZ e ee fed <ee Z e ee fed <e j ed <ed d ZdS)SessionrJrerQ is_activerwrxrRrgrhrSrUriTrjN)rDrErFr rQr-rYrrxr/rhrrUrlr r[rr'r%rrWs$///D#///OOO%NCCCNCCCC!&*"""JS#X%*E$$?$?$?M4S>???!!!!:tLLLr'rceZdZUedZeed<eddZeed<edd Z eed <ed Z eed <ed Z eed<edZ eed<dS)SummaryzThe summary textr{rz=The internal ID of the message that this summary covers up toT)r|exclude message_idz;The public ID of the message that this summary covers up tor|rfmessage_public_idz#The type of summary (short or long) summary_typez:The timestamp of when the summary was created (ISO format)riz(The number of tokens in the summary textrN) rDrErFr rr-rYrrrrrirrr'r%rrfs5%7888GS888eSJ#UQ(s*OPPPL#PPPePJu)STTTKTTTTTr'rceZdZUedZeed<eeed<eddZ e dzed<edd Z edzed <edd Z eedzed <e d d ZdS)SessionContextrJrerQrNzThe summary if availablersummaryz^A curated subset of a peer representation, if context is requested from a specific perspectivepeer_representationzBThe peer card, if context is requested from a specific perspectiverTrj)rDrErFr rQr-rYr1rrrrrr r[rr'r%rrws$///D#///7m#e"<GWt^',et'''t#(%X###ItCy4 :tLLLr'rceZdZUdZedZeed<edZeed<edd Z edzed <edd Z e edzed <dS) PeerContextz;Context for a peer, including representation and peer card.zThe ID of the peerr{rz(The ID of the target peer being observed target_idNzYA curated subset of the representation of the target peer from the observer's perspectiverrzAThe peer card for the target peer from the observer's perspectiver) rDrErFrr rr-rYrrrr1rr'r%rrsEE5%9:::GS:::U'QRRRIsRRR!&o"""NC$J#(%W###ItCy4r'rceZdZUedZeed<eddZedzed<eddZ edzed <e d d Z dS) SessionSummariesrJrerQNzThe short summary if availabler short_summaryzThe long summary if available long_summaryTrj) rDrErFr rQr-rYrrrr r[rr'r%rrs$///D#///$)E"B%%%M7T>$)5"A$$$L'D.:tLLLr'rc8eZdZUdZdZeeefdzed<dS) ConclusionGetz5Schema for listing conclusions with optional filters.Nr^) rDrErFrr^r/r-rrYrr'r%rrs5??%)GT#s(^d ")))))r'rceZdZUdZeed<eed<eddZeed<edd Zeed <ed d Z ed zed<e j ed<e ddZ d S) Conclusionz2Conclusion response - external view of a document.rJrz The peer who made the conclusion observer_idrobserver The peer the conclusion is about observed_idobservedNr})rrfrriTrj) rDrErFrr-rYr rrrrlr r[rr'r%rrs<< GGG LLLE6)HcE6)Hc %uT|TTTL#*TTT!!!!:LLLr'rceZdZUdZeddZeed<edddd Ze ed <ed d dd Z e d zed<ed dZ e eefd zed<d S)ConclusionQueryz4Query parameters for semantic search of conclusions..zSemantic search queryr{query r4rNumber of results to returnrtop_kNrrz-Maximum cosine distance threshold for resultsdistancezAdditional filters to applyrr^)rDrErFrr rr-rYrrrrr^r/rrr'r%rrs>>s(?@@@E3@@@ 1 E3 #U  C Hedl &+U1&&&GT#s(^d "r'rcNeZdZUdZeeeddfed<eddZeed <edd Z eed <ed d Z ed zed<e dZ e ed<eddededefdZeddefdZd S)ConclusionCreatez(Schema for creating a single conclusion.r4irr.zThe peer making the conclusionr{rrrNz5A session ID to store the conclusion in, if specifiedrr}rr _token_countrrrrc.|ddSrrrs r%rz!ConclusionCreate.sanitize_contentrr'ctjd}||j}t ||_|jt jjkr0tdt jjdzd|jdz|S)z;Validate that content doesn't exceed embedding token limit.rz1Content exceeds maximum embedding token limit of  z(got z tokens)) rrrrr;rr EMBEDDINGMAX_INPUT_TOKENSr:)rrtokenss r%validate_token_countz%ConclusionCreate.validate_token_counts(66..KK  x1B B BC'8;;;<5$+5556   r')rDrErFrrr-r rYrrr}r rrrrrrrrrr'r%rrsL22 sEEQ5AAAA BBBBuS.NOOOKOOOuS.PQQQKQQQ"UKJd  $ A...L#..._YW---%%%%%[.-%_'""" d   #"   r'rc \eZdZUdZedddeddZeee d<dS) ConclusionBatchCreatezCSchema for batch conclusion creation with a max of 100 conclusions..r4r conclusions observations)rNrOvalidation_aliasN) rDrErFrr rrr1rrYrr'r%rrsaMM*/% %m^DD +++K&'r'rceZdZUeeeddfed<eddZeee fdzed<ed d d d Z e ed<e dde dedefdZdS)MessageSearchOptions.z Search queryr{rNzFilters to scope the searchrr^rr4rrrlimitrrrrc.|ddSrrrs r%sanitize_queryz#MessageSearchOptions.sanitize_queryrr')rDrErFrr-r rYr^r/rr rrrrrr'r%r r s S%%@@@@ AAAA%*U"?&&&GT#s(^d " 1 E3_W7+++%s%s%%%[,+%%%r'r ceZdZUeddZedzed<eddZedzed<eeeddd fed <d Z e ed <eddZ e ed<e d dededefdZdS)DialecticOptionsNz0ID of the session to scope the representation tor{r}zNOptional peer to get the representation for, from the perspective of this peerr~r4i'zDialectic API Prompt)rNrOr|rFstreamlowz?Level of reasoning to apply: minimal, low, medium, high, or maxrreasoning_levelrrrrc.|ddSrrrs r%rzDialecticOptions.sanitize_query9rr')rDrErFr r}r-rYr~rrrrrrrrrr'r%rr(s4"U LJd  dFC$J  UUaE?U V V VV FD&+eU'''O^ _W7+++%s%s%%%[,+%%%r'rc eZdZUedzed<dS)DialecticResponseNrrrr'r%rr?s" 4Zr'rc(eZdZUdZdZedzed<dS)DialecticStreamDeltaz/Delta object for streaming dialectic responses.Nr)rDrErFrrr-rYrr'r%rrCs,99GS4Zr'rc,eZdZUdZeed<dZeed<dS)DialecticStreamChunkz(Chunk in a streaming dialectic response.deltaFdoneN)rDrErFrrrYrrrr'r%rrMs422 D$r'rceZdZUdZeddZedzed<edZe ed<ed Z e ed <ed Z e ed <ed Z e ed<dS)SessionQueueStatusz:Status for a specific session within the processing queue.Nz!Session ID if filtered by sessionrr}Total work unitsr{total_work_unitszCompleted work unitscompleted_work_units$Work units currently being processedin_progress_work_units"Work units waiting to be processedpending_work_units) rDrErFrr r}r-rYr rr!r#r%rr'r%rrYsDD"U7Jd "E.@AAAcAAA %2H I I I#III"'%:###C$e0TUUUUUUUUr'rceZdZUdZedZeed<edZeed<edZ eed<ed Z eed <ed d Z e e efd zed<d S) QueueStatusa4Aggregated processing queue status. Tracks user-facing task types only: representation, summary, and dream. Internal infrastructure tasks (reconciler, webhook, deletion) are excluded. Note: completed_work_units reflects items since the last periodic queue cleanup, not lifetime totals. rr{r z2Completed work units (since last periodic cleanup)r!r"r#r$r%Nz/Per-session status when not filtered by sessionrsessions)rDrErFrr r rrYr!r#r%r(r/r-rrr'r%r'r'hs"E.@AAAcAAA %H!!!##(%:###C$e0TUUUUUU5:UE666Hd3**+d2r'r'ceZdZUeddZeed<eddZedzed<eddZe ed <edd Z edzed <dS) ScheduleDreamRequest.zObserver peer namer{rNz:Observed peer name (defaults to observer if not specified)rzType of dream to schedule dream_typez-Session ID to scope the dream to if specifiedr}) rDrErFr rr-rYrr+rr}rr'r%r*r*sE#+?@@@Hc@@@ 5 VHcDj"E#3NOOOJ OOO"U IJd r'r*ceZdZdS)WebhookEndpointBaseNrCrr'r%r-r-rGr'r-c\eZdZUeed<edededefdZdS)WebhookEndpointCreateurlrrc8t|}t|j|jgst d|jdvrt d|jrA t j|j}|jrt dn#t$rYnwxYw|S)NzInvalid URL format)httphttpsz$Only HTTP and HTTPS URLs are allowedz$Private IP addresses are not allowed) rallschemenetlocr:hostname ipaddress ip_address is_private)rrparsedr9s r%validate_webhook_urlz*WebhookEndpointCreate.validate_webhook_urls!FM6=122 3122 2 = 1 1 1CDD D ?  &1&/BB (M$%KLLLM    s/B BBN)rDrErFr-rYrrr<rr'r%r/r/s` HHH_USS[r'r/czeZdZUeed<edZedzed<eed<ejed<edd Z dS) WebhookEndpointrJrwreNrxr0riTrj) rDrErFr-rYr rxrlr r[rr'r%r>r>sf GGG!&>!J!J!JNC$JJJJ HHH!!!!:dTJJJLLLr'r>)Urrlr8typingrrrr urllib.parserrpydanticrr r r r r rr src.configrrsrc.schemas.configurationrrrrrrXr<r9r!r/r-rr>r@rZrBrIr]rbrdrnrprrrtrvrzrrrrrrrrrrrrrrrrrrrrrrrrr rrrrrr'r*r-r/r>rr'r%rDs  ------------!!!!!!                    0///////,  s  s      sCx. *!#!#!!!!tCH~?Q/R/RRS     I    5 5 5 5 5m 5 5 5*****=***88888m888        $     y   55555555*****h***00000000     8   !!!!!I!!!HYy     )   &     )   K6********/////K///k"MMMMMMMM 55555)555      )    5 5 5 5 5K 5 5 5********66666K666     k   UUUUUiUUU"[(     )        {   &*****I*** ,i,yDI %%%%%9%%%.%%%%%y%%%. 99 V V V V V V V V):9      )   /6KKKKK)KKKKKr'