yjF ddlZddlZddlmZddlmZddlmZddlm Z m Z m Z m Z m Z mZddlmZddlmZddlmZdd lmZdd lmZmZdd lmZdd lmZmZdd lm Z m!Z!ddl"m#Z#ddl$m%Z%m&Z&ddl'm(Z(m)Z)ddl*m+Z+ddl,m-Z-m.Z.m/Z/ddl0m1Z1ddl2m3Z3ej4e5Z6e ddgZ7e78deej9e e)dge de ddefde:dej;dzd efd!Ze7?d)ej9e e)dd*+ge de de dd,efde:d*e:d&ej@d efd-ZAe78d.eejBe e)dd*+ge de de dd/efde:d*e:dejCdzd efd0ZDe78d1d2d3d4ejEFiid5iie e)dd*+g6e de de dfde:d*e:dejGfd7ZHe78d8ejIe e)dd*+ge de de dd9fde:d*e:dejJfd:ZKe7Ld;ejMe e)dd*+ge de dd<e dd=efde:d*e:d>e:dzd efd?ZNe7?d;ejMe e)dd*+ge de dd<e dd@e ddAefde:d*e:dBejOd>e:dzd ef dCZPe7LdDejQe e)dd*+ge de dd<e ddEe ddFe ddGdHdIJe ddKdLdMJe dNdOPe ddGdHdQJef de:d*e:d>e:dzdRe:dzdSeRdzdTeSdzdUeTdVeRdzd efdWZUe78dXeVejWe e)dd*+ge de de ddYfde:d*e:dZejXfd[ZYdS)\N) AsyncIterator)suppress) perf_counter) APIRouterBodyDependsPathQueryResponse)StreamingResponse)Page) apaginate) AsyncSession)crudschemas)settings)db tracked_db) agentic_chatagentic_chat_stream)embedding_client)AuthenticationExceptionResourceNotFoundException) JWTParams require_auth)prometheus_metrics)EmbeddingCallPurposeGetContextEventemit)search)embedding_call_purposez /workspaces/{workspace_id}/peerspeers)prefixtagsz/list workspace_id)workspace_name)response_model dependencies.z$Filtering options for the peers list) descriptionoptionsrcKd}|rt|dr|j}|ikrd}t|tj||d{Vd{VS)z?Get all Peers for a Workspace, paginated with optional filters.Nfilters)r&r,)hasattrr,rr get_peers)r%r*r filter_params 9/DATA/AppData/hermes/projects/honcho/src/routers/peers.pyr.r.!sL 77I..  2  L nL,OOOOOOOOO      )r'zPeer creation parametersresponsepeer jwt_paramscK|js!|j|j|krtd|jr.|js&|j|j|jkrtdn"|jstd|j|_t j|||gd{V}|jd{V|d{V|j rdnd|_ |j dS)z Get a Peer by ID or create a new Peer with the given ID. If peer_id is provided as a query parameter, it uses that (must match JWT workspace_id). Otherwise, it uses the peer_id from the JWT. NzUnauthorized access to resourcez+Peer ID not found in query parameter or JWTr&r"r) adwrnameprget_or_create_peerscommit post_commitcreated status_coderesource)r3r%r4r5rresults r0get_or_create_peerrE:s8$ =IZ\5*,,:V:V%&GHHH y!} M!9jldi>W>W)*KLL L| Y)*WXX XL + <vF ")++     "(.933cH ?1 r1z /{peer_id}peer_id)r& peer_namezUpdated peer parameterscFKtj||||d{V}|S)z.Update a Peer's metadata and/or configuration.)r&rGr4N)r update_peer)r%rFr4r updated_peers r0rIrI`sN) <7L r1z/{peer_id}/sessionsz'Filtering options for the sessions listcKd}|rt|dr|j}|ikrd}t|tj|||d{Vd{VS)z=Get all Sessions for a Peer, paginated with optional filters.Nr,)r&rGr,)r-r,rrget_sessions_for_peer)r%rFr*rr/s r0rLrLts L 77I..  2  L ('                r1z/{peer_id}/chatr9contentschema)zapplication/jsontext/event-stream) responsesr(c Ktd4d{V}tj||tj|gd{V}|d{Vdddd{Vn#1d{VswxYwY|d{V|jrdttdttfd}tj j rtj||jt!|t#||j|j||j|jn||j d St+||j|j||j|jn||j d{V}tj j rtj||jtj|r|nd S) z Query a Peer's representation using natural language. Performs agentic search and reasoning to comprehensively answer the query based on all latent knowledge gathered about the peer from their messages and conclusions. zpeers.chat.get_or_create_peerN)r<r7chunksreturncK|23d{V}dtjd|idddWV(6dtjddidWVdS) zFormat chunks as SSE events.Nzdata: rMF)deltadonez rVT)jsondumps)rRchunks r0format_sse_streamzchat..format_sse_streams & ^ ^ ^ ^ ^ ^ ^e]tzY4FPU*V*VWW]]]]]]]] &;4:vtn55;;; ; ; ; ; ; ;s-)r&reasoning_level)r& session_namequeryobserverobservedr[rO) media_type)rM)rrr>r PeerCreater?r@streamrstrrMETRICSENABLEDrrecord_dialectic_callr[r r session_idr]targetrDialecticResponse)r%rFr*peer_db peers_resultrZr3s r0chatrls49::g!5 '%73334         nn   " " $ $$$$$$$$~  <!#& < 3  < < < <   #   4+ ' 7     !  ##/!(!3!-$/6~/IW^^w$+$;   +    "#'m$+>#=7/         H 0'#3     $-KXXt L L LLsA A66 BBz/{peer_id}/representationz+Options for getting the peer representationc K d}|jrtt5ttjj|d5tj|jd{V}dddn #1swxYwYdddn #1swxYwYtj |||j |j n||j |j||j |j|j|jnd|j|jnt"jjd d{V}t)j|S#t.$rB}t0d|dt5|t7d |d}~wwxYw) aGet a curated subset of a Peer's Representation. A Representation is always a subset of the total knowledge about the Peer. The subset can be scoped and filtered in various ways. If a session_id is provided in the body, we get the Representation of the Peer scoped to that Session. If a target is provided, we get the Representation of the target from the perspective of the Peer. If no target is provided, we get the omniscient Honcho Representation of the Peer. Napir&parent_categoryF r^r_r\include_semantic_query embeddingsemantic_search_top_ksemantic_search_max_distanceinclude_most_derivedmax_observationsrp)representationz&Failed to get representation for peer : zPeer or session not found) search_queryr Exceptionr!r SEARCH_MEMORYvaluerembedrget_working_representationrhrg search_top_ksearch_max_distanceinclude_most_frequentmax_conclusionsrDERIVER'WORKING_REPRESENTATION_MAX_OBSERVATIONSrRepresentationResponseformat_as_markdown ValueErrorloggerwarningrcr)r%rFr*rsrxes r0get_representationrsk,$L(,   O## O O&(6<#/$) O O#3"89M"N"NNNNNNN  O O O O O O O O O O O O O O O O O O O O O O O O O O O O O O $> '.~'AW^^w +#*#7")"6)0)D,8")!>!>&2%44!I!         "-)<<>>     LLLSSS3q66SSTTT'(CDD!KLs_D*"B A/# B/A3 3B6A3 7B: D*B  D* B BD** E64=E11E6z/{peer_id}/cardzID of the observer peerz~Optional target peer to retrieve a card for, from the observer's perspective. If not provided, returns the observer's own cardrhcxK||n|}tj||||d{V}tj|S)zGet a peer card for a specific peer relationship. Returns the peer card that the observer peer has for the target peer if it exists. If no target is specified, returns the observer's own peer card. Nr^r_ peer_card)r get_peer_cardrPeerCardResponse)r%rFrhrr_rs r0rr*sj, +vvH( L7XI  #i 8 8 88r1zPeer card data to setzvOptional target peer to set a card for, from the observer's perspective. If not provided, sets the observer's own cardpeer_card_datacK||n|}tj|||j||d{Vtj|jS)zSet a peer card for a specific peer relationship. Sets the peer card that the observer peer has for the target peer. If no target is specified, sets the observer's own peer card. N)rr^r_r)r set_peer_cardrrr)r%rFrrhrr_s r0rrHsy2 +vvH    *      #n.F G G GGr1z/{peer_id}/contextzOptional target peer to get context for, from the observer's perspective. If not provided, returns the observer's own context (self-observation)zJOptional query to curate the representation around semantic search resultsdzcOnly used if `search_query` is provided. Number of semantic-search-retrieved conclusions to include)geler)gg?z_Only used if `search_query` is provided. Maximum distance for semantically relevant conclusionsTzFWhether to include the most frequent conclusions in the representation)defaultr)z>Maximum number of conclusions to include in the representationrzrrrrc xK||n|} t} d} |rtt5ttjj|d5tj|d{V} dddn #1swxYwYdddn #1swxYwYtj ||| d|| |||||ntj j d d{V} tj|||| d{V} tj|| | | }t%t'|d|| t)|j| du|du||||t| z dz |S#t,$rB}t.d |d t3|t5d |d}~wwxYw) a Get context for a peer, including their representation and peer card. This endpoint returns a curated subset of the representation and peer card for a peer. If a target is specified, returns the context for the target from the observer peer's perspective. If no target is specified, returns the peer's own context (self-observation). This is useful for getting all the context needed about a peer without making multiple API calls. Nrnrorqr)rF target_idrxrr4i) r& context_scoperG target_namehas_representation has_peer_cardsearch_query_providedrrrrtotal_duration_mszFailed to get context for peer ryzPeer not found)rrr{r!rr|r}rr~rrrrrrr PeerContextrrrboolrxrrrrcr)r%rFrhrzrrrrrr_context_startedrsrxrr3rs r0get_peer_contextrnsl +vvH"nnO<A(,  G## G G&(6<#/$) G G#3"8"F"FFFFFFF  G G G G G G G G G G G G G G G G G G G G G G G G G G G G G G $> #/".)>     +$!$#'(?#@#@'t3&2$&>)$7&; /#/>>O#Ct"K       AAALLLCFFLLMMM'(899q@As_E-"BA9- B9A= =BA= B E-BE-BCE-- F97=F44F9z/{peer_id}/searchzQMessage search parameters. Use `limit` to control the number of results returned.bodycpK|jpi}||d<||d<t|j||jd{VS)zBSearch a Peer's messages, optionally filtered by various criteria.r%rF)r,limitN)r,r r]r)r%rFrr,s r0 search_peerrsS"l bG*GN GI G4:FFF F F F F F FFr1)ZrWloggingcollections.abcr contextlibrtimerfastapirrrr r r fastapi.responsesr fastapi_paginationr !fastapi_pagination.ext.sqlalchemyrsqlalchemy.ext.asynciorsrcrr src.configrsrc.dependenciesrrsrc.dialectic.chatrrsrc.embedding_clientrsrc.exceptionsrr src.securityrr src.telemetryrsrc.telemetry.eventsrrrsrc.utils.searchr src.utils.typesr! getLogger__name__rrouterpostPeerrcPeerGetr.rarEput PeerUpdaterISession SessionGetrLrimodel_json_schemaDialecticOptionsrlrPeerRepresentationGetrgetrr PeerCardSetrrintfloatrrlistMessageMessageSearchOptionsrr1r0rs  ))))))CCCCCCCCCCCCCCCC//////######777777//////++++++++@@@@@@@@111111MMMMMMMM00000000,,,,,,LLLLLLLLLL######222222  8 $ $  -       %',,nEEEFFG  S &*d @'''  _t #    (<  S #tC5OPPP#GLLNN33       D< NiPPPQQ S 499#tC5NOOO             ( NiPPPQQ S 499)- C***   $ &  4 g7IIKK%&(     NiPPPQQ "S 499(,S EMEMEM EM %EMEMEM! EMP1 NiPPPQQ S 499-1T F...3L3L3L 3L *3L3L3L3Ll+ NiPPPQQ S 4)BCCC U999 9 $J9 9999.+ NiPPPQQ S 4)BCCC*.$ 0+++ MHHH H'H $J H HHHH>& NiPPPQQ S 4)BCCC g %u `    %u y     ).  u ))) #(%\####(% T ### CnAnAnA nA $JnA*nA*nA"#nA. /nA64Z7nAB CnAnAnAnAb( NiPPPQQ S 499)- g*** G G G  G  & G G G G G Gr1