%jU1dZddlmZddlmZddlmZGddZGddZGd d Z Gd d Z Gd deZ dS)z Wrapper for Hindsight client that adds API namespaces. Provides organized access to different parts of the Hindsight API through namespaces like .banks, .mental_models, etc. ) annotations)Any) HindsightcJeZdZdZddZ dddZddZddZddZddZ dS)BanksAPIzqNamespace for bank-related operations. Provides methods to create, delete, and manage memory banks. clientrc||_dSN_clientselfrs `/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/hindsight/client_wrapper.py__init__zBanksAPI.__init__  Nbank_idstrname str | Nonemission dispositiondict[str, Any] | Nonereturnrc>|j||||S)alCreate a new bank. Args: bank_id: Unique identifier for the bank. name: Optional display name for the bank. mission: Optional mission statement for the bank. disposition: Optional disposition configuration dict. Returns: Bank creation response from the API. )rrrr)r create_bank)rrrrrs rcreatezBanksAPI.creates/$|''# (   rc8|j|S)zDelete a bank. Args: bank_id: The ID of the bank to delete. Returns: Deletion response from the API. )r)r delete_bank)rrs rdeletezBanksAPI.delete1s|'''888rc:|j||S)zSet or update the mission for a bank. Args: bank_id: The ID of the bank. mission: The mission statement to set. Returns: API response confirming the update. )rr)r set_mission)rrrs rr"zBanksAPI.set_mission<s|'''IIIrdict[str, Any]c:|j||S)zSet or update the disposition for a bank. Args: bank_id: The ID of the bank. disposition: The disposition configuration dict. Returns: API response confirming the update. )rr)r set_disposition)rrrs rr%zBanksAPI.set_dispositionHs|++G+UUUrc\ddlm}||jjS)zZList all banks. Returns: List of banks from the API. r) _run_async)!hindsight_client.hindsight_clientr'r _banks_api list_banks)rr's rlistz BanksAPI.listTs8 A@@@@@z$,1<<>>???rrrNNN) rrrrrrrrrr)rrrr)rrrrrr)rrrr#rr)rr) __name__ __module__ __qualname____doc__rrr r"r%r+rrrrs   "-1      2 9 9 9 9 J J J J V V V V@@@@@@rrcXeZdZdZddZ dddZdddZddZddZ dddZ ddZ dS)MentalModelsAPIzNamespace for mental model operations. Mental models are reusable knowledge structures that guide agent behavior. rrc||_dSr r r s rrzMentalModelsAPI.__init__errNrrrcontenttagslist[str] | Nonerrc>|j||||S)anCreate a new mental model. Args: bank_id: The ID of the bank to add the model to. name: Name for the mental model. content: The content/instructions for the mental model. tags: Optional list of tags for categorization. Returns: Creation response from the API. rrr6r7)r create_mental_modelrrrr6r7s rrzMentalModelsAPI.createhs/$|// 0   rc:|j||S)zList all mental models for a bank. Args: bank_id: The ID of the bank. tags: Optional filter by tags. Returns: List of mental models. rr7)r list_mental_modelsrrr7s rr+zMentalModelsAPI.lists|..wT.JJJrmental_model_idc:|j||S)zGet a specific mental model. Args: bank_id: The ID of the bank. mental_model_id: The ID of the mental model. Returns: The mental model details. rrA)r get_mental_modelrrrAs rgetzMentalModelsAPI.gets|,,Wo,^^^rc:|j||S)zRefresh a mental model. Args: bank_id: The ID of the bank. mental_model_id: The ID of the mental model to refresh. Returns: Refresh response from the API. rC)r refresh_mental_modelrEs rrefreshzMentalModelsAPI.refreshs |00Ra0bbbrrc@|j|||||S)adUpdate a mental model. Args: bank_id: The ID of the bank. mental_model_id: The ID of the mental model to update. name: Optional new name. content: Optional new content. tags: Optional new tags list. Returns: Update response from the API. )rrArr6r7)r update_mental_model)rrrArr6r7s rupdatezMentalModelsAPI.updates2(|//+ 0   rc:|j||S)zDelete a mental model. Args: bank_id: The ID of the bank. mental_model_id: The ID of the mental model to delete. Returns: Deletion response from the API. rC)r delete_mental_modelrEs rr zMentalModelsAPI.deletes |//Q`/aaarr,r rrrrr6rr7r8rrrrr7r8rr)rrrArrrr-) rrrArrrr6rr7r8rr) r.r/r0r1rrr+rFrIrLr r2rrr4r4_s "&      2 K K K K K _ _ _ _ c c c c  "!%      8 b b b b b brr4cPeZdZdZddZ dddZdddZddZ dddZddZ dS) DirectivesAPIzqNamespace for directive operations. Directives are explicit instructions that guide agent behavior. rrc||_dSr r r s rrzDirectivesAPI.__init__rrNrrrr6r7r8rrc>|j||||S)aaCreate a new directive. Args: bank_id: The ID of the bank to add the directive to. name: Name for the directive. content: The directive content/instructions. tags: Optional list of tags for categorization. Returns: Creation response from the API. r:)r create_directiver<s rrzDirectivesAPI.creates/$|,, -   rc:|j||S)zList all directives for a bank. Args: bank_id: The ID of the bank. tags: Optional filter by tags. Returns: List of directives. r>)r list_directivesr@s rr+zDirectivesAPI.lists|++G$+GGGr directive_idc:|j||S)zGet a specific directive. Args: bank_id: The ID of the bank. directive_id: The ID of the directive. Returns: The directive details. rrX)r get_directiverrrXs rrFzDirectivesAPI.gets|))' )UUUrrc@|j|||||S)a[Update a directive. Args: bank_id: The ID of the bank. directive_id: The ID of the directive to update. name: Optional new name. content: Optional new content. tags: Optional new tags list. Returns: Update response from the API. )rrXrr6r7)r update_directive)rrrXrr6r7s rrLzDirectivesAPI.updates2(|,,% -   rc:|j||S)zDelete a directive. Args: bank_id: The ID of the bank. directive_id: The ID of the directive to delete. Returns: Deletion response from the API. rZ)r delete_directiver\s rr zDirectivesAPI.delete$s|,,W<,XXXrr,r rOrP)rrrXrrrr-) rrrXrrrr6rr7r8rr) r.r/r0r1rrr+rFrLr r2rrrRrRs "&      2 H H H H H V V V V  "!%      8 Y Y Y Y Y YrrRc,eZdZdZddZ dddZdS) MemoriesAPIzfNamespace for memory operations. Provides methods to query and retrieve stored memories. rrc||_dSr r r s rrzMemoriesAPI.__init__7rrNdrrrtyper search_querylimitintoffsetrrc@|j|||||S)aList memories in a bank. Args: bank_id: The ID of the bank to query. type: Optional filter by memory type. search_query: Optional search query for filtering. limit: Maximum number of results to return (default: 100). offset: Number of results to skip for pagination (default: 0). Returns: List of memories matching the criteria. )rrerfrgri)r list_memories)rrrerfrgris rr+zMemoriesAPI.list:s2(|))% *   rr,)NNrdr) rrrerrfrrgrhrirhrr)r.r/r0r1rr+r2rrrbrb1sY   #'        rrbceZdZdZdfd Zedd Zedd Zedd ZeddZ xZ S)HindsightClienta9 Enhanced Hindsight client with organized API namespaces. This wrapper extends the auto-generated Hindsight client with organized access to different parts of the API through namespaces. Example: ```python from hindsight import HindsightClient client = HindsightClient(base_url="http://localhost:8888") # Core operations (inherited from Hindsight) client.retain(bank_id="test", content="Hello") results = client.recall(bank_id="test", query="Hello") # Organized API access through namespaces client.banks.create(bank_id="test", name="Test Bank") models = client.mental_models.list(bank_id="test") directives = client.directives.list(bank_id="test") memories = client.memories.list(bank_id="test") ``` Attributes: banks: Namespace for bank management operations. mental_models: Namespace for mental model operations. directives: Namespace for directive operations. memories: Namespace for memory listing operations. argsrkwargsrNonecrtj|i|d|_d|_d|_d|_dSr )superr_banks_namespace_mental_models_namespace_directives_namespace_memories_namespace)rrnro __class__s rrzHindsightClient.__init__vsE$)&)))15@D%;?"7;   rrcF|jt||_|jS)zxAccess bank management operations. Returns: BanksAPI instance for bank operations. )rsrrs rbankszHindsightClient.banks}s$  ($,TNND !$$rr4cF|jt||_|jS)zAccess mental model operations. Returns: MentalModelsAPI instance for mental model operations. )rtr4rys r mental_modelszHindsightClient.mental_modelss&  ( 0,;D,A,AD ),,rrRcF|jt||_|jS)z|Access directive operations. Returns: DirectivesAPI instance for directive operations. )rurRrys r directiveszHindsightClient.directivess&  % -)6t)<)rsY#"""""&&&&&&M@M@M@M@M@M@M@M@`lblblblblblblblb^`Y`Y`Y`Y`Y`Y`Y`YF# # # # # # # # LP(P(P(P(P(iP(P(P(P(P(r