Spaces:
Sleeping
Sleeping
Commit
·
722a601
1
Parent(s):
e4efa7a
fix: do not create vector extension
Browse files- custom_pgvector.py +0 -10
custom_pgvector.py
CHANGED
@@ -114,7 +114,6 @@ class CustomPGVector(VectorStore):
|
|
114 |
Initialize the store.
|
115 |
"""
|
116 |
# self._conn = self.connect()
|
117 |
-
self.create_vector_extension()
|
118 |
|
119 |
self.EmbeddingStore = Article
|
120 |
|
@@ -122,15 +121,6 @@ class CustomPGVector(VectorStore):
|
|
122 |
def embeddings(self) -> Embeddings:
|
123 |
return self.embedding_function
|
124 |
|
125 |
-
def create_vector_extension(self) -> None:
|
126 |
-
try:
|
127 |
-
with Session(self._conn) as session:
|
128 |
-
statement = sqlalchemy.text("CREATE EXTENSION IF NOT EXISTS vector")
|
129 |
-
session.execute(statement)
|
130 |
-
session.commit()
|
131 |
-
except Exception as e:
|
132 |
-
self.logger.exception(e)
|
133 |
-
|
134 |
def drop_tables(self) -> None:
|
135 |
with self._conn.begin():
|
136 |
Base.metadata.drop_all(self._conn)
|
|
|
114 |
Initialize the store.
|
115 |
"""
|
116 |
# self._conn = self.connect()
|
|
|
117 |
|
118 |
self.EmbeddingStore = Article
|
119 |
|
|
|
121 |
def embeddings(self) -> Embeddings:
|
122 |
return self.embedding_function
|
123 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
124 |
def drop_tables(self) -> None:
|
125 |
with self._conn.begin():
|
126 |
Base.metadata.drop_all(self._conn)
|