leo-bourrel commited on
Commit
6e27dd8
·
1 Parent(s): c6cacc4

clean: remove useless method

Browse files
Files changed (1) hide show
  1. vector_store.py +1 -5
vector_store.py CHANGED
@@ -15,7 +15,7 @@ from sqlalchemy.orm import Session
15
 
16
  from utils import str_to_list
17
  from models.distance import DistanceStrategy
18
- from model import Base, Article
19
 
20
  DEFAULT_DISTANCE_STRATEGY = DistanceStrategy.COSINE
21
 
@@ -97,10 +97,6 @@ class CustomVectorStore(VectorStore):
97
  def embeddings(self) -> Embeddings:
98
  return self.embedding_function
99
 
100
- def drop_tables(self) -> None:
101
- with self._conn.begin():
102
- Base.metadata.drop_all(self._conn)
103
-
104
  @contextlib.contextmanager
105
  def _make_session(self) -> Generator[Session, None, None]:
106
  """Create a context manager for the session, bind to _conn string."""
 
15
 
16
  from utils import str_to_list
17
  from models.distance import DistanceStrategy
18
+ from model import Article
19
 
20
  DEFAULT_DISTANCE_STRATEGY = DistanceStrategy.COSINE
21
 
 
97
  def embeddings(self) -> Embeddings:
98
  return self.embedding_function
99
 
 
 
 
 
100
  @contextlib.contextmanager
101
  def _make_session(self) -> Generator[Session, None, None]:
102
  """Create a context manager for the session, bind to _conn string."""