danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix checking settings

Browse source
This commit is contained in:
Yiorgis Gozadinos 2025-09-02 13:01:36 +03:00
parent 9e5f11060d
commit ecf8894755
No known key found for this signature in database
1 changed files with 38 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

61
src/haiku/rag/store/repositories/settings.py
View file

@ -97,32 +97,47 @@ class SettingsRepository:
def validate_config_compatibility(self) -> None: def validate_config_compatibility(self) -> None:
"""Validate that the current configuration is compatible with stored settings.""" """Validate that the current configuration is compatible with stored settings."""
try: stored_settings = self.get_current_settings()
stored_settings = self.get_current_settings()
current_config = Config.model_dump(mode="json")
# Check if embedding provider or model has changed # If no stored settings, this is a new database - save current config and return
stored_provider = stored_settings.get("embedding_provider") if not stored_settings:
current_provider = current_config.get("embedding_provider") self.save_current_settings()
return
stored_model = stored_settings.get("embedding_model") current_config = Config.model_dump(mode="json")
current_model = current_config.get("embedding_model")
if (stored_provider and stored_provider != current_provider) or ( # Check if embedding provider or model has changed
stored_model and stored_model != current_model stored_provider = stored_settings.get("EMBEDDINGS_PROVIDER")
): current_provider = current_config.get("EMBEDDINGS_PROVIDER")
# Provider or model changed - need to recreate embeddings
from rich.console import Console
console = Console() stored_model = stored_settings.get("EMBEDDINGS_MODEL")
console.print( current_model = current_config.get("EMBEDDINGS_MODEL")
"[yellow]Warning: Embedding provider/model changed. "
"You may need to recreate embeddings for optimal performance.[/yellow]"
)
# Optionally recreate embeddings table stored_vector_dim = stored_settings.get("EMBEDDINGS_VECTOR_DIM")
# self.store.recreate_embeddings_table() current_vector_dim = current_config.get("EMBEDDINGS_VECTOR_DIM")
except Exception: # Check for incompatible changes
# If we can't validate, just continue incompatible_changes = []
pass
if stored_provider and stored_provider != current_provider:
incompatible_changes.append(
f"Embedding provider changed from '{stored_provider}' to '{current_provider}'"
)
if stored_model and stored_model != current_model:
incompatible_changes.append(
f"Embedding model changed from '{stored_model}' to '{current_model}'"
)
if stored_vector_dim and stored_vector_dim != current_vector_dim:
incompatible_changes.append(
f"Vector dimension changed from {stored_vector_dim} to {current_vector_dim}"
)
if incompatible_changes:
error_msg = (
"Database configuration is incompatible with current settings:\n"
+ "\n".join(f" - {change}" for change in incompatible_changes)
)
error_msg += "\n\nPlease rebuild the database using: haiku-rag rebuild"
raise ConfigMismatchError(error_msg)