yjn8UddlZddlZddlZddlmZddlmZddlmZddlm Z m Z ddl m Z ddl mZddlmZdd lmZmZmZdd lmZdd lmZdd lmZdd lmZddlmZddlm Z m!Z!m"Z"m#Z#m$Z$ddl%m&Z&ddl'm(Z(m)Z)ddl*m+Z+ddl,m-Z-m.Z.m/Z/ej0e1Z2dZ3dej4dej4fdZ5de6e7e ffdZ8de6e7e ffdZ9de6e7e ffdZ:de6e7e ffdZ;de deZ?ej>e@e7e7e7fejAfeBd<ejAZCd e7d!e7d"e7dejAfd#ZDeGd$d%ZEeGd&d'ZF dd(e7d)ed3d4d5d?e;d7id4gd8d9d@d@dAd3d-d5dBdCdDid7id-gd8d9dEdEdFd3idGd9dHdHdId3dDdJdKdLdMdNdOdPdQgd8d9dRdRdSd3dTd5dCdDidUdVidTgd8d9dWdWdXd3dDdYdKdLdZd[dOd\dQgd8d9d]d]d^d3dDd_dKdLd`d[dOdLdadbdOdcddgd8d9dededfd3dDdgdKdDdhdKdLdidNdOdDdjdkgdldkdmdndGd9dododpd3dDdqdKdDdrdKdDdsdKdLdtd[dOdLdudbdOdvdQgd8d9dwdwdxd3dLdyd[dOdzd{d|dOd}dGd9d~d~dd3ddLdyd[dOidGd9dddd3ddDddgdddmidGd9dddd3idGd9dddd3dd5dCdDiddVidgd8d9dddd3ddDddKidgd8d9ddd3idGd9ddd3dDddKdDgddddmddgd8d9dZLe6e7e6e7e ffeBd<eLdHeLdWeLdReLd]eLdeeLdoeLdgZMeNe6e7e feBd<eLdHeLdWgZOeNe6e7e feBd<eLdeLdweLd~eLdHeLdeLd1eLdeLd@eLdWeLdReLdeLdg ZPeNe6e7e feBd<eLdweLdHeLdWeLd:eLdeLd@gZQeNe6e7e feBd<eLdweLdHeLdWeLd=eLd@gZReNe6e7e feBd< dd4eNej4d!e7d"e7de7dzd e7dTeNe<de7de7dzde7dzdeFfdZS dded e7de7dzd"e7dzde>)S[->->-@-@!A> B BBc dddddgddddddid d dddid d dddid d dgd dddgddddS)NstringzThe observation contenttype description)explicit deductive inductive contradictionzLevel: 'explicit' for direct facts, 'deductive' for logical necessities, 'inductive' for patterns, 'contradiction' for conflicting statementsr-enumr.arrayr-zDocument IDs of source or premise observations. Required and must be non-empty for deductive, inductive, and contradiction observations.r-itemsr.z7(For deductive) Human-readable premise text for displayzD(For inductive/contradiction) Human-readable source text for display preferencebehavior personalitytendency correlationz5(For inductive only) Type of pattern being identifiedhighmediumlowz[(For inductive only) Confidence level: 'high' for 5+ sources, 'medium' for 3-4, 'low' for 2)r#level source_idspremisessources pattern_type confidencerHr)r'_base_observation_propertiesrI,s4  +   h'"   h'T  h'a  S   ---2   c9 9 9r)c dtddgdddddiiidd gd d d id dd d d id ddddddddiiigdd d d iddd d d idddddddddiiiddgd d d iddd d d idddddgdS)Nobjectr#rBF propertiesconstr0rCrDr5r-r+)r-r7minItems)rCrD)requiredrL)ifthenr1)rCrErFrG)rCrEr2rE)r-rLrPadditionalPropertiesallOf)rIrHr)r' _generic_observation_item_schemarVis244( %$g/E%FG!-z :%,&,h%7()'' %,&,h%7()%% # #  &$g/E%FG!!!%,&,h%7()'' %,&,h%7()$$ # #  0$g/I%JK!-y 9%,&,h%7()'' %,&,h%7()$$ # #  U< B B Br)c>dddddddiddd dddidd d d gd d d S)NrKr+z6The deductive conclusion as a self-contained statementr,r5r-rNzJRequired non-empty list of source observation IDs supporting the deductionr-r7rOr.zERequired human-readable premise text matching the source observations)r#rCrDFr-rLrPrTrHrHr)r'"_deductive_observation_item_schemarZsx!W   (+k   (+f   $:99 %+  r)c Zdddddddiddd dddidd d dgd d d dgddd dgdddS)NrKr+zEThe inductive pattern or generalization as a self-contained statementr,r5r-rSzKRequired list of at least two source observation IDs supporting the patternrXzFRequired human-readable evidence text matching the source observationsr8zRequired pattern categoryr3r>z1Required confidence level based on evidence count)r#rCrErFrGFrYrHrHr)r'"_inductive_observation_item_schemar\s!f   (+l   (+g ! ;  !111R9! ! DWVV %K& & &r)valuedefaultc` t|S#tttf$r|cYSwxYw)zCoerce a tool input value to int, returning default on failure. LLMs sometimes pass non-numeric strings (e.g. 'Infinity') for integer parameters which would crash ``min()`` comparisons. )int TypeError ValueError OverflowError)r]r^s r' _safe_intrds> 5zz z= 1s --_observation_locksworkspace_nameobserverobservedcK|||f}t4d{Vt|}|tj}|t|<|cdddd{VS#1d{VswxYwYdS)u Get or create a lock for a specific workspace/observer/observed combination. This ensures that concurrent tool executors operating on the same observation space share a lock, preventing race conditions during document creation. The lock is stored as a weak reference — it stays alive as long as at least one ToolContext (via create_tool_executor) holds a strong reference. Once all executors for a key finish and are garbage collected, the entry is automatically removed from the registry. Args: workspace_name: Workspace identifier observer: The observing peer observed: The peer being observed Returns: An asyncio.Lock shared by all executors for this combination N)_registry_lockregetasyncioLock)rfrgrhkeylocks r'get_observation_lockrp s , 8X .C!%%c** <<>>D&* s # s;A## A-0A-c(eZdZUdZeed<eed<dS)ObservationFailurez9Records a single observation that failed during creation.content_previewerrorN)__name__ __module__ __qualname____doc__str__annotations__rHr)r'rrrr,s+CC JJJJJr)rrcJeZdZUdZeed<eeed<eeed<dS)ObservationsCreatedResultz+Result of a batch create_observations call. created_countcreated_levelsfailedN) rurvrwrxr`rzlistryrrrHr)r'r|r|4sF55I # $$$$$$r)r|output max_charsc|tjj}t|}||kr||dfS|d|d|dd|ddz}||dfS)aITruncate tool output to prevent token explosion. Returns (text, original_chars, was_truncated). Callers thread the truncation signal into `ToolResult.metadata` so `AgentToolCallCompletedEvent` can report `was_truncated` and `result_chars_before_truncation` instead of always emitting them as None/False. NFz [OUTPUT TRUNCATED - showing ,z of z characters]T)rLLMMAX_TOOL_OUTPUT_CHARSlen)rroriginal_chars truncateds r'_truncate_tool_outputr=syL6 [[N""~u,,z z \Y \ \ \n \ \ \ \ ] nd **r)zstr | ToolResultcXt|\}}}|s|St|d|dS)a~Run `_truncate_tool_output` and wrap in `ToolResult` only when the output was actually clamped, so the truncation signal reaches the `AgentToolCallCompletedEvent` emitter (which reads `was_truncated` / `result_chars_before_truncation` from `ToolResult.metadata`). Returns a bare `str` in the common no-truncation case to keep the handler contract unchanged. T) was_truncatedresult_chars_before_truncationr#metadata)rr)rr#rrs r'_maybe_truncated_resultrTsP.C6-J-J*G^]  !.<     r)r#cl|tjj}t||kr|S|d|dzS)zBTruncate individual message content (simple beginning truncation).N...)rrMAX_MESSAGE_CONTENT_CHARSr)r#rs r'_truncate_message_contentrhs<L:  7||y   :I:  &&r)patterncddl}|tjj}t ||kr|S|||||j}|s |d|dzS|}| }||z }||z }|dz} || z } td|| z } tt ||| z} | dkrtt ||} n3| t |kr tdt ||z } || | } | dkrdnd}| t |krdnd}|| |S)zExtract snippet around a regex pattern match. For grep/exact text search, finds the pattern and extracts context around it. rNrrS) rerrrrsearchescape IGNORECASEstartendmaxmin)r#rrrmatch match_start match_end match_len remainingbeforeafterrrsnippetprefixsuffixs r'_extract_pattern_snippetrqs~IIIL:  7||y   IIbii(('2= A AE +z z"U**++--K IK'II%I !^F  E ;' ( (E c'llI- . .C zz#g,, ** G  As7||i/00eCi GaiiUURFCLL((UUbF 'g 'v ' ''r)create_observationsa Create observations at any level: explicit (facts), deductive (logical necessities), inductive (patterns), or contradiction (conflicting statements). For deductive, inductive, and contradiction observations, missing or empty source_ids are invalid and will be rejected.rK observationsr5zList of observations to create)r-r.r7)r-rLrP)namer. input_schemacreate_observations_deductivezCreate new deductive observations discovered while answering the query. Every observation must include non-empty source_ids and premise text. Use this only for novel deductions grounded in existing observations.z,List of new deductive observations to createcreate_observations_inductivezCreate new inductive observations discovered while answering the query. Every observation must include source_ids, source text, pattern_type, and confidence. Use this only for patterns supported by multiple observations.z,List of new inductive observations to createupdate_peer_cardzUpdate the peer card with durable profile facts about the observed peer. Only include stable biographical facts, standing instructions, and long-lived preferences/traits. Do not include one-off conclusions, temporary events, or duplicate entries.znComplete deduplicated peer card list (max 40 entries). Each entry should be a concise standalone profile fact.r-r+get_recent_historyzPRetrieve recent conversation history to get more context about the conversation.)r-rL search_memoryzSearch for observations in memory using semantic similarity. Use this to find relevant information about the peer when you need to recall specific details.zSearch query textr,integerz=(Optional) number of results to return (default: 20, max: 40))r-r.r^)querytop_krget_observation_contextzRetrieve messages for given message IDs along with surrounding context. Takes message IDs (from an observation's message_ids field) and retrieves those messages plus the messages immediately before and after each one to provide conversation context. message_idszZList of message IDs to retrieve (get these from observation.message_ids in search results)r6search_messagesaSearch for messages using semantic similarity and retrieve conversation snippets. Returns matching messages with surrounding context (2 messages before and after). Nearby matches within the same session are merged into a single snippet to avoid repetition.z+Search query text to find relevant messageszDMaximum number of matching messages to return (default: 10, max: 20) )rlimit grep_messageszSearch for messages containing specific text (case-insensitive). Unlike semantic search, this finds EXACT text matches. Use for finding specific names, dates, phrases, or keywords mentioned in conversations. Returns messages with surrounding context.z5Text to search for (case-insensitive substring match)z1Maximum messages to return (default: 10, max: 30)zBNumber of messages before/after each match to include (default: 2)rS)textrcontext_windowrget_messages_by_date_rangezGet messages from a specific date range. Use this to find what was discussed during a particular time period, or to compare information before vs after a date. Essential for knowledge update questions.zNStart date (ISO format, e.g., '2024-01-15'). Returns messages after this date.z9End date (ISO format). Returns messages before this date.z1Maximum messages to return (default: 20, max: 50)ascdesczKSort order: 'asc' for oldest first, 'desc' for newest first (default: desc))r-r4r.r^) after_date before_daterordersearch_messages_temporalaYSemantic search for messages with optional date filtering. Combines the power of semantic search with time constraints. Use after_date to find recent mentions of a topic, or before_date to find what was said about something before a certain point. Best for knowledge update questions where you need to find the MOST RECENT discussion of a topic.zSemantic search queryzEOnly return messages after this date (ISO format, e.g., '2024-01-15')z2Only return messages before this date (ISO format)z1Maximum messages to return (default: 10, max: 20)z-Messages before/after each match (default: 2))rrrrrget_recent_observationszgGet the most recent observations about the peer. Useful for understanding what's been learned recently.z6Maximum number of observations to return (default: 10)booleanzKIf true, only return observations from the current session (default: false)F)r session_onlyget_most_derived_observationszGet observations that have been reinforced most frequently across conversations. These represent the most established facts about the peer.rget_session_summaryzhGet the session summary (short or long form). Useful for understanding the overall conversation context. summary_typeshortlongz,Type of summary to retrieve (default: short) get_peer_cardzgGet the peer card containing known biographical information about the peer (name, age, location, etc.).delete_observationszDelete observations by their IDs. Use the exact ID shown in [id:xxx] format from search results. Example: if observation shows '[id:abc123XYZ]', pass 'abc123XYZ' to delete it.observation_idszTList of observation IDs to delete (use the exact ID from [id:xxx] in search results)finish_consolidationzSignal that consolidation is complete. Call this when you have finished your consolidation work and are ready to stop. You MUST call this tool when done - do not keep exploring indefinitely.summaryzkBrief summary of what was accomplished (peer card updates, observations consolidated, observations deleted)extract_preferencesaExtract user preferences and standing instructions from conversation history. This tool performs both semantic and text searches for preferences, instructions, and communication style preferences, then returns them for adding to the peer card. Call this FIRST during consolidation.get_reasoning_chaina Get the reasoning chain for an observation - traverse the tree to find premises (for deductive) or sources (for inductive), and/or find conclusions derived from this observation. Use this to understand how an observation was derived or what conclusions depend on it.zAThe document ID of the observation to get the reasoning chain forrD conclusionsbothzu'premises' to get what this observation is based on, 'conclusions' to get what depends on it, 'both' for full contextr)observation_id directionr)rrTOOLSDIALECTIC_TOOLSDIALECTIC_TOOLS_MINIMAL DREAMER_TOOLSDEDUCTION_SPECIALIST_TOOLSINDUCTION_SPECIALIST_TOOLS session_namemessage_created_atrun_idparent_categoryc K|s,tdtdggSd|D} | s,tdtdggSt d4d{V} t j| |||d{Vdddd{Vn#1d{VswxYwYd | D} d} ttj j ||| 5tj | d{V} dddn #1swxYwYtttt!| | d } n2#t"$r%}td |Yd}~nd}~wwxYwg}g}t%| D]\}}| | |}n ttj j ||| 5tj|jd{V}dddn #1swxYwYnl#t"$r_}td|j||t/|jddd|Yd}~d}~wwxYwt1j|||jdvr|jnd|jdkr|jnd|jdvr|jnd|jdkr|jnd|jdkr |jpdnd}t1j|j||j|||jdvr|jnd}||g}|rt d4d{V} t j | ||||d d{V}dddd{Vn#1d{VswxYwYtdt!||||tt!|d|D|S)ap Create multiple observations (documents) in the memory system in a single call. Uses short-lived DB sessions to avoid holding connections during embedding API calls. Args: observations: List of validated observation inputs observer: The peer making the observation observed: The peer being observed session_name: Session identifier workspace_name: Workspace identifier message_ids: List of message IDs these observations are based on message_created_at: Timestamp of the message that triggered these observations run_id: Agent run id, threaded onto the embedding-call ContextVar so EmbeddingCallCompletedEvents emitted here can be joined back to the originating agent run. Returns: ObservationsCreatedResult with created count and any per-observation failures z*create_observations called with empty listr)r}r~rc^g|]*}|jt|+SrH)r#r&r(.0r s r' z'create_observations..NsD  ;    %c**r)z#No non-empty observations to createzcreate_observations.collectionNrgrhcg|] }|j SrH)r#rs r'rz'create_observations..as??? ???r)rfrrTstrictz]Batch embedding failed for create_observations; falling back to per-observation embedding: %sz6Error embedding observation content for level '%s': %s2zEmbedding failed: rsrt)r0r1r2r0)r1r2r1r@)rrrCrDrErFrG)r#rrBr embeddingrCzcreate_observations.save) documentsrfrgrh deduplicatez#Created %d observations in %s/%s/%scg|] }|j SrH)rB)rdocs r'rz'create_observations..s666c 666r))!loggerwarningr|inforr get_or_create_collectionrrCREATE_OBSERVATIONSr]rsimple_batch_embeddictzipranger Exception enumerateembedr#rBappendrrr DocumentMetadatarCrDrErFrGDocumentCreatecreate_documents)rrgrhrrfrrrrnormalized_observationsdbcontentsembeddings_by_index embeddingserrir rrraccepteds r'rr+s> XCDDD(qTVWWWW #X 9:::(qTVWWWW:;;       r+                                        @?'>???H9= # 4 :)+     M M 0B8LLLLLLLLJ  M M M M M M M M M M M M M M M# c12233Z M M M       k          /1I')F344663  *+A.II +(<B#1!$3 JJ '7&B;"E3D EDE!D">##''))HX&&'''  6> " " U6>0NB C C U6>+x7 8 8 Xfn/4466 7 7 U2YY zz$''''''''((,,..//HX&&''' r)levelsrc Kd}|rdd|ii}tjd|||||||d{V}tj|S)u Search for observations in memory using semantic similarity. Does not require a DB session — ``query_documents`` manages its own short-lived sessions so no connection is held during external calls. Args: workspace_name: Workspace identifier observer: The peer who made the observations observed: The peer who was observed query: Search query text limit: Maximum number of results levels: Optional list of observation levels to filter by (e.g., ["explicit"], ["deductive", "inductive", "contradiction"]) embedding: Optional pre-computed embedding to avoid redundant API calls Returns: Representation object containing relevant observations NrBin)rrfrgrhrrfiltersr)r query_documentsrfrom_documents) rfrgrhrrr rr#rs r'rrs:&*G ,T6N+* %         I  ( 3 33r)cK|sgSd}|r|sddlm}||||d{V}|sgSttjjtjj|ktjj |}|r)|tjj |k}n9|7|tjj |}| d}ttjtjj|kt|j j|j jtjjz ddtjj}|r)|tjj |k}n9|7|tjj |}||d{V} t%| } | S)a Retrieve messages for given message IDs along with surrounding context. Takes message IDs (from an observation's message_ids field) and retrieves those messages plus the messages immediately before and after each one to provide conversation context. Args: db: Database session workspace_name: Workspace identifier session_name: Session identifier (optional) message_ids: List of message IDs to retrieve observer: When provided and session_name is None, scope results to sessions this peer belongs to Returns: List of messages in chronological order, including the requested messages and surrounding context Nr)get_peer_session_names target_seqsrN)src.crud.messager'r r rseq_in_sessionrrf public_idin_rctecbetweenexistsrrrrrr) rrfrrrgallowed_session_namesr'rtarget_seqs_cterrs r'rr/sz2  /3 ;;;;;;&<&< ' ' ! ! ! ! ! ! % I v~,-- v~,> ? ? v~'++K88 9 9 Rzz&.5EFF  *zz&.599:OPPQQhh}--O v~ v~,> ? ?  ?$3 4 4 U#%4v~7TT'"a.. VXX   &./3355 6 6 Rzz&.5EFF  *zz&.599:OPPQQ::d## # # # # # #FFNN$$((**++H Or)c Kg}t}gd}d} tj|d{V}tt ||d}n2#t $r%} t d| Yd} ~ nd} ~ wwxYw|D]} tj ||| dd|| | nd|d{V} | D]\} } | D]w}|j |krj|j dd  }||vrE|||d |j d x#t $r&} t d | | Yd} ~ d} ~ wwxYwgg|dd d S)ah Extract user preferences and standing instructions from conversation history. Uses semantic search to find messages that might contain preferences or instructions. This is language-agnostic and doesn't rely on keyword matching. Args: workspace_name: Workspace identifier session_name: Session identifier (optional) observed: The peer whose preferences to extract observer: When provided and session_name is None, scope results to sessions this peer belongs to Returns: Dict with 'messages' list containing potentially relevant messages )z(user preferences and communication stylez)standing instructions and rules to followz"how user wants responses formattedz!user requirements and constraintsz"things user wants or does not wantNTrzjBatch embedding failed for extract_preferences; falling back to per-query embedding in search_messages: %srrrfrrrrrrgd'z%Error in semantic search for '%s': %s) instructions preferencesr)setrrrrrrrr rrkrr#loweraddrr&)rfrrhrgr seen_contentsemantic_queriesquery_embeddings_by_queryquery_embeddingsr rsnippetsmatches_msg content_keys r'rrso,H UUL@D  !1!DEU!V!VVVVVVV$(  "24 @ @ @% % !!     x          "NN N!1-) 1<.11%888!         H' H H "HHC}00&)k$3$&7&=&=&?&? &l::(,,[999$OO,F 0A0A0C0C,F,F,FGGG H H N N N NNBE1 M M M M M M M M NSbSM  s09A BA>>B B:E E6E11E6ceZdZUdZeed<eed<eed<edzed<eejdzed<e ed<e ed <e j ed <dZ edzed <dZedzed <dZedzed <dZedzed<dS) ToolContextz'Context object passed to tool handlers.rfrgrhNrcurrent_messagesinclude_observation_idshistory_token_limitdb_lock configurationr agent_typer)rurvrwrxryrzrr rboolr`rlrmrMrrrNrrHr)r'rHrHs11MMMMMM*6>*T1111!!!!\26M(4/666FC$J!Jd !!!"&OS4Z&&&&&r)rH forced_levelctx tool_inputrQcK|dg}|sdS|jrdnd}|D] }|||d< |d|!g}g}|D]} tj|}ng#t $rZ} |tt|dddd d |  Yd} ~ d} ~ wwxYw|jrF|j dkr;|t|j dd d |j d  |||s$d d|D} d| S|jr1d|jD} t|jdj } ng} t} |j4d{Vt!||j|j|j|j| | |j|j d{V} dddd{Vn#1d{VswxYwY|| jz}| j}|d}|d}|d}|d}|jrb|jr[|jrTt7t9|jt;|j|j|j|j|j| j| d| jd|jd|jd|d|d|d|d}|r7d d|D} |d t?|d!| z }d"d#l m!}||| j|d$%S)&z Handle create_observations tool.rz!ERROR: observations list is emptyr/r0NrBr#rrzValidation failed: rz6Deriver can only create 'explicit' observations, got 'r7z; c3:K|]}d|jd|jVdSr7z': Nrrfs r' z3_handle_create_observations_impl..I$ $ 45 /! / /ag / /$ $ $ $ $ $ r)z+ERROR: All observations failed validation: cg|] }|j SrH)idrrEs r'rz4_handle_create_observations_impl..s>>>#sv>>>r)r)) rrgrhrrfrrrrr1r2 r iterationrrNrfrgrhconclusion_countr zCreated z observations for  by  (z explicit, z deductive, z inductive, z contradiction)c3:K|]}d|jd|jVdSrVrrWs r'rYz3_handle_create_observations_impl..QrZr)z Failed : rr)r}r r)"rkrI setdefaultr ObservationInputmodel_validaterrrrryrBr#joinrrrLrrgrhrrfrrrr~countrNrrrr}rsrc.utils.typesr)rRrSrQraw_observations default_levelr rvalidation_failures validatedr failure_detailsrrr all_failuresr explicit_countdeductive_countinductive_countcontradiction_countresponsers r' _handle_create_observations_implrwsI"~~nb99 322#&"6GJJKM33  #'CLL NN7M 2 2 2 246L46'' 0??DDII     & &"$' 2(>(>$?$?$D333    HHHH    IOz$A$A  & &"$-$5crc$:eS\Sbeee    I&&&& O))$ $ 9L$ $ $   O_NNN +>>)=>>>  !5b!9!DEE (]]{         *%\\)-#1:/                                      '6L "F\\*--Nll;//Oll;//O ,,77 z  cn  )<   ,z/11 # 3>"1!'!5     M6' M M3< M MS\ M M  M M'6 M M  M M(; M M M G))$ $ 9E$ $ $    FL 1 1FF_FFF +***** :#)#76JJ   s+A44 C>ACC9=H HHc2Kt||d{VSNrwrRrSs r'_handle_create_observationsr|bs*2#zBB B B B B B BBr)c6Kt||dd{VS)Nr0rPrzr{s r'%_handle_create_observations_deductiver~hF2        r)c6Kt||dd{VS)Nr1rPrzr{s r'%_handle_create_observations_inductiverrrr)cK|j4|jjjs#td|j dS|d}|"td|jdSg}t}t|tr ttt|nt|g}|D]}t| }|s&d|}||vrd|||||s"td|jd St'|t(krCtd t(t'|t(|dt(}|j4d{Vt-d 4d{V} t/j| |j||j|j d{Vdddd{Vn#1d{VswxYwYdddd{Vn#1d{VswxYwYtd |jd|jd|j|jri|jrb|jr[t=t?|jtA|j|j|j|j|jt'|ddl!m"} | d |jd|jdt'|dS)zHandle update_peer_card tool.Nz3Peer card creation disabled for %s, skipping updatezHPeer card creation is disabled for this workspace/session configuration.r#zGPeer card update called with None content for %s, keeping existing cardz1Peer card content was empty, no update performed. zBPeer card update normalized to empty for %s, keeping existing cardzEPeer card content was empty after normalization, no update performed.zBPeer card update exceeded max facts (%s), truncating from %s to %sztool.update_peer_card)rf peer_cardrgrhzUpdated peer card for /)rr_rrNrfrgrh facts_countrreraT)peer_card_updatedrr)#rMrcreaterrrfrkrr; isinstancerrryr&rir<splitr=rrMAX_PEER_CARD_FACTSrLrr set_peer_cardrgrhrrNrrrrrkr) rRrSraw_peer_card_contentnormalized_peer_cardseenr7itemlinenormalized_keyrrs r'_handle_update_peer_cardr|s  $S->-H-O$ A     W  'NN955$ U     CB')UUD +T 2 2 *T#Y-...'(( )  * *4yy    $**,,"4"4"6"677 T ! !     ##D)))) W P     WV   #666 P  $ % %     44H5H4HI{        J'>??       2 -*\\                                                                  KKS!3SSclSSS\SS  z  cn  )<   )z/11 # 3>"1 455    +***** :III3<II'+C@T.s3 W W W! E E8CC E E W W Wr)z from session zfrom z across sessionszConversation history (z messages ): ) rrrfrrhrKrirr)rRrSrDrhistory history_textscopers r'_handle_get_recent_historyrs A344        .@ -)\/ / / / ) ) ) ) ) )  76               yy W Ww W W W                                 4*(*** 3S\ 3 3 3 Xc'll W We W W W WF "6 * **s2BB BBc Kddlm}tt|ddd}|d} t t jj|j |j |j 5tj |d{V}dddn #1swxYwYn'#t$rd tjjd zcYSwxYw|d d t%|d }t'jd|j |j|j|||d{V}t/j|}|} | dkri|ddi} |jdkrtt|ddd} d} t'j|j |j|| d||jd{V} | rt;| d|d} | r%i| dt3| i}|d| |S|d|d| S|d|d| S|jr|ntA|}| |d<|d| d|d||S)zHandle search_memory tool.rrerrrrrNz,ERROR: Query exceeds maximum token limit of z. Please use a shorter query.TrN)rused_embeddingembedding_query_count query_tokens)rrfrgrhrrr results_count dialecticr5 for query 'r7z.No observations yet. Message search results: rz!No observations found for query 'zb', and no messages found in history. Try a different phrasing or use grep_messages for exact text.Found z observations for query 'z': )!rkrrrdrkrr SEARCH_MEMORYr]rfrrrrrbr EMBEDDINGMAX_INPUT_TOKENS_estimate_tokens_safer r$rgrhrr%rrNrr_format_message_snippetsrJ str_with_idsry)rRrSrrrquery_embedding search_metarmem total_count zero_hit_metarmessage_outputrB fallback_metamem_strs r'_handle_search_memoryrs+*****  *..11266 ; ;E w E  # . 4-:/     B B %5$:5$A$AAAAAAAO  B B B B B B B B B B B B B B B    :#4SSS T    !"-e44 ##K* )!I  ' 2 2C''))Ka<;;;; >[ ( ( *.."9"92>>CCE!N!1"1 - )H !94E444""  Q= Q/3x== Q Q !z_~__*:]]]]'  z@@@@"    %($?Mc   SXXG#.K :UUUuUUGUU   s61B'4B B'BB'"B#B''!C  C cKtd4d{V}t||j|j|d|jd{V}|sd|dcdddd{VSdd|D}dddd{Vn#1d{VswxYwYdt |d |}t|S) z$Handle get_observation_context tool.ztool.get_observation_contextNr)rfrrrgzNo messages found for IDs rchg|]/}tt|j|j|j0SrHrrr#rrrs r'rz3_handle_get_observation_context..csK     /-ai88LK   r)z Retrieved z messages with context: )rrrfrrgrirr)rRrSrr messages_textrs r'_handle_get_observation_contextrTs899       R0 -)"=1\           LK =0IKK                  "                                  (R#h-- Q Q- Q QF "6 * **s> !1 6u = ======= >>>>>>>>>>>>>>>))%!H!"-e44X ##K  z > >>s3BB!Bc K|dd}|sdStt|ddd}tt|ddd}tj|j|j||||j d {V}|sd |d Sg}td |D}t|dD]\}\} } g} | D]E} t| j |} | t| | j| jF| r | djnd}| d|d|dt!| dd| zd|d|dt!|dd|z}t%|S)zHandle grep_messages tool.rrz#ERROR: 'text' parameter is requiredrrr8rrS)rfrrrrrgNzNo messages found containing 'r7c3:K|]\}}t|VdSryrrrCrDs r'rYz(_handle_grep_messages..,@@!G @@@@@@r)rNrunknown --- Snippet (session: ,  match(es)) --- rrz messages containing 'z' in z conversation snippets:  )rkrrdr rrfrrgsumrrr#rrrrrrir)rRrSrrrrB snippet_texts total_matchesr rCcontextlinesrErsessrs r'_handle_grep_messagesrs; >>&" % %D 544  *..11266 ; ;E*..!122A66N')% % H 877777 "M@@x@@@@@M!*8Q!7!7    GW  C0dCCI LL.y#.#-XX    +2@wqz&&y P1 P P P PW P P Pii     lkkdkkXkkk ++m $ $ %  #6 * **r)date_str param_namec|sdS tj|ddS#t$r d|d|dcYSwxYw)z?Parse a date string, returning datetime, None, or error string.NZz+00:00zERROR: Invalid z format 'z&'. Use ISO format (e.g., '2024-01-15'))r fromisoformatreplacerb)rrs r' _parse_daterst tg%h&6&6sH&E&EFFF gggfffhffffffgs'.AAc K|d}|d}tt|ddd}|dd}t|d}t |t r|St|d}t |t r|St d4d{V}tj||j |j |||||j d{V} t| } | rd d | Dnd } dddd{Vn#1d{VswxYwYg} |r| d ||r| d|| s| rd | nd} d| S| rd | nd} |dkrdnd}d| d| d|d| }t|S)z'Handle get_messages_by_date_range tool.rrrrrrztool.get_messages_by_date_rangeN)rfrrrrrrgrchg|]/}tt|j|j|j0SrHrrs r'rz6_handle_get_messages_by_date_range..sI31!)<3344475O55666 11;RW\\*---AR 0J000-7Gj)))ZJ#(E>>~J WVVzVVZVV}VV  #6 * **sA$E  EEc K|dd}|sdS|d}|d}tt|ddd}tt|dd d }t|d}t |t r|St|d}t |t r|St tjj |j |j |j 5tj|d {V} d d d n #1swxYwYtj|j |j|||||| |j d {V} g} |r| d ||r| d|| rdd| dnd} |ddt+|t-| d} | st/d|d| | St1| d|d| }t/|| S)z%Handle search_messages_temporal tool.rrz$ERROR: 'query' parameter is requiredrrrrrrSrN) rfrrrrrrrrgrrrbr)TrNrrr7rr)rkrrdrrryrrrr]rfrrrrr rrrgrrirrrr)rRrSrrrrrrrrrB date_filter filter_descrrs r' _handle_search_messages_temporalrs NN7B ' 'E 655^^L11N nn]33O  *..11266 ; ;E:>>2B#C#CQGGKKN^\::J*c""o}==K+s## ,2)z+    >> !1 6u = ======= >>>>>>>>>>>>>>>2)%%!         H K64N4455586_667777BJ3w||K003333K !"-e44X ##K  IEIIKII     )3V3V3V3V3VWWI i+ > > >>s(EEEcHK|dd}td4d{V}tj||j|j|jtt|ddd|r|j ndd{V}tj |}dddd{Vn#1d{VswxYwY| }|d krd S|rd nd }|j r|nt|}d |d|d|S)z$Handle get_recent_observations tool.rFztool.get_recent_observationsNrrr6)rrfrgrhrrrzNo recent observations foundz this sessionz all sessionsrz recent observations from : )rkrr query_documents_recentrfrgrhrrdrrr%rrJrry) rRrSrrrrepresentationrrrepr_strs r'_handle_get_recent_observationsrTs>>.%88L899 B B B B B B BR5-\\i w 7 7<NNE  & !##%%%   RK Q Q5 Q Qx Q QQsA??BBBBBBB2;-\\i w 7 7<>.'::L 6 ! ! ##  # ) 455       ". "C$4b                                    211 Pw~6 P PGI.s9:: T ::::::r))rr rrfrgrhri)rRrSrDrrs r'_handle_get_peer_cardrsnA.//       2, -\\                                      <;S\;;; -CL - - - :: ):::11 s.A A$'A$cK|dg}|sdS|j4d{Vtd4d{V}tj||j||j|jd{V}dddd{Vn#1d{VswxYwYdddd{Vn#1d{VswxYwYd|D}|D]!}||vrt d|"t|}|dkrn|j rg|j r`|j rYtt|j t!|j |j |j|j|j|d |D dd lm}|d |d |d|DdS)z Handle delete_observations tool.rz$ERROR: observation_ids list is emptyNztool.delete_observations)rf document_idsrgrhch|]\}}|SrHrH)rdoc_idrDs r' z._handle_delete_observations..s333ifa6333r)zLFailed to delete observation %s (not found, already deleted, or wrong scope)rcg|]\}}|SrHrHrrDrBs r'rz/_handle_delete_observations..s666(!U666r)r^rezDeleted z observationscg|]\}}|SrHrHrs r'rz/_handle_delete_observations..s555Eu555r)) deleted_countr r)rkrLrr delete_documentsrfrgrhrrrrrNrrrrrkr) rRrSrrdeleted deleted_idsobs_idrrs r'_handle_delete_observationsr s!nn%6;;O 655{        J'ABB       b- -(\\                                                                437333K!  $ $ NN^    LLMqSZCNs?R ,z/11 # 3>"1!.66g666    +***** :7=777*55W555     s5B$/B/B$ B B$B B$$ B.1B.c@K|}|dd}d|S)z!Handle finish_consolidation tool.rzConsolidation completezCONSOLIDATION_COMPLETE: )rk)rRrSrDrs r'_handle_finish_consolidationrs/ AnnY(@AAG /g / //r)cK|}ttjj|j|j|j5t|j|j|j |j d{V}dddn #1swxYwY| dg}|sdSdt|dd d |Dd g}d |S) z Handle extract_preferences tool.r)rfrrhrgNrzYNo potentially relevant preference or instruction messages found in conversation history.z!**Potentially Relevant Messages (z):**rc3 K|] }d|V dSrrHr]s r'rYz._handle_extract_preferences..s(11*s**111111r)a **Action Required:** Review these messages and extract any preferences or standing instructions to add to the peer card using `update_peer_card`. Summarize as clear rules (e.g., 'INSTRUCTION: Always include cultural context') or preferences (e.g., 'PREFERENCE: Brief responses').r)rrPREFERENCE_EXTRACTIONr]rfrrrrrhrgrkrri)rRrSrDresultsr output_partss r'_handle_extract_preferencesrsW A 28)z+       ,-)\\                          {{:r**H kjj @CMM??? 1111111 RL ;;| $ $$s.A00A47A4rBc *g}td|D}t|dD]\}\}}g}|D]D}t|j} |t | |j|jE|r |djnd} |d|d| dt|dd |zd |d t|d |d d |z} t| dS)aFormat message snippets for output. Returns bare `str` because callers concatenate it into other strings or place it into `ToolResult.content`. Callers that need the truncation telemetry signal route their own output through `_maybe_truncated_result` themselves. c3:K|]\}}t|VdSryrrs r'rYz+_format_message_snippets..rr)rNrrrrrrrrz matching messages in z conversation snippets rr) rrrr#rrrrrrrir) rBrrrr rCrrrErrrs r'rrs\ "M@@x@@@@@M!*8Q!7!7    GW  C1#+>>I LL.y#.#-XX    +2@wqz&&y P1 P P P PW P P Pii     hggc(mmgg\`ggg ++m $ $ %  ! ( ( ++r)c VK|d}|sdS|dd}|dvrd|dStd4d {V}tj||j|gd {V}|r|d sd |d cd d d d {VS|d }g}|jpd }|d|jd|d|j|dvr|dkr|j rtj||j|j d {V} | rvg} | D]3} | jpd } | d| jd| d| j4|dt| dd | zn4|dt|j dn|dkr|j rtj||j|j d {V} | rug}| D]3}|jpd }|d|jd|d|j4|dt| dd |zn]|dt|j dn1|d kr|dn|d|dvrtj ||j||j |j d {V}|rug}|D]3}|jpd }|d|jd|d|j4|d!t|dd |zn|d"d d d d {Vn#1d {VswxYwYd |S)#z Handle get_reasoning_chain tool.rz#ERROR: 'observation_id' is requiredrrrzERROR: Invalid direction 'z/'. Must be 'premises', 'conclusions', or 'both'ztool.get_reasoning_chainNrzERROR: Observation 'z ' not foundr/z**Observation [id:z] (z):** )rDrr0z - [id:z): z **Premises (rz **Premises:** Referenced z' premise IDs but none found in databaser1z **Sources (z **Sources:** Referenced z& source IDs but none found in databasezC **Premises/Sources:** N/A (explicit observations have no premises)z$ **Premises/Sources:** None recorded)rrrz **Derived Conclusions (z$ **Derived Conclusions:** None found)rkrr get_documents_by_idsrfrBrr\r#rCrriget_child_observationsrgrh)rRrSrrrdocsrrrBrD premise_linespp_levelrE source_linesss_levelchildren child_linesr/c_levels r'_handle_get_reasoning_chainr$1s* ^^$455N 544{F33I;;;fIffff455JMJMJMJMJMJMJM.r33EGWXXXXXXXX F47 FE.EEEJMJMJMJMJMJMJMJMJMJMJMJMJMJM Q"$  'ZVVVEVVVVWWW , , , ###!%!:*CN"" /1M%YY"#'"7Z%,,-Wqt-W-W-W-WAI-W-WXXXX ''>X>>>))M223 !''rc#.6I6Irrr+%%#.% $ 9*CN!! .0L$XX"#'"7Z$++,Vad,V,Vw,V,V19,V,VWWWW ''<G <<.execute_tools                  ,i @Q@Q9R9RSSS!!## #%L $((33G +'.wsJ'?'?!?!?!?!?!?!?nj990!/!7J-6HH!/J  > OO8==??++ :i99 z***%    7666JH  ' ' 'KKKKKJH NN: & & & & & & & & ' ' 'LLLLLJH NN: & & & & & & & & 4 4 4Q PPa9IPPQPP H LLdL 3 3 3 3 3 3 3 3 4 ,,..6$>K # "8 , , , +#'%!!//117799&G&G&I&I      ,,..6$>K # "8 , , , +#'%!!//117799&G&G&I&I     sV,B:D'&I'!G9$E1,I1 G9>$F'"I' G94;G4/I4G99IAJ))rprHryrr)rfrgrhrrIrJrKrMrrNr shared_lockr<rRs @r'create_tool_executorr>sP-^XxPPPPPPPPK %!) 7/#'   CpcptCH~p#ppppppd r)r'r.r/rr0r_r1r2c |jr|jr|jsdS ddlm} m} | | did|jd|d|d|d|jd |jd |jd |d |d |dt|d|ddt|dt|ddd|dd|dd|dd|ddt|dpddS#t$r tddYdSwxYw)uBuild and emit AgentToolCallCompletedEvent. Best-effort; swallows errors. Skipped when the executor was constructed without agent identifiers (run_id / agent_type / parent_category) — telemetry attribution requires all three. Nr)AgentToolCallCompletedEventrrr_r1r2rrNrfr'r.r0 result_charsrresult_tokens_estimaterFrrrrrz*Failed to emit AgentToolCallCompletedEventTr,rH)rrNrsrc.telemetry.eventsr@rrfrrk_estimate_tokensrOr`rrdebug) rRr'r.r/rr0r_r1r2r@rs r'r9r9S s0$ J3>c.ARJJJJJJJJ  ' '   zz #) ,m '<&;   !$ 3 3   >>   #11 $) (K " !___ 08||4000 (8 'C'C'C  #8<<#G#GHHH! "&\\.999# $ll7+++% &'ll?;;;' ( (||,<===) *'*(,,7N*O*O*TST&U&U&U+      2 RRR AD QQQQQQRsD D;;&E%$E%c|sdS ddl}|d}t||S#t$r#t dt|dzcYSwxYw)z?Tiktoken-based size proxy for tool result strings. Best-effort.rN cl100k_baserN)tiktoken get_encodingrencoderr)rrIencodings r'rDrD s q &((778??4(())) &&&1c$ii1n%%%%%&s:A*A.-A.c(|sdSt|S)zWrapper around `_estimate_tokens` that returns None on falsy input. Used by search-handler metadata where we want `query_tokens=None` when the query is empty rather than 0 (which could be confused with a real measurement). N)rD)rs r'rr s t D ! !!r)ry)NN)Nr)NNFrNNNN)vrlloggingweakrefcollections.abcr dataclassesrrtypingrrpydanticr sqlalchemyr sqlalchemy.ext.asyncior srcr r r src.configrsrc.dependenciesrsrc.embedding_clientr src.modelsr src.schemasrrCrrrrr src.utilsrsrc.utils.formattingrrsrc.utils.representationrrkrrr getLoggerrurrrgr(rryrIrVrZr\r`rdWeakValueDictionaryretuplermrzrjrprrr|rOrrrrrrrrrrrrrrfloatrrrrHrwr|r~rrrrrrrrrrrrrrr rrrr$r%r>r9rDrrHr)r'rcs$$$$$$!!!!!!$$$$$$//////%%%%%%%%%%''''''111111------! LLLLLLLL333333UUUUUUUUUU  8 $ $C  !C CCCC:d38n::::zC$sCx.CCCCLDcN4'DcN''''T S 3 3    * G!!G/c3m0Dgl0RS#&25 \>   %%%%%%% %*.++ +$J+ 3T>++++.C,>(''s'sTz'S''''9=*(*( *(*(+.:*(*(*(*(*(ZF$%g##C==??!!((   F$ $/m##Q??AA!!((   &&!F$>$/v##Q??AA!!((   &&?F$\" \ #T%h/  #    ]F$H$i  IF$Xu%#6 &#b!  !   YF$~)S#$h/#  '     F$\!Z%#P &#i!  !   ]F$BT%#Z &#V! &#g ##  %  CF$r!,c%#s %#^   &#V! %"FO#p%   ##sF$j*s%#: %#j %#W   &#V! &#R ##%0! 5  !!kF$j)A&#[! &#p$!!      kF$P$/e%#[!    & &QF$l%B$$f-#Q& !!   mF$JA  KF$Z%I!#$h/#y$$++   [F$x&X$$Q #      yF$V &s  &d%#f## %???$[%   **  e F$F$F$tCc3h  FFFT  /  #$ / &' $%   )d38n%    / 1d38n-   #$ )* / /       #$   !!' tDcN#2 #$ /  )*    4Dc3h0   " #$ /  )* 4Dc3h0&"&\\w/0\\\* \  \ c \\ $J\4Z\\\\\F  3333*3Dj 3  3  &. 3333x $$(,4,4,4,4,4  ,4  ,4 I  ,4E{T!,4,4,4,4,4h NNNN*Nc N Dj N  &. NNNNj MMM*MMDj M  #tCy. MMMM` ''''''' '4 $ xxx xS#Xx* x  xxxxvC C"&sCx.CCCCC  "&sCx. "&sCx.` `"&sCx.`````F+ +"&sCx.+++++6W W"&sCx.WWWWWt+ +"&sCx.+++++8'? '?"&sCx.'?'?'?'?'?T++ ++"&sCx.++++++++++\g#*g#g(T/C:Ogggg:+ :+"&sCx.:+:+:+:+:+zB? B?"&sCx.B?B?B?B?B?JR R"&sCx.RRRRR6b b"&sCx.bbbbb0Q Q"&sCx.QQQQQ*[d38nQT"2 2"&sCx.22222j0 0"&sCx.00000 %  %"&sCx. % % % % %F ,5fn-tFN/CCDE ,MP , , , , ,FY# Y#"&sCx.Y#Y#Y#Y#Y#zK6K#%JK$%JK0 K 4 K * K>K.K*K!"DK @K>K$%JK6K*K 6!K"8#K$76'KKKS(Kc3h#@##EFFG6 $48$)#26!"&kkkkk* k 6>*T1 k " kk)4/k $Jkd k4ZksDcN#S()kkkk\1R 1R1R 1R  1R 38n 1R1R1R1R:1R 1R1R1R1Rh &3 &3 & & & & "d "sTz " " " " " "r)