clean: isort

app.py

@@ -10,10 +10,10 @@ from langchain.embeddings import GPT4AllEmbeddings
 from langchain.llms import OpenAI
 
 from chat_history import insert_chat_history, insert_chat_history_articles
+from connection import connect
 from css import load_css
-from vector_store import CustomVectorStore
 from message import Message
-from connection import connect
+from vector_store import CustomVectorStore
 
 st.set_page_config(layout="wide")
 

connection.py

@@ -1,6 +1,5 @@
 import sqlalchemy
 
-
 CONNECTION_STRING = "postgresql+psycopg2://postgres@/sorbobot?host=0.0.0.0"
 
 

model.py

@@ -1,7 +1,6 @@
 import sqlalchemy
-
-from sqlalchemy.orm import declarative_base
 from pgvector.sqlalchemy import Vector
+from sqlalchemy.orm import declarative_base
 
 Base = declarative_base()  # type: Any
 

vector_store.py

@@ -13,9 +13,9 @@ from langchain.vectorstores.base import VectorStore
 from sqlalchemy import delete, text
 from sqlalchemy.orm import Session
 
-from utils import str_to_list
-from models.distance import DistanceStrategy
 from model import Article
+from models.distance import DistanceStrategy
+from utils import str_to_list
 
 DEFAULT_DISTANCE_STRATEGY = DistanceStrategy.COSINE