Spaces:
Sleeping
Sleeping
Léo Bourrel
commited on
Commit
·
6ddc1e6
1
Parent(s):
279f3c6
feat: move article to models folder
Browse files- models/article.py +17 -0
- 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
|
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 |
|