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

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

View file

@ -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]]: