chore: relock dependencies

superlinear-ai · Dec 15, 2024 · 7b05cb0 · 7b05cb0
1 parent d654872
commit 7b05cb0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/src/raglite/_database.py b/src/raglite/_database.py
@@ -368,11 +368,9 @@ def create_database_engine(config: RAGLiteConfig | None = None) -> Engine:
         with Session(engine) as session:
             metrics = {"cosine": "cosine", "dot": "ip", "euclidean": "l2", "l1": "l1", "l2": "l2"}
             session.execute(
-                text(
-                    """
-                    CREATE INDEX IF NOT EXISTS keyword_search_chunk_index ON chunk USING GIN (to_tsvector('simple', body));
-                    """
-                )
+                text("""
+                CREATE INDEX IF NOT EXISTS keyword_search_chunk_index ON chunk USING GIN (to_tsvector('simple', body));
+                """)
             )
             create_vector_index_sql = f"""
                 CREATE INDEX IF NOT EXISTS vector_search_chunk_index ON chunk_embedding
@@ -382,7 +380,7 @@ def create_database_engine(config: RAGLiteConfig | None = None) -> Engine:
                 );
                 SET hnsw.ef_search = {20 * 4 * 8};
             """
-            # Add iterative scan if version >= 0.8.0
+            # Enable iterative scan for pgvector v0.8.0 and up.
             pgvector_version = _pgvector_version(session)
             if pgvector_version and pgvector_version >= version.parse("0.8.0"):
                 create_vector_index_sql += f"\nSET hnsw.iterative_scan = {'relaxed_order' if config.reranker else 'strict_order'};"