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

Fix info to not use old .env settings

Browse source
This commit is contained in:
Yiorgis Gozadinos 2025-10-29 13:31:54 +02:00
parent b289aa04e0
commit f9b6894974
No known key found for this signature in database
2 changed files with 9 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/haiku/rag/app.py
View file

@ -81,13 +81,10 @@ class HaikuRAGApp:
raw = rows[0].get("settings") or "{}" raw = rows[0].get("settings") or "{}"
data = json.loads(raw) if isinstance(raw, str) else (raw or {}) data = json.loads(raw) if isinstance(raw, str) else (raw or {})
stored_version = str(data.get("version", stored_version)) stored_version = str(data.get("version", stored_version))
embed_provider = data.get("EMBEDDINGS_PROVIDER") embeddings = data.get("embeddings", {})
embed_model = data.get("EMBEDDINGS_MODEL") embed_provider = embeddings.get("provider")
vector_dim = ( embed_model = embeddings.get("model")
int(data.get("EMBEDDINGS_VECTOR_DIM")) # pyright: ignore[reportArgumentType] vector_dim = embeddings.get("vector_dim")
if data.get("EMBEDDINGS_VECTOR_DIM") is not None
else None
)
num_docs = 0 num_docs = 0
if "documents" in table_names: if "documents" in table_names:

8
tests/test_info.py
View file

@ -40,9 +40,11 @@ async def test_app_info_outputs_and_read_only(temp_db_path, capsys):
settings=json.dumps( settings=json.dumps(
{ {
"version": "1.2.3", "version": "1.2.3",
"EMBEDDINGS_PROVIDER": "openai", "embeddings": {
"EMBEDDINGS_MODEL": "text-embedding-3-small", "provider": "openai",
"EMBEDDINGS_VECTOR_DIM": 3, "model": "text-embedding-3-small",
"vector_dim": 3,
},
} }
), ),
) )