fleet-memory/hindsight-embed/hindsight_embed/embed_manager.py
Nicolò Boschi d3302c95b9
feat: HindsightEmbedded python SDK (#293)
* feat: HindsightEmbedded python SDK

* feat: HindsightEmbedded python SDK

* fixes

* improve

* ci

* improvemnts

* fix test

* fix test

* fix: update tests to use Pydantic model attributes instead of dict access

- Fixed test_server_integration.py to access Pydantic model attributes directly
- Changed dict-style access (response["field"]) to attribute access (response.field)
- Fixed .get() calls on Pydantic models
- Updated recall() calls to access .results attribute
- Updated reflect() calls to access .text attribute
- Fixed test_list_banks to use namespace API instead of deleted default_api
- Fixed attribute shadowing in HindsightClient wrapper (renamed _*_api to _*_namespace)

* fix: add list() method to BanksAPI namespace

* fix: remove leftover async cleanup code from test_list_banks

* docs: remove Advanced Configuration section from embed.md
2026-02-04 14:41:19 +01:00

83 lines
2 KiB
Python

"""
Abstract interface for managing Hindsight embedded servers and profiles.
This module provides a clean interface for daemon lifecycle and profile management,
abstracting away the implementation details.
"""
from abc import ABC, abstractmethod
from typing import Optional
class EmbedManager(ABC):
"""Abstract interface for managing Hindsight embedded servers and profiles."""
@abstractmethod
def ensure_running(self, config: dict, profile: str) -> bool:
"""
Ensure daemon is running for the given profile with config.
Args:
config: Environment configuration dict (HINDSIGHT_API_* vars)
profile: Profile name for isolation
Returns:
True if daemon is running (started or already running), False on failure
"""
pass
@abstractmethod
def get_url(self, profile: str) -> str:
"""
Get the URL for the daemon serving this profile.
Args:
profile: Profile name
Returns:
URL string (e.g., "http://127.0.0.1:54321")
Raises:
RuntimeError: If daemon is not running
"""
pass
@abstractmethod
def stop(self, profile: str) -> bool:
"""
Stop the daemon for this profile.
Args:
profile: Profile name
Returns:
True if stopped successfully, False otherwise
"""
pass
@abstractmethod
def is_running(self, profile: str) -> bool:
"""
Check if daemon is running for this profile.
Args:
profile: Profile name
Returns:
True if daemon is running and responsive
"""
pass
@abstractmethod
def get_database_url(self, profile: str, db_url: Optional[str] = None) -> str:
"""
Get the database URL for this profile.
Args:
profile: Profile name
db_url: Optional override database URL
Returns:
Database connection string
"""
pass