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

fix: use 'SearchType' for 'ChunkRepository.search'

Browse source
This commit is contained in:
Tres Seaver 2026-05-18 06:50:49 -04:00
parent 70da24b691
commit 5079205d3d
No known key found for this signature in database
GPG key ID: F2A968348913D1D8
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
haiku_rag_slim/haiku/rag/store/repositories/chunk.py
View file

@ -10,6 +10,7 @@ if TYPE_CHECKING:
from lancedb.index import FTS from lancedb.index import FTS
from lancedb.rerankers import RRFReranker 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.engine import Store, query_to_pydantic
from haiku.rag.store.models.chunk import Chunk from haiku.rag.store.models.chunk import Chunk
@ -220,7 +221,7 @@ class ChunkRepository:
self, self,
query: str = "", query: str = "",
limit: int = 5, limit: int = 5,
search_type: str = "hybrid", search_type: SearchType = "hybrid",
filter: str | None = None, filter: str | None = None,
query_vector: list[float] | None = None, query_vector: list[float] | None = None,
) -> list[tuple[Chunk, float]]: ) -> list[tuple[Chunk, float]]: