Yj|XNdZddlmZddlZddlZddlmZddlmZm Z ddl Z ddl m Z m Z mZmZmZddlmZdd lmZmZmZmZmZmZmZmZdd lmZmZdd lm Z m!Z!m"Z"dd l#m$Z$dd l%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ej/e0Z1dddZ2Gdde e&Z3dS)zSync Honcho client.) annotationsN)Mapping)AnyLiteral) BaseModel ConfigDictField PrivateAttr validate_call HonchoAio)MessageResponse PeerConfig PeerResponseQueueStatusResponseSessionConfigurationSessionResponseWorkspaceConfigurationWorkspaceResponse)PeerBase SessionBase)AsyncHonchoHTTPClientHonchoHTTPClientroutes)Message)MetadataConfigMixin)SyncPagePeerSession) resolve_idzhttp://localhost:8000zhttps://api.honcho.dev)local productionc eZdZUdZedZedddZded <e d Z d ed <e d Z ded<e Z ded<e d Z ded<e Zded<e Zded<e d Zded<ed{dZed|dZdZd}dZd}dZd~d Zdd#Zdd%Zeedd&'fdd*Zed}d+Zedd-Zedd/Zeed012d d ed d3'ed dded d4d56ed d4d78d d ed d9'f dfdI ZddJZ ddKZ!eedddLfed dM'ed dN'dOddTZ" dddWZ#eedddXfed dY'ed dZ'dOdd]Z$ ddd_Z% dddaZ&eedddbfddcZ'eedddded de'edfddgdhifddnZ(eed012 ddduZ)eed012 dddxZ*d}dyZ+d}dzZ,xZ-S)Honchoa Main client for the Honcho SDK. Provides access to peers, sessions, and workspace operations with configuration from environment variables or explicit parameters. This is the primary entry point for interacting with the Honcho conversational memory platform. Attributes: workspace_id: Workspace ID for scoping operations metadata: Cached metadata for this workspace. May be stale if not recently fetched. Call get_metadata() for fresh data. configuration: Cached configuration for this workspace. May be stale if not recently fetched. Call get_configuration() for fresh data. allow)extra.r z#Workspace ID for scoping operations) min_length descriptionstr workspace_idN)defaultdict[str, object] | None _metadataWorkspaceConfiguration | None_configurationr_httpzAsyncHonchoHTTPClient | None _async_httpdict[str, Any] _http_config _base_urlFbool_workspace_ensuredreturnc|jS)zTCached metadata for this workspace. May be stale. Use get_metadata() for fresh data.)r0selfs T/home/ubuntu/.hermes/hermes-agent/venv/lib/python3.11/site-packages/honcho/client.pymetadatazHoncho.metadataI ~c|jS)z^Cached configuration for this workspace. May be stale. Use get_configuration() for fresh data.)r2r<s r> configurationzHoncho.configurationNs ""rAc|jSN)r3r<s r>_get_http_clientzHoncho._get_http_clientTs zrAc(tjSrE)r workspacesr<s r>_get_fetch_routezHoncho._get_fetch_routeWs """rAc4tj|jSrE)r workspacer-r<s r>_get_update_routezHoncho._get_update_routeZs 1222rAcd|jiS)Nidr-r<s r>_get_fetch_bodyzHoncho._get_fetch_body]sd'((rAdata+tuple[dict[str, object], dict[str, object]]crtj|}|jpi|jdfS)NT exclude_none)rmodel_validater?rC model_dumpr=rQrKs r>_parse_responsezHoncho._parse_response`sG&4T:: !'R)@)K)K*L* *   rArc|||}t j|}|jpi|_|j|_ |j S)z Get configuration from the server and update the cache. Returns: A WorkspaceConfiguration object containing the configuration settings. body) rFpostrIrPrrVr?r0rCr2rXs r>get_configurationzHoncho.get_configurationis|$$&&++  ! ! # #$*>*>*@*@,  &4T:: "+1r'5""rAzConfiguration to set)r+rCNonec||d|di||_dS)z Set configuration on the server and update the cache. Args: configuration: A WorkspaceConfiguration object with configuration settings. rCTrTr[N)rFputrLrWr2)r=rCs r>set_configurationzHoncho.set_configurationxse ##  " " $ $!=#;#;#;#N#NO $   ,rAc|jS)zThe base URL of the Honcho API.)r7r<s r>base_urlzHoncho.base_urlr@rArcJ|jtdi|j|_|jS)z/Lazily create and return the async HTTP client.N)r4rr6r<s r>_async_http_clientzHoncho._async_http_clients/   #4IIt7HIID rArc t|S)a Access async versions of all Honcho methods. Returns an HonchoAio view that provides async versions of all methods while sharing state with this Honcho instance. Example: ```python honcho = Honcho(workspace_id="my-workspace") # Async operations peer = await honcho.aio.peer("user-123") async for p in honcho.aio.peers(): print(p.id) ``` r r<s r>aioz Honcho.aios$rAT)arbitrary_types_allowed)configzBase URL for the Honcho APIrzTimeout in seconds)gtr+zMaximum number of retries)ger+zCustom HTTP clientapi_key str | None environment%Literal['local', 'production'] | Nonerdtimeout float | None max_retries int | Nonedefault_headersMapping[str, str] | None default_queryMapping[str, object] | None http_clienthttpx.Client | Nonec |ptjdd} t| |ptjd} |r|} n0|rt|} n tjdtd} | |_| | d} ||| d <||| d <|t || d <|t || d <t | |_| | | d <tdi| |_ dS)a Initialize the Honcho client. Args: api_key: API key for authentication. If not provided, will attempt to read from HONCHO_API_KEY environment variable environment: Environment to use (local or production) base_url: Base URL for the Honcho API. If not provided, will attempt to read from HONCHO_URL environment variable or default to the production API URL workspace_id: Workspace ID to use for operations. If not provided, will attempt to read from HONCHO_WORKSPACE_ID environment variable or default to "default" timeout: Optional custom timeout for the HTTP client. max_retries: Optional custom maximum number of retries for the HTTP client. default_headers: Optional custom default headers for the HTTP client. default_query: Optional custom default query parameters for the HTTP client. http_client: Optional custom httpx client. HONCHO_WORKSPACE_IDr.rOHONCHO_API_KEY HONCHO_URLr%)rdrnNrrrtrvrxrzrf) osgetenvsuper__init__ ENVIRONMENTSr7dictr6rr3)r=rnrprdr-rrrtrvrxrzresolved_workspace_idresolved_api_keyresolved_base_url http_kwargs __class__s r>rzHoncho.__init__sE`!-!  !91 1  &;<<<#Abi0@&A&A  T (    T ,[ 9   " , \8R S S **'' '  %,K "  ")4K &  &-1/-B-BK) *  $+/ +>+>K (!--  ")4K &%44 44 rAc|jrdS|jtjd|jid|_dS)ad Ensure the workspace exists on the server. The Honcho API uses get-or-create semantics for workspaces via `POST /v3/workspaces`. This SDK uses that endpoint once per client instance to guarantee that subsequent workspace-scoped calls (peers, sessions, queue status, etc.) operate on an existing workspace. NrNr[T)r9r3r]rrHr-r<s r>_ensure_workspacezHoncho._ensure_workspacesK  "  F )++49J2KLLL"&rAcK|jrdS|jtjd|jid{Vd|_dS)z Async version of `_ensure_workspace`. This performs the same get-or-create call, but via the async HTTP client used by `honcho.aio`. NrNr[T)r9rgr]rrHr-r<s r>_ensure_workspace_asynczHoncho._ensure_workspace_asyncs|  "  F%**    tT->&?+         #'rAzUnique identifier for the peerzqOptional metadata dictionary to associate with this peer. If set, will get/create peer immediately with metadata.zaOptional configuration to set for this peer. If set, will get/create peer immediately with flags.r?rCrNr?PeerConfig | Noner c(t||||S)a Get or create a peer with the given ID. Creates a Peer object that can be used to interact with the specified peer. This method does not make an API call unless `configuration` or `metadata` is provided. Args: id: Unique identifier for the peer within the workspace. Should be a stable identifier that can be used consistently across sessions. metadata: Optional metadata dictionary to associate with this peer. If set, will get/create peer immediately with metadata. configuration: Optional configuration to set for this peer. If set, will get/create peer immediately with flags. Returns: A Peer object that can be used to send messages, join sessions, and query the peer's knowledge representations Raises: ValidationError: If the peer ID is empty or invalid rCr?rr=rNr?rCs r>peerz Honcho.peer!sLBMHMMMMrAfiltersSyncPage[PeerResponse, Peer]cjtjjrdind}d fd dfd t |tS)a Get all peers in the current workspace. Makes an API call to retrieve all peers that have been created or used within the current workspace. Returns a paginated result that transforms inner client Peer objects to SDK Peer objects as they are consumed. Returns: A SyncPage of Peer objects representing all peers in the workspace rNr[rrr:r cHt|j|j|jSNr)r rNr?rC)rr=s r> transformzHoncho.peers..transform\s-"0  rApageintrcjtjjrdindd|i}t |t SNrr)r\query)r3r]r peers_listr-rrr next_data fetch_nextrr=rs r>rz Honcho.peers..fetch_nextds] !$"344-4>i))$tn(I I|Y KK KrA)rrr:r )rrr:r)rr3r]rrr-rrr=rrQrrs`` @@r>peersz Honcho.peersIs    z  d/ 0 0)0:)W%%d          L L L L L L L L LlIzBBBrAz!Unique identifier for the sessionzwOptional metadata dictionary to associate with this session. If set, will get/create session immediately with metadata.zgOptional configuration to set for this session. If set, will get/create session immediately with flags.SessionConfiguration | Noner"c(t||||S)a Get or create a session with the given ID. Creates a Session object that can be used to manage conversations between multiple peers. This method does not make an API call unless `configuration` or `metadata` is provided. Args: id: Unique identifier for the session within the workspace. Should be a stable identifier that can be used consistently to reference the same conversation metadata: Optional metadata dictionary to associate with this session. If set, will get/create session immediately with metadata. configuration: Optional configuration to set for this session. If set, will get/create session immediately with flags. Returns: A Session object that can be used to add peers, send messages, and manage conversation context Raises: ValidationError: If the session ID is empty or invalid rr!rs r>sessionzHoncho.sessionnsNr4}xPPPPrA"SyncPage[SessionResponse, Session]cjtjjrdind}d fd dfd t |tS)aL Get all sessions in the current workspace. Makes an API call to retrieve all sessions that have been created within the current workspace. Returns: A SyncPage of Session objects representing all sessions in the workspace. Returns an empty page if no sessions exist rNr[rrr:r"cHt|j|j|jSr)r"rNr?rC)rr=s r>rz"Honcho.sessions..transforms.  )%3  rArrrcjtjjrdindd|i}t |t Sr)r3r]r sessions_listr-rrrs r>rz#Honcho.sessions..fetch_nexts] $T%677-4>i))$tn(I I :NN NrA)rrr:r")rrr:r)rr3r]rrr-rrrs`` @@r>sessionszHoncho.sessionss    z  !2 3 3)0:)W%%d          O O O O O O O O Ooy*EEErA SyncPage[WorkspaceResponse, str]cjtjrdind}d ddfd t |t S)z Get all workspace IDs from the Honcho instance. Makes an API call to retrieve all workspace IDs that the authenticated user has access to. Returns: A paginated SyncPage of workspace ID strings rNr[rKrr:r,c|jSrE)rN)rKs r>rz$Honcho.workspaces..transforms < rArrrcjtjrdindd|i}t |t Srr3r]rworkspaces_listrrrs r>rz%Honcho.workspaces..fetch_nextsY &((-4>i))$tn(I I'8)ZPP PrA)rKrr:r,)rrr:rrrs`` @@r>rHzHoncho.workspacessz  " $ $)0:)W%%d        Q Q Q Q Q Q Q Q Q/JGGGrAzID of the workspace to deletec^|jtj|dS)z Delete a workspace. Makes an API call to delete the specified workspace. This action cannot be undone. Args: workspace_id: The ID of the workspace to delete N)r3deleterrK)r=r-s r>delete_workspacezHoncho.delete_workspaces+ &*<8899999rAzThe search query to usezFilters to scope the search dzNumber of results to return)r.rmler+rlimitr list[Message]c||jtj|j|||d}d|DS)aC Search for messages in the current workspace. Makes an API call to search for messages in the current workspace. Args: query: The search query to use filters: Filters to scope the search. See [search filters documentation](https://docs.honcho.dev/v3/documentation/core-concepts/features/using-filters). limit: Number of results to return (1-100, default: 10) Returns: A list of Message objects representing the search results. Returns an empty list if no messages are found. )rrrr[cZg|](}tjtj|)Srf)rfrom_api_responserrV).0items r> z!Honcho.search.. s>     %o&DT&J&J K K   rA)rr3r]rworkspace_searchr-)r=rrrrQs r>searchz Honcho.searchso2    z  #D$5 6 6 WuEE        rAobserverstr | PeerBase | Nonesenderrstr | SessionBase | NonercH|t|}t|}t|}i}|r||d<|r||d<|r||d<|jt j|j|r|nd}tj|S)a Get the queue processing status, optionally scoped to an observer, sender, and/or session. Args: observer: Optional observer (ID string or Peer object) to scope the status check sender: Optional sender (ID string or Peer object) to scope the status check session: Optional session (ID string or Session object) to scope the status check observer_id sender_id session_idN)r) rr#r3getrworkspace_queue_statusr-rrV) r=rrrresolved_observer_idresolved_sender_idresolved_session_idrrQs r> queue_statuszHoncho.queue_statuss    )(33'//(11 "  8#7E-  4!3E+   6"5E, z~~  )$*; < < *%%d  #1$777rAstr | PeerBaseobservedc|t|}t|}|rt|n|}|jt j|j|||dddS)a Schedule a dream task for memory consolidation. Dreams are background processes that consolidate observations into higher-level insights and update peer cards. This method schedules a dream task for immediate processing. Args: observer: The observer peer (ID string or Peer object) whose perspective to use for the dream. session: Optional session (ID string or Session object) to scope the dream to. observed: Optional observed peer (ID string or Peer object). If not provided, defaults to the observer (self-reflection). omni)rrr dream_typer[N)rr#r3r]rworkspace_schedule_dreamr-)r=rrrrrresolved_observed_ids r>schedule_dreamzHoncho.schedule_dream0s*    )(33(11$, FJx 2F    +D,= > >001$       rAc(d|jd|jdS)z Return a string representation of the Honcho client. Returns: A string representation suitable for debugging zHoncho(workspace_id='z ', base_url='z'))r-r7r<s r>__repr__zHoncho.__repr__Vs% WD$5 V VDN V V V rAcd|jdS)z Return a human-readable string representation of the Honcho client. Returns: A string showing the workspace ID zHoncho Client (workspace: )rOr<s r>__str__zHoncho.__str__asAD,=@@@@rA)r:r/)r:r1)r:r,)r:r5)rQr5r:rR)r:r)rCrr:r_)r:r)r:r)rnrorprqrdror-rorrrsrtrurvrwrxryrzr{r:r_)r:r_)rNr,r?r/rCrr:r rE)rr/r:r)rNr,r?r/rCrr:r")rr/r:r)rr/r:r)r-r,r:r_)rr,rr/rrr:r)NNN)rrrrrrr:r)NN)rrrrrrr:r_).__name__ __module__ __qualname____doc__r model_configr r-__annotations__r r0r2r3r4r6r7r9propertyr?rCrFrIrLrPrYr^r rbrdrgrirrrrrrrrHrrrrrr __classcell__)rs@r>r'r'*s  :G,,,L 9L +6+d*C*C*CICCCC4?K4M4M4MNMMMM)kmmE++++0; D0I0I0IKIIII#.;==L0000 []]I""""*{59999999 X###X# ####3333))))     # # # #16 31 1 1 ,,,,],$X   X X&]**TBBBCCC#=A$uT7TUUU#(5 Q,Q$ $ $ !&dq>R S S S"'% Q$?# # # 5959+05 2, , , W5W5W5W5W5W5DCW5r ' ' ' ' ' ' ' '% A+K   %N .3U L. . . ,15 {, , , %N%N%N%N%N]%NP37#C#C#C#C#CJ% A+N   &Q .3U R. . . 6;U B6 6 6 &Q&Q&Q&Q&Q]&QR37#F#F#F#F#FL37HHHHH<"E A+J   ::::]: U31:STTT,1E ;- - - U12O       ]  D]**TBBBCCC+/(,,0 8888DC8B]**TBBBCCC-1*. # # # # DC# J     AAAAAAAArAr')4r __future__rloggingrcollections.abcrtypingrrhttpxpydanticrrr r r rir api_typesrrrrrrrrbaserrhttprrrmessagermixinsr paginationrrr rr"utilsr# getLoggerrloggerrr'rfrAr>rs"""""" ###### MMMMMMMMMMMMMM                    ('''''''AAAAAAAAAA''''''   8 $ $%* ~A~A~A~A~AY+~A~A~A~A~ArA