%j dZddlZddlZddlZddlZddlZddlmZddlZddlm Z ddl m Z ddl m Z ejeZdefdZGd d Z ddededededeededeededededefdZdS)z Server module for running Hindsight in a background thread. Provides a simple way to start and stop the Hindsight HTTP API server without blocking the main thread. N)Optional)Config) MemoryEngine) create_appreturnctjtjtj5}|d|d|d}dddn #1swxYwY|S)zFind a free port on localhost.)rN)socketAF_INET SOCK_STREAMbindlisten getsockname)sports X/home/rurouni/.hermes/hermes-agent/venv/lib/python3.11/site-packages/hindsight/server.py_find_free_portrs v~v'9 : :"a w  }}q!""""""""""""""" KsAA;;A?A?ceZdZdZ dd ed ed ed edeededeededefdZe defdZ dZ dde ddfdZ d de ddfdZd!dZd"dZdS)#Servera Hindsight server that runs in a background thread. Example: ```python from hindsight import Server server = Server( db_url="pg0", llm_provider="groq", llm_api_key="your-api-key", llm_model="openai/gpt-oss-120b" ) server.start() print(f"Server running at {server.url}") # Use the server... server.stop() ``` pg0groqr openai/gpt-oss-120bN 127.0.0.1Finfodb_url llm_provider llm_api_key llm_model llm_base_urlhostr mcp_enabled log_levelc *||_||_||_||_||_||_|p t |_||_| |_ d|_ d|_ d|_ tj|_tj|_dS)aw Initialize the Hindsight server. Args: db_url: Database URL. Use "pg0" for embedded PostgreSQL. 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 host: Host to bind to (default: 127.0.0.1) port: Port to bind to (default: auto-select free port) mcp_enabled: Whether to enable MCP server log_level: Uvicorn log level (default: warning) N)rrrrr r!rrr"r#_memory_server_thread threadingEvent_started_stopped) selfrrrrr r!rr"r#s r__init__zServer.__init__8s4 (&"( -O-- &"/3 15 37 !)) !)) rc&d|jd|jS)zGet the server URL.zhttp://:)r!rr,s rurlz Server.urlbs100TY000r.ctj}tj| t|j|j|j|j|j|_ t|j |j d}t||j |j|jd}t!j||_|j||jn0#t.$r#}t0d|d}~wwxYw |j r,||j ||jdS#|j r,||j ||jwxYw)z&Run the server in a background thread.)rmemory_llm_providermemory_llm_api_keymemory_llm_modelmemory_llm_base_urlT)memorymcp_api_enabledinitialize_memoryasyncio)appr!rr#loopzServer error: N)r;new_event_loopset_event_looprrrrrr r%rr"rr!rr#uvicornrr&r*setrun_until_completeserve Exceptionloggererrorcloser+)r,r=r<configes r _run_serverzServer._run_servergs%''t$$$) '{$($5#'#3!%$($5 DL| $ 0"&CYY. F#>&11DL M       # #DL$6$6$8$8 9 9 9 9    LL-!-- . . .   :| >'' (:(:(<(<=== JJLLL M       | >'' (:(:(<(<=== JJLLL M      s+CC98F 9 D&D!!D&&F A"G.>@timeoutcL|j(|jrtd|j|jt j|jd|_|j |j |tj }tj |z |kr tj |j|jfd5t d|j|cdddS#1swxYwYn6#t&tjt*f$rtjdYnwxYwtj |z |ktd |d ) a Start the server in a background thread. Args: timeout: Maximum time to wait for server to start (seconds) Returns: self (for chaining) Raises: RuntimeError: If server fails to start within timeout NzServer is already runningT)targetdaemonrLr zHindsight server started at g?zServer failed to start within z seconds)r'is_alive RuntimeErrorr*clearr+r(ThreadrJstartwaittimer create_connectionr!rrErr2ConnectionRefusedErrorrLOSErrorsleep)r,rL start_times rrUz Server.starts < # (=(=(?(? #:;; ;   't/?MMM   7+++Y[[ ikkJ&00 -ty$).DaPPP  KK Itx I IJJJ                 +FNGD    3 ikkJ&00MGMMMNNNs6$"E$D7* E7D;;E>D;?E0E65E6$@c*|jdSd|j_|jN|j||jrt dd|_d|_t ddS)zx Stop the server. Args: timeout: Maximum time to wait for server to stop (seconds) NTrPz"Server thread did not stop cleanlyzHindsight server stopped)r& should_exitr'joinrQrEwarningr)r,rLs rstopz Server.stops <  F$(   < # L  g  . . .|$$&& ECDDD   ./////r.c*|S)zContext manager entry.)rUr1s r __enter__zServer.__enter__szz||r.c.|dS)zContext manager exit.N)rb)r,exc_typeexc_valexc_tbs r__exit__zServer.__exit__s r.) rrr rNrNFr)rK)r])rr)rN)__name__ __module__ __qualname____doc__strrintboolr-propertyr2rJfloatrUrbrdrir.rrr sj2".&*"!(*(*(*(* (*  (* sm (*(*sm(*(*(*(*(*(*T1S111X1. . . `#O#OU#Oh#O#O#O#OJ00E0T0000.r.rrrr rrFrarKrrrrr r!rr"r#rLc ^t||||||||| } | | S)a Start a Hindsight server in a background thread. This is a convenience function that creates and starts a Server instance. Args: db_url: Database URL. Use "pg0" for embedded PostgreSQL. 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 host: Host to bind to (default: 127.0.0.1) port: Port to bind to (default: auto-select free port) mcp_enabled: Whether to enable MCP server log_level: Uvicorn log level (default: warning) timeout: Maximum time to wait for server to start (seconds) Returns: Running Server instance Example: ```python from hindsight import start_server, Client server = start_server( db_url="pg0", llm_provider="groq", llm_api_key="your-api-key", llm_model="openai/gpt-oss-120b" ) client = Client(base_url=server.url) client.put(agent_id="assistant", content="User likes Python") server.stop() ``` ) rrrrr r!rr"r#rP)rrU) rrrrr r!rr"r#rLservers r start_serverrvsJb!!     F <<< ( ((r.) rrr rNrNFrarK)rmr;loggingr r(rWtypingrr@r hindsight_apirhindsight_api.apir getLoggerrjrErorrrnrprrrvrsr.rr|s   &&&&&&((((((  8 $ $yyyyyyyyz*"&<)<) <)<)<) <) 3- <)  <) 3-<)<)<)<) <)<)<)<)<)<)r.