rust/lancedb
1
0
Fork 0
mirror of https://github.com/lancedb/lancedb.git synced 2026-01-08 12:52:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Qian/make vector col optional (#950)

Browse Source 
remote SDK tests were completed through lancedb_integtest
This commit is contained in:
QianZhu
2024-02-12 16:35:44 -08:00
committed by GitHub
parent 0c21f91c16
commit 1b990983b3
8 changed files with 154 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
python/tests/test_embeddings_slow.py
View File

@@ -69,10 +69,14 @@ def test_basic_text_embeddings(alias, tmp_path):
)
query = "greetings"
actual = table.search(query).limit(1).to_pydantic(Words)[0]
actual = (
table.search(query, vector_column_name="vector").limit(1).to_pydantic(Words)[0]
)
vec = func.compute_query_embeddings(query)[0]
expected = table.search(vec).limit(1).to_pydantic(Words)[0]
expected = (
table.search(vec, vector_column_name="vector").limit(1).to_pydantic(Words)[0]
)
assert actual.text == expected.text
assert actual.text == "hello world"
assert not np.allclose(actual.vector, actual.vector2)
@@ -116,7 +120,11 @@ def test_openclip(tmp_path):
)
# text search
actual = table.search("man's best friend").limit(1).to_pydantic(Images)[0]
actual = (
table.search("man's best friend", vector_column_name="vector")
.limit(1)
.to_pydantic(Images)[0]
)
assert actual.label == "dog"
frombytes = (
table.search("man's best friend", vector_column_name="vec_from_bytes")
@@ -130,7 +138,11 @@ def test_openclip(tmp_path):
query_image_uri = "http://farm1.staticflickr.com/200/467715466_ed4a31801f_z.jpg"
image_bytes = requests.get(query_image_uri).content
query_image = Image.open(io.BytesIO(image_bytes))
actual = table.search(query_image).limit(1).to_pydantic(Images)[0]
actual = (
table.search(query_image, vector_column_name="vector")
.limit(1)
.to_pydantic(Images)[0]
)
assert actual.label == "dog"
other = (
table.search(query_image, vector_column_name="vec_from_bytes")