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>

bindings/python/examples/psycopg_example.py

@@ -6,7 +6,7 @@ import psycopg
 from pgvecto_rs.psycopg import register_vector
 
 URL = "postgresql://{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"),
@@ -18,7 +18,7 @@ with psycopg.connect(URL) as conn:
     conn.execute("CREATE EXTENSION IF NOT EXISTS vectors;")
     register_vector(conn)
     conn.execute(
-        "CREATE TABLE documents (id SERIAL PRIMARY KEY, text TEXT NOT NULL, embedding vector(3) NOT NULL);"
+        "CREATE TABLE documents (id SERIAL PRIMARY KEY, text TEXT NOT NULL, embedding vector(3) NOT NULL);",
     )
     conn.commit()
     try:
@@ -39,7 +39,8 @@ with psycopg.connect(URL) as conn:
 
         # Select the row "hello pgvecto.rs"
         cur = conn.execute(
-            "SELECT * FROM documents WHERE text = %s;", ("hello pgvecto.rs",)
+            "SELECT * FROM documents WHERE text = %s;",
+            ("hello pgvecto.rs",),
         )
         target = cur.fetchone()[2]
 

bindings/python/examples/sdk_example.py

@@ -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)

bindings/python/examples/sqlalchemy_example.py

@@ -7,7 +7,7 @@ from sqlalchemy.orm import DeclarativeBase, Mapped, Session, mapped_column
 from pgvecto_rs.sqlalchemy import Vector
 
 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"),
@@ -53,7 +53,7 @@ with Session(engine) as session:
     stmt = select(
         Document.text,
         Document.embedding.squared_euclidean_distance(target.embedding).label(
-            "distance"
+            "distance",
         ),
     ).order_by("distance")
     for doc in session.execute(stmt):