From 14168db79563b188ece9b9916ca1ea5fea91ae59 Mon Sep 17 00:00:00 2001 From: Yiorgis Gozadinos Date: Fri, 26 Sep 2025 13:52:06 +0300 Subject: [PATCH] Fix embedding client test --- tests/test_client.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/test_client.py b/tests/test_client.py index 44ec66c9..338075b5 100644 --- a/tests/test_client.py +++ b/tests/test_client.py @@ -7,6 +7,7 @@ import pytest from datasets import Dataset from haiku.rag.client import HaikuRAG +from haiku.rag.config import Config from haiku.rag.store.models.chunk import Chunk @@ -504,7 +505,7 @@ async def test_client_create_document_with_custom_chunks(temp_db_path): Chunk( content="This is the second chunk", metadata={"custom": "metadata2"}, - embedding=[0.1] * 1024, + embedding=[0.1] * Config.EMBEDDINGS_VECTOR_DIM, order=1, ), # With embedding Chunk(