File tree Expand file tree Collapse file tree 5 files changed +11
-10
lines changed Expand file tree Collapse file tree 5 files changed +11
-10
lines changed Original file line number Diff line number Diff line change 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
4
4
5
5
__all__ = [
6
- "PgVectorConfigurator " ,
7
- "PgVectorSearcher " ,
8
- "PgVectorUploader " ,
6
+ "PgvectorConfigurator " ,
7
+ "PgvectorSearcher " ,
8
+ "PgvectorUploader " ,
9
9
]
Original file line number Diff line number Diff line change 1
1
import os
2
2
3
- PGVECTOR_PORT = int (os .getenv ("PGVECTOR_PORT" , 9200 ))
3
+ PGVECTOR_PORT = int (os .getenv ("PGVECTOR_PORT" , 5432 ))
4
4
PGVECTOR_DB = os .getenv ("PGVECTOR_DB" , "postgres" )
5
5
PGVECTOR_USER = os .getenv ("PGVECTOR_USER" , "postgres" )
6
6
PGVECTOR_PASSWORD = os .getenv ("PGVECTOR_PASSWORD" , "passwd" )
9
9
def get_db_config (host , connection_params ):
10
10
return {
11
11
"host" : host or "localhost" ,
12
+ "port" : PGVECTOR_PORT ,
12
13
"dbname" : PGVECTOR_DB ,
13
14
"user" : PGVECTOR_USER ,
14
15
"password" : PGVECTOR_PASSWORD ,
Original file line number Diff line number Diff line change 8
8
from engine .clients .pgvector .config import get_db_config
9
9
10
10
11
- class PgVectorConfigurator (BaseConfigurator ):
11
+ class PgvectorConfigurator (BaseConfigurator ):
12
12
DISTANCE_MAPPING = {
13
13
Distance .L2 : "vector_l2_ops" ,
14
14
Distance .COSINE : "vector_cosine_ops" ,
Original file line number Diff line number Diff line change 11
11
from engine .clients .pgvector .parser import PgVectorConditionParser
12
12
13
13
14
- class PgVectorSearcher (BaseSearcher ):
14
+ class PgvectorSearcher (BaseSearcher ):
15
15
conn = None
16
16
cur = None
17
17
distance = None
Original file line number Diff line number Diff line change 8
8
from engine .clients .pgvector .config import get_db_config
9
9
10
10
11
- class PgVectorUploader (BaseUploader ):
11
+ class PgvectorUploader (BaseUploader ):
12
12
conn = None
13
13
cur = None
14
14
upload_params = {}
You can’t perform that action at this time.
0 commit comments