diff --git a/docs/src/reranking/cohere.md b/docs/src/reranking/cohere.md index b4924947..01f564de 100644 --- a/docs/src/reranking/cohere.md +++ b/docs/src/reranking/cohere.md @@ -25,18 +25,19 @@ data = [ {"text": "hello world"}, {"text": "goodbye world"} ] -tbl = db.create_table("test", data) -reranker = CohereReranker(api_key="your_api_key") +tbl = db.create_table("test", schema=Schema, mode="overwrite") +tbl.add(data) +reranker = CohereReranker(api_key="key") # Run vector search with a reranker -result = tbl.query("hello").rerank(reranker).to_list() +result = tbl.search("hello").rerank(reranker=reranker).to_list() # Run FTS search with a reranker -result = tbl.query("hello", query_type="fts").rerank(reranker).to_list() +result = tbl.search("hello", query_type="fts").rerank(reranker=reranker).to_list() # Run hybrid search with a reranker -tbl.create_fts_index("text") -result = tbl.query("hello", query_type="hybrid").rerank(reranker).to_list() +tbl.create_fts_index("text", replace=True) +result = tbl.search("hello", query_type="hybrid").rerank(reranker=reranker).to_list() ``` diff --git a/docs/src/reranking/colbert.md b/docs/src/reranking/colbert.md index d9a26e4b..a382c494 100644 --- a/docs/src/reranking/colbert.md +++ b/docs/src/reranking/colbert.md @@ -10,7 +10,7 @@ import numpy import lancedb from lancedb.embeddings import get_registry from lancedb.pydantic import LanceModel, Vector -from lancedb.rerankers import CrossEncoderReranker +from lancedb.rerankers import ColbertReranker embedder = get_registry().get("sentence-transformers").create() db = lancedb.connect("~/.lancedb") @@ -23,18 +23,19 @@ data = [ {"text": "hello world"}, {"text": "goodbye world"} ] -tbl = db.create_table("test", data) -reranker = CrossEncoderReranker() +tbl = db.create_table("test", schema=Schema, mode="overwrite") +tbl.add(data) +reranker = ColbertReranker() # Run vector search with a reranker -result = tbl.query("hello").rerank(reranker).to_list() +result = tbl.search("hello").rerank(reranker=reranker).to_list() # Run FTS search with a reranker -result = tbl.query("hello", query_type="fts").rerank(reranker).to_list() +result = tbl.search("hello", query_type="fts").rerank(reranker=reranker).to_list() # Run hybrid search with a reranker -tbl.create_fts_index("text") -result = tbl.query("hello", query_type="hybrid").rerank(reranker).to_list() +tbl.create_fts_index("text", replace=True) +result = tbl.search("hello", query_type="hybrid").rerank(reranker=reranker).to_list() ``` diff --git a/docs/src/reranking/cross_encoder.md b/docs/src/reranking/cross_encoder.md index b9692166..4ba97b57 100644 --- a/docs/src/reranking/cross_encoder.md +++ b/docs/src/reranking/cross_encoder.md @@ -23,18 +23,19 @@ data = [ {"text": "hello world"}, {"text": "goodbye world"} ] -tbl = db.create_table("test", data) -reranker = CrossEncoderReranker(api_key="your_api_key") +tbl = db.create_table("test", schema=Schema, mode="overwrite") +tbl.add(data) +reranker = CrossEncoderReranker() # Run vector search with a reranker -result = tbl.query("hello").rerank(reranker).to_list() +result = tbl.search("hello").rerank(reranker=reranker).to_list() # Run FTS search with a reranker -result = tbl.query("hello", query_type="fts").rerank(reranker).to_list() +result = tbl.search("hello", query_type="fts").rerank(reranker=reranker).to_list() # Run hybrid search with a reranker -tbl.create_fts_index("text") -result = tbl.query("hello", query_type="hybrid").rerank(reranker).to_list() +tbl.create_fts_index("text", replace=True) +result = tbl.search("hello", query_type="hybrid").rerank(reranker=reranker).to_list() ``` diff --git a/docs/src/reranking/linear_combination.md b/docs/src/reranking/linear_combination.md index 9829cd71..aa3ff242 100644 --- a/docs/src/reranking/linear_combination.md +++ b/docs/src/reranking/linear_combination.md @@ -24,12 +24,13 @@ data = [ {"text": "hello world"}, {"text": "goodbye world"} ] -tbl = db.create_table("test", data) +tbl = db.create_table("test", schema=Schema, mode="overwrite") +tbl.add(data) reranker = LinearCombinationReranker() # Run hybrid search with a reranker -tbl.create_fts_index("text") -result = tbl.query("hello", query_type="hybrid").rerank(reranker).to_list() +tbl.create_fts_index("text", replace=True) +result = tbl.search("hello", query_type="hybrid").rerank(reranker=reranker).to_list() ``` diff --git a/docs/src/reranking/openai.md b/docs/src/reranking/openai.md index c9535f57..c7940b23 100644 --- a/docs/src/reranking/openai.md +++ b/docs/src/reranking/openai.md @@ -25,18 +25,19 @@ data = [ {"text": "hello world"}, {"text": "goodbye world"} ] -tbl = db.create_table("test", data) +tbl = db.create_table("test", schema=Schema, mode="overwrite") +tbl.add(data) reranker = OpenaiReranker() # Run vector search with a reranker -result = tbl.query("hello").rerank(reranker).to_list() +result = tbl.search("hello").rerank(reranker=reranker).to_list() # Run FTS search with a reranker -result = tbl.query("hello", query_type="fts").rerank(reranker).to_list() +result = tbl.search("hello", query_type="fts").rerank(reranker=reranker).to_list() # Run hybrid search with a reranker -tbl.create_fts_index("text") -result = tbl.query("hello", query_type="hybrid").rerank(reranker).to_list() +tbl.create_fts_index("text", replace=True) +result = tbl.search("hello", query_type="hybrid").rerank(reranker=reranker).to_list() ```