1
0
mirror of https://github.com/tensorchord/pgvecto.rs.git synced 2025-08-12 23:42:45 +03:00

feat: add more ruff rules (#138)

* feat: add more ruff rules

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

* chore: modified readme

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

* rename error class

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

---------

Signed-off-by: 盐粒 Yanli <mail@yanli.one>
This commit is contained in:
盐粒 Yanli
2023-11-17 17:47:28 +08:00
committed by GitHub
parent f8344dd039
commit f6e382d0fc
16 changed files with 138 additions and 86 deletions

View File

@@ -5,7 +5,7 @@ from openai import OpenAI
from pgvecto_rs.sdk import PGVectoRs, Record, filters
URL = "postgresql+psycopg://{username}:{password}@{host}:{port}/{db_name}".format(
port=os.getenv("DB_PORT", 5432),
port=os.getenv("DB_PORT", "5432"),
host=os.getenv("DB_HOST", "localhost"),
username=os.getenv("DB_USER", "postgres"),
password=os.getenv("DB_PASS", "mysecretpassword"),
@@ -43,7 +43,8 @@ try:
# Query (With a filter from the filters module)
print("#################### First Query ####################")
for record, dis in client.search(
target, filter=filters.meta_contains({"src": "one"})
target,
filter=filters.meta_contains({"src": "one"}),
):
print(f"DISTANCE SCORE: {dis}")
print(record)
@@ -51,7 +52,8 @@ try:
# Another Query (Equivalent to the first one, but with a lambda filter written by hand)
print("#################### Second Query ####################")
for record, dis in client.search(
target, filter=lambda r: r.meta.contains({"src": "one"})
target,
filter=lambda r: r.meta.contains({"src": "one"}),
):
print(f"DISTANCE SCORE: {dis}")
print(record)