Skip to content

Commit 9ad5b1a

Browse files
Fixed pgvector class naming
1 parent a9a7488 commit 9ad5b1a

File tree

5 files changed

+11
-10
lines changed

5 files changed

+11
-10
lines changed

engine/clients/pgvector/__init__.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
1-
from engine.clients.pgvector.configure import PgVectorConfigurator
2-
from engine.clients.pgvector.search import PgVectorSearcher
3-
from engine.clients.pgvector.upload import PgVectorUploader
1+
from engine.clients.pgvector.configure import PgvectorConfigurator
2+
from engine.clients.pgvector.search import PgvectorSearcher
3+
from engine.clients.pgvector.upload import PgvectorUploader
44

55
__all__ = [
6-
"PgVectorConfigurator",
7-
"PgVectorSearcher",
8-
"PgVectorUploader",
6+
"PgvectorConfigurator",
7+
"PgvectorSearcher",
8+
"PgvectorUploader",
99
]

engine/clients/pgvector/config.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import os
22

3-
PGVECTOR_PORT = int(os.getenv("PGVECTOR_PORT", 9200))
3+
PGVECTOR_PORT = int(os.getenv("PGVECTOR_PORT", 5432))
44
PGVECTOR_DB = os.getenv("PGVECTOR_DB", "postgres")
55
PGVECTOR_USER = os.getenv("PGVECTOR_USER", "postgres")
66
PGVECTOR_PASSWORD = os.getenv("PGVECTOR_PASSWORD", "passwd")
@@ -9,6 +9,7 @@
99
def get_db_config(host, connection_params):
1010
return {
1111
"host": host or "localhost",
12+
"port": PGVECTOR_PORT,
1213
"dbname": PGVECTOR_DB,
1314
"user": PGVECTOR_USER,
1415
"password": PGVECTOR_PASSWORD,

engine/clients/pgvector/configure.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from engine.clients.pgvector.config import get_db_config
99

1010

11-
class PgVectorConfigurator(BaseConfigurator):
11+
class PgvectorConfigurator(BaseConfigurator):
1212
DISTANCE_MAPPING = {
1313
Distance.L2: "vector_l2_ops",
1414
Distance.COSINE: "vector_cosine_ops",

engine/clients/pgvector/search.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
from engine.clients.pgvector.parser import PgVectorConditionParser
1212

1313

14-
class PgVectorSearcher(BaseSearcher):
14+
class PgvectorSearcher(BaseSearcher):
1515
conn = None
1616
cur = None
1717
distance = None

engine/clients/pgvector/upload.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from engine.clients.pgvector.config import get_db_config
99

1010

11-
class PgVectorUploader(BaseUploader):
11+
class PgvectorUploader(BaseUploader):
1212
conn = None
1313
cur = None
1414
upload_params = {}

0 commit comments

Comments
 (0)