YjddlmZddlmZmZddlmZmZmZddl m Z erddl m Z GddeZ Gd d eZGd d eZd S)) annotations) TYPE_CHECKINGClassVar) BaseModel ConfigDictField)Message)PeerceZdZUdZeddZded<eddZded<edd Zded <edd Z ded <edd Z ded<dS)Summaryz1Represents a summary of a session's conversation..zThe summary text descriptionstrcontentz4The ID of the message that this summary covers up to message_idz#The type of summary (short or long) summary_typez:The timestamp of when the summary was created (ISO format) created_atz(The number of tokens in the summary textint token_countN) __name__ __module__ __qualname____doc__rr__annotations__rrrr]/home/ubuntu/.hermes/hermes-agent/venv/lib/python3.11/site-packages/honcho/session_context.pyr r s;;5*<===G====e OJc/TUUULUUUUe UJu CKrr ceZdZUdZeddZded<eddZd ed <edd Zd ed <dS) SessionSummariesz5Contains both short and long summaries for a session..zThe session IDrridNzThe short summary if availableSummary | None short_summaryzThe long summary if available long_summary) rrrrrr!rr#r$rrrr r s??eC%5666B6666$)E :%%%M$)5 9$$$Lrr c eZdZUdZedZded<eddZd ed <edd Z d ed <eddZ ded<eddZ ded<eddZ ded<d!dZ d!dZd"dZd#d ZdS)$SessionContexta Represents the context of a session containing a curated list of messages. The SessionContext provides methods to convert message history into formats compatible with different LLM providers while staying within token limits and providing optimal conversation context. Attributes: messages: List of Message objects representing the conversation context T)arbitrary_types_allowedzClassVar[ConfigDict] model_config.z)ID of the session this context belongs torr session_idz1List of Message objects to include in the contextz list[Message]messagesNz:Summary of the session history prior to the message cutoffr"summaryzLThe peer representation, if context is requested from a specific perspectivez str | Nonepeer_representationzBThe peer card, if context is requested from a specific perspectivezlist[str] | None peer_card assistant str | Peerreturnlist[dict[str, str]]czt|tr|n|jfd|jD}g}|jr#dd|jdd}|||jr#dd|jdd}|||jr(dd|jjd d}||||zS) a Convert the context to OpenAI-compatible message format. Transforms the message history and summary into the format expected by OpenAI's Chat Completions API, with proper role assignments based on the assistant's identity. Args: assistant: The assistant peer (Peer object or peer ID string) to use for determining message roles. Messages from this peer will be marked as "assistant", others as "user" Returns: A list of dictionaries in OpenAI format, where each dictionary contains "role" and "content" keys suitable for the OpenAI API cJg|]}|jkrdnd|j|jd S)r.user)rolenamerpeer_idr.0message assistant_ids r z,SessionContext.to_openai..`sO    (/,'F'F F"?     rsystemr5r isinstancerr!r*r,appendr-r+rselfr.r*system_messagespeer_representation_messagepeer_card_messagesummary_messager<s @r to_openaizSessionContext.to_openaiIs$,%/y#$>$>PyyIL      =    13  # @ c43Kccc++ '  " "#> ? ? ? > 6 EEEE!!   " "#4 5 5 5 < 4 Gt|';GGGO  " "? 3 3 3))rczt|tr|n|jfd|jD}g}|jr#dd|jdd}|||jr#dd|jdd}|||jr(dd|jjd d}||||zS) a Convert the context to Anthropic-compatible message format. Transforms the message history into the format expected by Anthropic's Claude API, with proper role assignments based on the assistant's identity. Args: assistant: The assistant peer (Peer object or peer ID string) to use for determining message roles. Messages from this peer will be marked as "assistant", others as "user" Returns: A list of dictionaries in Anthropic format, where each dictionary contains "role" and "content" keys suitable for the Anthropic API Note: Future versions may implement role alternation requirements for Anthropic's API compatibility cbg|]+}|jkr d|jdnd|jd|jd,S)r.rAr4z: r7r9s rr=z/SessionContext.to_anthropic..sr     ,..$"?    %oBBBB    rr4r?r@rArBrCrDrErFrIs @r to_anthropiczSessionContext.to_anthropics$2%/y#$>$>PyyIL       =    13  # @c43Kccc++ '  " "#> ? ? ? > 6EEEE!!   " "#4 5 5 5 < 4Gt|';GGGO  " "? 3 3 3))rrcBt|j|jrdndzS)z Return the number of messages in the context. Returns: The number of messages in this context r rlenr*r+rJs r__len__zSessionContext.__len__s$4=!!$,%=QQA>>rcJdt|jd|jrdnddS)z Return a string representation of the SessionContext. Returns: A string representation suitable for debugging zSessionContext(messages=z , summary=presentNone)rTrVs r__repr__zSessionContext.__repr__s6q#dm*<*<ppUYUaHm gmppppr)r.r/r0r1)r0r)r0r)rrrrrr(rrr)r*r+r,r-rOrRrWr\rrrr&r&)s  *4D)Q)Q)QLQQQQe DJ$e LH$e VG',e b'''#(% X###I 6*6*6*6*p=*=*=*=*~????qqqqqqrr&N) __future__rtypingrrpydanticrrrr;r peerr r r r&rrrras""""""********1111111111     i        y   gqgqgqgqgqYgqgqgqgqgqr