From 5079205d3d5e956968078e47b150d8174ad32033 Mon Sep 17 00:00:00 2001 From: Tres Seaver Date: Mon, 18 May 2026 06:50:49 -0400 Subject: [PATCH] fix: use 'SearchType' for 'ChunkRepository.search' --- haiku_rag_slim/haiku/rag/store/repositories/chunk.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/haiku_rag_slim/haiku/rag/store/repositories/chunk.py b/haiku_rag_slim/haiku/rag/store/repositories/chunk.py index f32121c8..91b94206 100644 --- a/haiku_rag_slim/haiku/rag/store/repositories/chunk.py +++ b/haiku_rag_slim/haiku/rag/store/repositories/chunk.py @@ -10,6 +10,7 @@ if TYPE_CHECKING: from lancedb.index import FTS from lancedb.rerankers import RRFReranker +from haiku.rag.client.models import SearchType from haiku.rag.store.engine import Store, query_to_pydantic from haiku.rag.store.models.chunk import Chunk @@ -220,7 +221,7 @@ class ChunkRepository: self, query: str = "", limit: int = 5, - search_type: str = "hybrid", + search_type: SearchType = "hybrid", filter: str | None = None, query_vector: list[float] | None = None, ) -> list[tuple[Chunk, float]]: