56489a58d1
* refactor(llamaindex): merge two packages into single hindsight-llamaindex Merge `llama-index-tools-hindsight` and `llama-index-memory-hindsight` into a single `hindsight-llamaindex` package following our naming convention. - Rename package to `hindsight-llamaindex` (Python module: `hindsight_llamaindex`) - Move HindsightToolSpec and HindsightMemory into the same package - Delete `llamaindex-memory/` directory - Add CI test job for llamaindex integration - Update docs, blog post, and integrations.json * fix(blog): update llamaindex blog post for merged package - Move date to 2026-03-30 - Add HindsightMemory (automatic BaseMemory) pattern - Fix "bank must exist first" pitfall — mission auto-creates - Align all code examples with docs page - Update architecture diagram to show both patterns * fix(docs): add llamaindex/openai icons, rename Codex - Add llamaindex.png and openai.png icons - Rename "OpenAI Codex CLI" to "Codex" in integrations.json and docs - Use openai.png icon for Codex integration
101 lines
3.2 KiB
Python
101 lines
3.2 KiB
Python
"""Global configuration for Hindsight-LlamaIndex integration."""
|
|
|
|
import os
|
|
from dataclasses import dataclass
|
|
from typing import Optional
|
|
|
|
DEFAULT_HINDSIGHT_API_URL = "https://api.hindsight.vectorize.io"
|
|
HINDSIGHT_API_KEY_ENV = "HINDSIGHT_API_KEY"
|
|
|
|
|
|
@dataclass
|
|
class HindsightLlamaIndexConfig:
|
|
"""Connection and default settings for the LlamaIndex integration.
|
|
|
|
Attributes:
|
|
hindsight_api_url: URL of the Hindsight API server.
|
|
api_key: API key for Hindsight authentication.
|
|
budget: Default recall budget level (low/mid/high).
|
|
max_tokens: Default maximum tokens for recall results.
|
|
tags: Default tags applied when storing memories.
|
|
recall_tags: Default tags to filter when searching memories.
|
|
recall_tags_match: Tag matching mode (any/all/any_strict/all_strict).
|
|
context: Source label for retain operations (default: "llamaindex").
|
|
mission: Bank mission for fact extraction context.
|
|
verbose: Enable verbose logging.
|
|
"""
|
|
|
|
hindsight_api_url: str = DEFAULT_HINDSIGHT_API_URL
|
|
api_key: Optional[str] = None
|
|
budget: str = "mid"
|
|
max_tokens: int = 4096
|
|
tags: Optional[list[str]] = None
|
|
recall_tags: Optional[list[str]] = None
|
|
recall_tags_match: str = "any"
|
|
context: str = "llamaindex"
|
|
mission: Optional[str] = None
|
|
verbose: bool = False
|
|
|
|
|
|
_global_config: Optional[HindsightLlamaIndexConfig] = None
|
|
|
|
|
|
def configure(
|
|
hindsight_api_url: Optional[str] = None,
|
|
api_key: Optional[str] = None,
|
|
budget: str = "mid",
|
|
max_tokens: int = 4096,
|
|
tags: Optional[list[str]] = None,
|
|
recall_tags: Optional[list[str]] = None,
|
|
recall_tags_match: str = "any",
|
|
context: str = "llamaindex",
|
|
mission: Optional[str] = None,
|
|
verbose: bool = False,
|
|
) -> HindsightLlamaIndexConfig:
|
|
"""Configure Hindsight connection and default settings.
|
|
|
|
Args:
|
|
hindsight_api_url: Hindsight API URL (default: production).
|
|
api_key: API key. Falls back to HINDSIGHT_API_KEY env var.
|
|
budget: Default recall budget (low/mid/high).
|
|
max_tokens: Default max tokens for recall.
|
|
tags: Default tags for retain operations.
|
|
recall_tags: Default tags to filter recall/search.
|
|
recall_tags_match: Tag matching mode.
|
|
context: Source label for retain operations.
|
|
mission: Bank mission for fact extraction context.
|
|
verbose: Enable verbose logging.
|
|
|
|
Returns:
|
|
The configured HindsightLlamaIndexConfig.
|
|
"""
|
|
global _global_config
|
|
|
|
resolved_url = hindsight_api_url or DEFAULT_HINDSIGHT_API_URL
|
|
resolved_key = api_key or os.environ.get(HINDSIGHT_API_KEY_ENV)
|
|
|
|
_global_config = HindsightLlamaIndexConfig(
|
|
hindsight_api_url=resolved_url,
|
|
api_key=resolved_key,
|
|
budget=budget,
|
|
max_tokens=max_tokens,
|
|
tags=tags,
|
|
recall_tags=recall_tags,
|
|
recall_tags_match=recall_tags_match,
|
|
context=context,
|
|
mission=mission,
|
|
verbose=verbose,
|
|
)
|
|
|
|
return _global_config
|
|
|
|
|
|
def get_config() -> Optional[HindsightLlamaIndexConfig]:
|
|
"""Get the current global configuration."""
|
|
return _global_config
|
|
|
|
|
|
def reset_config() -> None:
|
|
"""Reset global configuration to None."""
|
|
global _global_config
|
|
_global_config = None
|