database/pgvecto.rs
1
0
Fork 0
mirror of https://github.com/tensorchord/pgvecto.rs.git synced 2025-08-01 06:46:52 +03:00
Code Activity

feat: enhance the __init__ of client (#164)

Browse Source 
* enhance the __init__ of client

Signed-off-by: 盐粒 Yanli <mail@yanli.one>

* bump version

Signed-off-by: 盐粒 Yanli <mail@yanli.one>

---------

Signed-off-by: 盐粒 Yanli <mail@yanli.one>
This commit is contained in:
盐粒 Yanli
2023-11-23 15:32:07 +08:00
committed by GitHub
parent eafb2f66f2
commit 20b6e0081f
4 changed files with 45 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
bindings/python/tests/test_sdk.py
View File

@ -32,19 +32,12 @@ class MockEmbedder:
@pytest.fixture(scope="module")
def client():
client = PGVectoRs(db_url=URL, collection_name="empty", dimension=3)
try:
records1 = [
Record.from_text(t, v, {"src": "src1"}) for t, v in MockTexts.items()
]
records2 = [
Record.from_text(t, v, {"src": "src2"}) for t, v in MockTexts.items()
]
client.insert(records1)
client.insert(records2)
yield client
finally:
client.drop()
client = PGVectoRs(db_url=URL, collection_name="empty", dimension=3, recreate=True)
records1 = [Record.from_text(t, v, {"src": "src1"}) for t, v in MockTexts.items()]
records2 = [Record.from_text(t, v, {"src": "src2"}) for t, v in MockTexts.items()]
client.insert(records1)
client.insert(records2)
return client
filter_src1 = filters.meta_contains({"src": "src1"})