%j;dZddlZddlZddlmZddlmZddlmZddl m Z m Z m Z m Z ejeZGdd ZdS) a Embedded Hindsight client with automatic daemon lifecycle management. This module provides HindsightEmbedded, a client that uses the same daemon management interface as hindsight-embed CLI, ensuring full compatibility. Example: ```python from hindsight import HindsightEmbedded # Daemon starts automatically on first use client = HindsightEmbedded( profile="myapp", llm_provider="groq", llm_api_key="your-api-key", ) # Use just like HindsightClient client.retain(bank_id="alice", content="Alice loves AI") results = client.recall(bank_id="alice", query="What does Alice like?") # Optional cleanup client.close() ``` Using context manager: ```python from hindsight import HindsightEmbedded with HindsightEmbedded(profile="myapp") as client: client.retain(bank_id="alice", content="Alice loves AI") # Daemon managed automatically ``` N)Optional) Hindsight)get_embed_manager)BanksAPI DirectivesAPI MemoriesAPIMentalModelsAPIceZdZdZ d*d ed ed ededeedeededededeedefdZdZ d+defdZ d+defdZ defdZ dZ dZd Zed!efd"Zed!efd#Zed!efd$Zed!efd%Zed!efd&Zed!efd'Zed!efd(Zed!efd)ZdS),HindsightEmbeddeda Hindsight client with automatic daemon lifecycle management. This client uses the same daemon management interface as hindsight-embed CLI, ensuring full compatibility and shared profiles. The daemon is started automatically on first use and manages profile-specific databases. Profile data is stored in: ~/.pg0/instances/hindsight-embed-{profile}/ All methods from HindsightClient are available: - retain(), retain_batch() - recall() - reflect() - create_bank(), set_mission(), delete_bank() - create_mental_model(), list_mental_models(), etc. - create_directive(), list_directives(), etc. - And all async variants (aretain, arecall, areflect, etc.) Args: profile: Profile name for data isolation (default: "default") llm_provider: LLM provider ("groq", "openai", "ollama", "gemini", "anthropic", "lmstudio") llm_api_key: API key for the LLM provider llm_model: Model name to use llm_base_url: Optional custom base URL for LLM API database_url: Optional database URL override (default: profile-specific pg0) idle_timeout: Seconds before daemon auto-exits when idle (default: 0, disabled) log_level: Daemon log level (default: "info") ui: Whether to start the control plane web UI alongside the daemon (default: False) ui_port: Port for the UI. Defaults to daemon_port + 10000. ui_hostname: Hostname to bind the UI to. Defaults to "0.0.0.0". defaultgroqopenai/gpt-oss-120bNrinfoF0.0.0.0profile llm_provider llm_api_key llm_model llm_base_url database_url idle_timeout log_leveluiui_port ui_hostnamec Z||_||||t|d|_|r ||jd<|r ||jd<| |_| |_| |_d|_tj|_ d|_ d|_ t|_ d|_d|_d|_d|_dS)a Initialize the embedded client (daemon starts on first use). Args: profile: Profile name for data isolation llm_provider: LLM provider llm_api_key: API key for the LLM provider llm_model: Model name to use llm_base_url: Optional custom base URL for LLM API database_url: Optional database URL override idle_timeout: Seconds before daemon auto-exits when idle (0 = disabled) log_level: Daemon log level ui: Whether to start the control plane web UI alongside the daemon ui_port: Port for the UI (defaults to daemon_port + 10000) ui_hostname: Hostname to bind the UI to (defaults to "0.0.0.0") )HINDSIGHT_API_LLM_PROVIDERHINDSIGHT_API_LLM_API_KEYHINDSIGHT_API_LLM_MODELHINDSIGHT_API_LOG_LEVEL#HINDSIGHT_EMBED_DAEMON_IDLE_TIMEOUTHINDSIGHT_API_LLM_BASE_URL HINDSIGHT_EMBED_API_DATABASE_URLNF)rstrconfig_ui_ui_port _ui_hostname_client threadingLock_lock_started_closedr_manager _banks_api_mental_models_api_directives_api _memories_api) selfrrrrrrrrrrrs Z/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/hindsight/embedded.py__init__zHindsightEmbedded.__init__Qs< +7)4'0'036|3D3D      E8DDK4 5  K>JDK: ; ',0 ^%%   )++ /3=A8<48c|jr|j|j|jrdSt d|j |jn,#t$rt ddYnwxYwd|_d|_|j 5|jr|j|j|jr ddddSt d|j |jn,#t$rt ddYnwxYwd|_d|_|j rtdt d |jd |j|j|j}|std |jd |j|j}t#| |_d|_t d||jrst d|jd |j|j|j|j}|s#t d|jd ddddS#1swxYwYdS)z>Ensure daemon is running (thread-safe), restarting if crashed.Nz>Daemon for profile '%s' is no longer responsive, restarting...zError closing stale clientTexc_infoFzJDaemon for profile '%s' is no longer responsive (lock path), restarting...z5Cannot use HindsightEmbedded after it has been closedz(Ensuring daemon is running for profile ''...z$Failed to start daemon for profile '')base_urlzConnected to daemon at zStarting UI for profile 'z Failed to start UI for profile ')r/r+r1 is_runningrloggerwarningclose Exceptiondebugr.r0 RuntimeErrorrensure_runningr'get_urlrr(start_uir)r*)r6success daemon_url ui_starteds r7_ensure_startedz!HindsightEmbedded._ensure_startedsm = "T\5}'' 55  NNP     J ""$$$$ J J J 9D IIIII JDL!DM Z* W* W} &!9=++DL99 * W* W* W* W* W* W* W* W `LNL&&(((( NNNLL!=LMMMMMN# % | "K KKU4<UUU V V Vm224; MMG "J4<JJJ ..t|<*>> ? ? ?x W J JJJKKK!]33L$-1B "WNN#Udl#U#U#UVVVU* W* W* W* W* W* W* W* W* W* W* W* W* W* W* W* W* W* WsOA++&BB,/J4( J4 D#"J4#&E  J4 E  EJ44J8;J8stop_daemon_on_closec`|jrdS|jd}|s)td|jd|_dS |jr |jdS|jT |jn2#t$r%t d|jdYnwxYwd|_|j rO|j rHt d|jd |j|j|j|rI|j rBt d |jd |j|jd|_|jdS#|jwxYw) z Cleanup client resources (idempotent). Args: stop_daemon_on_close: If True, stops the daemon. Otherwise, daemon continues running (it will auto-stop after idle timeout). Ng@)timeoutzeCleanup lock acquisition timed out for profile '%s'; marking closed, daemon will idle-stop on its ownTz%Error closing client for profile '%s'r;zStopping UI for profile 'r=zStopping daemon for profile ')r0r.acquirerArBrreleaser+rCrDrEr(r/rr1stop_uir)stop)r6rNacquireds r7_cleanupzHindsightEmbedded._cleanups <  F:%%c%22  NNC      DL F !| 2 J   /|'L&&(((( LL? !%! $ x CDM C J JJJKKK %%dlDMBBB$ 1  1 NDLNNNOOO ""4<000DL J   DJ   s7F4F<BF,CFCB2FF- stop_daemonc2||dS)a Explicitly close the client. Args: stop_daemon: If True, stops the daemon. Otherwise, daemon continues running and will auto-stop after idle timeout (default: False). Note: The daemon may be shared with other clients or the CLI, so stopping it might affect other users. By default, we rely on the daemon's idle timeout. )rNNrV)r6rWs r7rCzHindsightEmbedded.closes ; 77777r9namecT|t|j|S)z Proxy all method calls to the underlying Hindsight client. This allows HindsightEmbedded to expose all HindsightClient methods without manually wrapping each one. )rMgetattrr+)r6rZs r7 __getattr__zHindsightEmbedded.__getattr__s( t|T***r9c.||S)z2Context manager entry - ensures server is started.)rMr6s r7 __enter__zHindsightEmbedded.__enter__s  r9c.|dS)z(Context manager exit - stops the server.N)rC)r6exc_typeexc_valexc_tbs r7__exit__zHindsightEmbedded.__exit__!s r9c.|dS)zCleanup on garbage collection.NrYr_s r7__del__zHindsightEmbedded.__del__%s r9returncF|jt||_|jS)a Access bank management operations. Each method call ensures the daemon is running before executing. Example: ```python from hindsight import HindsightEmbedded embedded = HindsightEmbedded(profile="myapp", ...) # Create a bank embedded.banks.create(bank_id="test", name="Test Bank") # Set mission embedded.banks.set_mission(bank_id="test", mission="Help users") ``` )r2rr_s r7bankszHindsightEmbedded.banks)s!( ? "&tnnDOr9cF|jt||_|jS)ab Access mental model operations. Each method call ensures the daemon is running before executing. Example: ```python from hindsight import HindsightEmbedded embedded = HindsightEmbedded(profile="myapp", ...) # Create a mental model embedded.mental_models.create( bank_id="test", name="User Preferences", content="User prefers dark mode" ) # List mental models models = embedded.mental_models.list(bank_id="test") ``` )r3r r_s r7 mental_modelszHindsightEmbedded.mental_modelsAs&0  " *&5d&;&;D #&&r9cF|jt||_|jS)a] Access directive operations. Each method call ensures the daemon is running before executing. Example: ```python from hindsight import HindsightEmbedded embedded = HindsightEmbedded(profile="myapp", ...) # Create a directive embedded.directives.create( bank_id="test", name="Response Style", content="Always be concise and friendly" ) # List directives directives = embedded.directives.list(bank_id="test") ``` )r4rr_s r7 directiveszHindsightEmbedded.directives]s&0   '#0#6#6D ##r9cF|jt||_|jS)a Access memory listing operations. Each method call ensures the daemon is running before executing. Example: ```python from hindsight import HindsightEmbedded embedded = HindsightEmbedded(profile="myapp", ...) # List memories memories = embedded.memories.list( bank_id="test", type="world", limit=50 ) ``` )r5r r_s r7memorieszHindsightEmbedded.memoriesys&*   %!,T!2!2D !!r9c8||jS)a Get the underlying Hindsight client for direct access. Ensures daemon is started (and restarts it if it has crashed) before returning the client. Returns: Hindsight: The underlying client instance Example: ```python from hindsight import HindsightEmbedded embedded = HindsightEmbedded(profile="myapp", ...) client = embedded.client banks = client.list_banks() ``` )rMr+r_s r7clientzHindsightEmbedded.clients* |r9ch||j|jS)z-Get the daemon URL (starts daemon if needed).)rMr1rHrr_s r7urlzHindsightEmbedded.urls. }$$T\222r9cp|jo/|j o'|jduo|j|jS)z@Check if the client is initialized and the daemon is responsive.N)r/r0r+r1r@rr_s r7r@zHindsightEmbedded.is_runningsF M 7L  7 D( 7 ((66  r9c@|j|jS)z Get the UI URL for this profile.)r1 get_ui_urlrr_s r7ui_urlzHindsightEmbedded.ui_urls}'' 555r9) r rrrNNrrFNr)F)__name__ __module__ __qualname____doc__r&rintboolr8rMrVrCr]r`rergpropertyrrjr rlrrnr rprrrrtr@rxr9r7r r 0sD!".&*&*!%$=9=9=9=9 =9  =9 sm =9sm=9=9=9 =9#=9=9=9=9=9~;W;W;Wz3!3!T3!3!3!3!j 8 8 8 8 8 8 + + + + + xX.''''X'6$M$$$X$6"+"""X"0 X.3S333X3  D   X 6666X666r9r )r|loggingr,typingrhindsight_clientrhindsight_embedrapi_namespacesrrr r getLoggerryrAr rr9r7rs!!F&&&&&&------QQQQQQQQQQQQ  8 $ $M6M6M6M6M6M6M6M6M6M6r9