update LanceDB integration (run-llama#16057)

raspawar · Oct 7, 2024 · bcf5eae · bcf5eae
1 parent ef6bf09
commit bcf5eae
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/...vector_stores/llama-index-vector-stores-lancedb/llama_index/vector_stores/lancedb/base.py b/...vector_stores/llama-index-vector-stores-lancedb/llama_index/vector_stores/lancedb/base.py
@@ -482,14 +482,20 @@ def query(
             else:
                 raise ValueError(f"Invalid query type: {query_type}")
 
-        lance_query = (
-            self._table.search(
+        if query_type == "hybrid":
+            lance_query = (
+                self._table.search(
+                    vector_column_name=self.vector_column_name, query_type="hybrid"
+                )
+                .vector(query.query_embedding)
+                .text(query.query_str)
+            )
+        else:
+            lance_query = self._table.search(
                 query=_query,
                 vector_column_name=self.vector_column_name,
             )
-            .limit(query.similarity_top_k * self.overfetch_factor)
-            .where(where)
-        )
+        lance_query.limit(query.similarity_top_k * self.overfetch_factor).where(where)
 
         if query_type != "fts":
             lance_query.nprobes(self.nprobes)

diff --git a/llama-index-integrations/vector_stores/llama-index-vector-stores-lancedb/pyproject.toml b/llama-index-integrations/vector_stores/llama-index-vector-stores-lancedb/pyproject.toml
@@ -31,7 +31,7 @@ version = "0.2.3"
 
 [tool.poetry.dependencies]
 python = ">=3.9,<4.0"
-lancedb = ">=0.8.0,<0.13.0"
+lancedb = ">=0.13.0"
 tantivy = "*"
 llama-index-core = "^0.11.0"