clean: isort

app.py

@@ -1,18 +1,19 @@
-import os
 import json
+import os
 
+import sqlalchemy
 import streamlit as st
 import streamlit.components.v1 as components
-from chat_history import insert_chat_history
-from css import load_css
-from custom_pgvector import CustomPGVector
 from langchain import OpenAI
 from langchain.callbacks import get_openai_callback
 from langchain.chains import ConversationalRetrievalChain
 from langchain.chains.conversation.memory import ConversationBufferMemory
 from langchain.embeddings import GPT4AllEmbeddings
+
+from chat_history import insert_chat_history
+from css import load_css
+from custom_pgvector import CustomPGVector
 from message import Message
-import sqlalchemy
 
 CONNECTION_STRING = "postgresql+psycopg2://localhost/sorbobot"
 

custom_pgvector.py

@@ -1,24 +1,16 @@
 from __future__ import annotations
-import json
-import pandas as pd
+
 import asyncio
 import contextlib
 import enum
+import json
 import logging
 from functools import partial
-from typing import (
-    Any,
-    Callable,
-    Dict,
-    Generator,
-    Iterable,
-    List,
-    Optional,
-    Tuple,
-    Type,
-)
+from typing import (Any, Callable, Dict, Generator, Iterable, List, Optional,
+                    Tuple, Type)
 
 import numpy as np
+import pandas as pd
 import sqlalchemy
 from langchain.docstore.document import Document
 from langchain.schema.embeddings import Embeddings
@@ -26,9 +18,8 @@ from langchain.utils import get_from_dict_or_env
 from langchain.vectorstores.base import VectorStore
 from langchain.vectorstores.pgvector import BaseModel
 from pgvector.sqlalchemy import Vector
-from sqlalchemy import delete
+from sqlalchemy import delete, text
 from sqlalchemy.orm import Session, declarative_base, relationship
-from sqlalchemy import text
 
 
 class DistanceStrategy(str, enum.Enum):

message.py

@@ -1,5 +1,5 @@
 from dataclasses import dataclass
-from typing import Literal, Optional, List
+from typing import List, Literal, Optional
 
 from langchain.schema import Document