Léo Bourrel commited on
Commit
6ddc1e6
·
1 Parent(s): 279f3c6

feat: move article to models folder

Browse files
Files changed (2) hide show
  1. models/article.py +17 -0
  2. vector_store.py +1 -1
models/article.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import sqlalchemy
2
+ from pgvector.sqlalchemy import Vector
3
+ from sqlalchemy.orm import declarative_base
4
+
5
+ Base = declarative_base() # type: Any
6
+
7
+
8
+ class Article(Base):
9
+ """Embedding store."""
10
+
11
+ __tablename__ = "article"
12
+
13
+ id = sqlalchemy.Column(sqlalchemy.Integer, primary_key=True, nullable=False)
14
+ title = sqlalchemy.Column(sqlalchemy.String, nullable=True)
15
+ abstract = sqlalchemy.Column(sqlalchemy.String, nullable=True)
16
+ embedding: Vector = sqlalchemy.Column("abstract_embedding", Vector(None))
17
+ doi = sqlalchemy.Column(sqlalchemy.String, nullable=True)
vector_store.py CHANGED
@@ -13,7 +13,7 @@ from langchain.vectorstores.base import VectorStore
13
  from sqlalchemy import delete, text
14
  from sqlalchemy.orm import Session
15
 
16
- from model import Article
17
  from models.distance import DistanceStrategy, distance_strategy_limit
18
  from utils import str_to_list
19
 
 
13
  from sqlalchemy import delete, text
14
  from sqlalchemy.orm import Session
15
 
16
+ from models.article import Article
17
  from models.distance import DistanceStrategy, distance_strategy_limit
18
  from utils import str_to_list
19