leo-bourrel commited on
Commit
722a601
·
1 Parent(s): e4efa7a

fix: do not create vector extension

Browse files
Files changed (1) hide show
  1. 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)