yjrddlZddlmZddlmZddlmZmZmZm Z m Z m Z ddl m Z ddlmZddlmZddlmZmZmZdd lmZdd lmZdd lmZdd lmZdd lmZddl m!Z!m"Z"m#Z#ddl$m%Z%m&Z&ddl'm(Z(m)Z)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4ej5e6Z7eddgZ8dddede9de9dzde9de9de9dzd e:dzd!e;dzd"edede9de9de9d%e=e9dzf d'Z?dede9d(e9d)e:d*eejKd?e%def d@ZLe8MdAejHee&dd(Bg4e d5e d5ed5dC7efde9d(e9d>ejNdefdDZOe8PdAdEee&dd(BgFe d5e d5efde9d(e9defdGZQe8GdHejHdIee&dd(BgJe d5e d5e ddK7efde9d(e9dLe9dzdefdMZRe8GdNejHee&dd(Bg4e d5e d5ed5dO7efde9d(e9dPeSe9ejTfdefdQZUe8MdNejHee&dd(Bg4e d5e d5ed5dR7efde9d(e9dPeSe9ejTfdefdSZVe8PdNejHee&dd(Bg4e d5e d5ed5dT7efde9d(e9dPe=e9defdUZWe8XdVejTee&dd(Bg4e d5e d5e d5efde9d(e9dWe9defdXZYe8MdVdYdee&dd(BgZe d5e d5e d5ed5d[7efde9d(e9dWe9d\ejTdef d]ZZe8XdNe ej[ee&dd(Bg4e d5e d5efde9d(e9defd^Z\e8Xd_ej]ee&dd(Bg4e d5e d5ee dej^j_d`ej^j_dabfe ddc7e dddedfge ddh7e ddi7e djdkle ddmdndope ddqdrdspe djdtle ddmdndupdv de9d(e9dedwe:dzdxe9dzd*e_get_peer_card_taskrCYsL&# #       r@ session_id token_limitinclude_summarycjKtj|||||d{V\}}d|D}||fS)aZ Get session context Args: db: Database session to use for queries workspace_id: The workspace identifier session_id: The session identifier token_limit: Maximum tokens for the context include_summary: Whether to include summary if available Returns: Tuple of (summary, messages) )r3r+rErFNcLg|]!}tj|"SrMessagemodel_validate.0msgs r> z-_get_session_context_task..s(OOOsw55c::OOOr@)rget_session_context)rr'rDrErFsummarymessagesmessage_schemass r>_get_session_context_taskrUtsr()< #' GXPOhOOOO O ##r@cKtj|||d{V\}}|rtj|nd}|rtj|nd}||fS)z Fetch both short and long summaries. Returns: Tuple of (short_summary, long_summary) as Pydantic schemas. r3r+N)rget_both_summariesto_schema_summary)rr'rD short_rawlong_rawshortlongs r>_get_both_summaries_taskr^s!+ = rPz2_get_messages_for_context_task..s( D D DCGO * *3 / / D D Dr@)rget_messages_id_range)rr'rDr_rErSs r>_get_messages_for_context_taskrcss(a /  H E D8 D D DDr@ short_summary long_summaryc|r|dkrddt|dfSt|dz}|r|jnd}|r|jnd}|r||kr||kr ||j||z fS|r||kr|dkr ||j||z fSdd|fS)a Pick the best summary that fits within the token budget using 40/60 allocation. Args: short_summary: The short summary, or None long_summary: The long summary, or None token_limit: Total token budget for summary + messages include_summary: Whether summaries should be considered Returns: Tuple of (chosen_summary, messages_start_id, messages_token_budget) rNg?)maxint token_count message_id)rdrerErFsummary_budgetlong_len short_lens r>_select_summary_for_contextrns$ ,kQ..QK++++s*++N+7>|''QH-:A ))I N22x)7K7K   # ( "   n44Q   $ ) #  K r@z/list)r3)response_model dependencies.z6Filtering and pagination options for the sessions list) descriptionoptionscKd}|r&t|dr|jr|j}|ikrd}t|tj||d{Vd{VS)zBGet all Sessions for a Workspace, paginated with optional filters.Nfilters)r3rt)hasattrrtr r get_sessions)r'rrr filter_params r>rvrvsL 77I.. 7?  2  L $#<VVV V V V V V V      r@)rozSession creation parametersresponsesession jwt_paramsc 0K|js!|j|j|krtd|jr.|js&|j|j|jkrtdn"|jstd|j|_ t j|||d{V}|jrdnd|_|j S#t$rT}t d|jdt|tt||d}~wwxYw) z Get a Session by ID or create a new Session with the given ID. If Session ID is provided as a parameter, it verifies the Session is in the Workspace. Otherwise, it uses the session_id from the JWT for verification. NzUnauthorized access to resourcez.Session ID not found in query parameter or JWT)r3rzz Failed to get or create session : )adwrnamesrget_or_create_sessioncreated status_coderesource ValueErrorloggerwarningstrr)ryr'rzr{rresultes r>rr s[( =IZ\5*,,:V:V%&GHHH| $  M (  ,,)*KLL L| )@ "| 11 |W         '-n=ss# 111R',RR#a&&RRSSS!#a&&))q01s3B77 DADDz /{session_id}rWzUpdated session parametersc K tj||||d{V}|S#t$rB}td|dt |t d|d}~wwxYw)z1Update a Session's metadata and/or configuration.)r3r+rzNzFailed to update session rSession not found)rupdate_sessionrrrrr)r'rDrzrupdated_sessionrs r>rr<s D $ 3 |*g! ! !        DDDI:IIQIIJJJ'(;<delete_sessionrVsH(D(ZFFFFFFFF!'#"           bikk 1, K KLLLLLLLLL GTTT9:: DDDI:IIQIIJJJ'(;<rrs" D#1 '",(            5zBBB DDDH*HHAHHIII'(;<add_peers_to_sessionrs" D1 )(           DDDOOOs1vvOOPPP'(;<set_session_peersrs*R( '#           1 )z:::'           =B;;Cz+List of peer IDs to remove from the sessionc K tj|||t|d{Vtj|t j||d{V}t d||jS#t$rB}t d|dt|td|d}~wwxYw) z"Remove Peers by ID from a Session.rNrrz*Removed peers from session %s successfullyz$Failed to remove peers from session rr) rremove_peers_from_sessionsetrrrrrrrrrrrs r>rrs! D, '#5zz           1 )z:::'           A:NNN DDDTjTTCPQFFTTUUU'(;<rrsL% #       r@)rrorpzNew peer configurationrc &K tj|||||d{Vtd||dS#t$rE}td|d|dt |td|d}~wwxYw)z.Set the configuration for a Peer in a Session.)r3r+ peer_namerNz1Set peer config for %s in session %s successfullyzFailed to set peer config for z in session rr)rset_peer_configrrrrrr)r'rDrrrrs r>rr&s D" '#             ?*      DDD XW X X* X XPSTUPVPV X X   ((;<get_session_peersrHsD 7'j         r;///////// DDDQ:QQQQQRRR'(;<A query string used to fetch semantically relevant conclusionsTzIWhether or not to include a summary *if* one is available for the sessionrR)defaultrqaliasaThe target of the perspective. If given without `peer_perspective`, will get the Honcho-level representation and peer card for this peer. If given with `peer_perspective`, will get the representation and card for this peer *from the perspective of that peer*.zA peer to get context for. If given, response will attempt to include representation and card from the perspective of that peer. Must be provided with `peer_target`.FzOnly used if `search_query` is provided. Whether to limit the representation to the session (as opposed to everything known about the target peer))rrqdz}Only used if `search_query` is provided. The number of semantic-search-retrieved conclusions to include in the representation)gerrqgg?zmOnly used if `search_query` is provided. The maximum distance to search for semantically relevant conclusionszoOnly used if `search_query` is provided. Whether to include the most frequent conclusions in the representationzkOnly used if `search_query` is provided. The maximum number of conclusions to include in the representation) search_queryrF peer_targetpeer_perspectivelimit_to_sessionr,r-include_most_frequentmax_conclusionstokensrrrrrrc bK||ntjj} t}|r|st d|st |||| |d{V\}}t j|||}tt|d||t||du|du| | | | ||t|z dz|S|p|}|}d}|rtt5ttjj|d5t#j|d{V}dddn #1swxYwYdddn #1swxYwYt'||||||r|nd| | | | | d{V}t)|||| d{V}t+|||d{V\}}| t-t/|z t-|z }t1||||\}}}t3|||||d{V}t j||||| }ttdid |d dd|d|d|d|dt|d|dudt7|jd|dud|dud| d| d| d| d|d|d|dudt|z dz|S) a Produce a context object from the Session. The caller provides an optional token limit which the entire context must fit into. If not provided, the context will be exhaustive (within configured max tokens). To do this, we allocate 40% of the token limit to the summary, and 60% to recent messages -- as many as can fit. Note that the summary will usually take up less space than this. If the caller does not want a summary, we allocate all the tokens to recent messages. NzrQrQ_sKH$&/*P#nnO  ! J    ";  j+# #       )    +''!'!(mm#4/&2$&>)$7&; / /!1#/>>O#Ct"K      $.;HH%)I C Y   C C "$;A+ %    C C/4\BBBBBBBBI C C C C C C C C C C C C C C C C C C C C C C C C C C C C C C< #3=ZZ!/2(         N% L8h      D)A L*))######M< oc.&9&9:::_T=R=RR 3N|^_33/G  4 L*&7H% *==?? H    '< #) $ h  !  $V  h---  t++  $H$@AAA d** #/d":": & !4 3 #8"7 ,O ,O! ".-# $'7d&B&B% & ,~~?4GG' . Os6"D;=D$ D;$D( (D;+D( ,D;;D?D?z/{session_id}/summariescKtj|||d{V\}}d}|rtj|}d}|rtj|}tj|||S)z Get available summaries for a Session. Returns both short and long summaries if available, including metadata like the message ID they cover up to, creation timestamp, and token count. rWN)rrdre)rrXrYrSessionSummaries)r'rDrrdreshort_summary_schemalong_summary_schemas r>get_session_summariesrs&)3(E #)))######M< K);MJJI(:<HH  # *(   r@z/{session_id}/searchzMessage search parametersbodycpK|jpi}||d<||d<t|j||jd{VS)zh Search a Session with optional filters. Use `limit` to control the number of results returned. r'rD)rtlimitN)rtr queryr)r'rDrrts r>search_sessionrHsj$l bG*GN&GL j      r@)elogging contextlibrtimerfastapirrrrr r fastapi_paginationr !fastapi_pagination.ext.sqlalchemyr sqlalchemy.ext.asyncior srcrrrsrc.cache.clientrsrc.crud.sessionrsrc.dependenciesrsrc.deriver.enqueuersrc.embedding_clientrsrc.exceptionsrrr src.securityrrsrc.telemetry.eventsrrr src.utilsrsrc.utils.representationrsrc.utils.searchr src.utils.tokensr!src.utils.typesr" getLogger__name__rrouterrrhfloatrlistr?rCtupleSummaryrKrUr^rcrnpostSession SessionGetrvrrput SessionUpdaterdeleterrdictSessionPeerConfigrrrgetrrPeerrrr:rrQrrMessageSearchOptionsrrIr@r>rsCCCCCCCCCCCCCCCC######777777//////%%%%%%%%%%............000000111111 10000000LLLLLLLLLL 333333######,,,,,,222222  8 $ $  0    $%)11111*1  1  1*1*111Dj1E{T!11111h     #Y 6$$$$ $  $  7?T !4#8 89 $$$$@ 7?T !7?T#9 9: &EEEE E  E  '/ EEEE@' ?T)' /D(' '  '  7?T !3 +, ' ' ' ' T (',,nEEEFFG  S )- R***   $ &    (?  S %)T 6&&&$GLLNN33+1+1+1+1 "+1  +1 +1+1+1 +1\? NVVVWW S d3ii%)T 5&&& DDDD "D  DDDD& NVVVWWS d3ii#D#D#D#D #D#D#D#DL? NVVVWW  S d3ii"U > DDDDd D  DDDD0? NVVVWW S d3ii26$ _333DDDD W.. /D DDDD2? NVVVWW S d3ii26$ `333RRRR W.. /R RRRRD? NVVVWWS d3iit F DDDD 9D  DDDD:*, NVVVWW S d3ii499          * NVVVWW  S d3ii499(,S>V(W(W(W DDDDD  % D  DDDD4 % NVVVWW S d3ii D D D D  D D D D ) NVVVWW S d3ii ? 1VflfufLVVV v %u T   "E_ $e Z$)5 |$$$#Ui %u T     ).  D ))) #(%F####(% B ###_vvvvv v $J v*vv&t'v.Dj/v67v>*?vJKvV Wv^4Z_vvvvr+ NVVVWW S d3ii   D( NVVVWW S d3ii)- 4***  &r@