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