From f9b689497463ed5355f4f5ffd852f92abd5838dd Mon Sep 17 00:00:00 2001 From: Yiorgis Gozadinos Date: Wed, 29 Oct 2025 13:31:54 +0200 Subject: [PATCH] Fix info to not use old .env settings --- src/haiku/rag/app.py | 11 ++++------- tests/test_info.py | 8 +++++--- 2 files changed, 9 insertions(+), 10 deletions(-) diff --git a/src/haiku/rag/app.py b/src/haiku/rag/app.py index e7609f8d..51e63751 100644 --- a/src/haiku/rag/app.py +++ b/src/haiku/rag/app.py @@ -81,13 +81,10 @@ class HaikuRAGApp: raw = rows[0].get("settings") or "{}" data = json.loads(raw) if isinstance(raw, str) else (raw or {}) stored_version = str(data.get("version", stored_version)) - embed_provider = data.get("EMBEDDINGS_PROVIDER") - embed_model = data.get("EMBEDDINGS_MODEL") - vector_dim = ( - int(data.get("EMBEDDINGS_VECTOR_DIM")) # pyright: ignore[reportArgumentType] - if data.get("EMBEDDINGS_VECTOR_DIM") is not None - else None - ) + embeddings = data.get("embeddings", {}) + embed_provider = embeddings.get("provider") + embed_model = embeddings.get("model") + vector_dim = embeddings.get("vector_dim") num_docs = 0 if "documents" in table_names: diff --git a/tests/test_info.py b/tests/test_info.py index 4a408420..dbc9669e 100644 --- a/tests/test_info.py +++ b/tests/test_info.py @@ -40,9 +40,11 @@ async def test_app_info_outputs_and_read_only(temp_db_path, capsys): settings=json.dumps( { "version": "1.2.3", - "EMBEDDINGS_PROVIDER": "openai", - "EMBEDDINGS_MODEL": "text-embedding-3-small", - "EMBEDDINGS_VECTOR_DIM": 3, + "embeddings": { + "provider": "openai", + "model": "text-embedding-3-small", + "vector_dim": 3, + }, } ), )