added a index for the VEctor DB

This commit is contained in:
2026-06-13 20:14:20 -04:00
parent 74e4c2e921
commit 07a9adfdd5
3 changed files with 81 additions and 3 deletions
+17 -1
View File
@@ -38,7 +38,14 @@ from python.ebook_search.search import (
search_ebooks,
)
from python.ebook_search.timing import RuntimeStep
from python.orm.richie import EbookChapter, EbookChunk, EbookEmbeddingModel, EbookSource, RichieBase
from python.orm.richie import (
EbookChapter,
EbookChunk,
EbookChunkEmbedding1024,
EbookEmbeddingModel,
EbookSource,
RichieBase,
)
def test_chunk_text_uses_overlap() -> None:
@@ -464,6 +471,15 @@ def test_ensure_embedding_models_registers_service_names() -> None:
]
def test_1024_embedding_table_has_cosine_hnsw_index() -> None:
indexes = {index.name: index for index in EbookChunkEmbedding1024.__table__.indexes}
index = indexes["ix_ebook_chunk_embedding_1024_embedding_cosine"]
assert [column.name for column in index.columns] == ["embedding"]
assert index.dialect_options["postgresql"]["using"] == "hnsw"
assert index.dialect_options["postgresql"]["ops"] == {"embedding": "vector_cosine_ops"}
def test_embedding_model_aliases_normalize_to_provider_names() -> None:
assert normalize_embedding_model() == "qwen3-embedding-0.6b"