added a index for the VEctor DB
This commit is contained in:
@@ -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"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user