clean: imports

app.py

@@ -1,3 +1,5 @@
+import os
+
 import streamlit as st
 import streamlit.components.v1 as components
 from css import load_css

custom_pgvector.py

@@ -6,7 +6,6 @@ import enum
 import logging
 from functools import partial
 from typing import (
-    TYPE_CHECKING,
     Any,
     Callable,
     Dict,
@@ -30,9 +29,6 @@ from pgvector.sqlalchemy import Vector
 from sqlalchemy import delete
 from sqlalchemy.orm import Session, declarative_base, relationship
 
-if TYPE_CHECKING:
-    from langchain.vectorstores._pgvector_data_models import CollectionStore
-
 
 class DistanceStrategy(str, enum.Enum):
     """Enumerator of the Distance strategies."""
@@ -356,7 +352,7 @@ class CustomPGVector(VectorStore):
                     metadata={
                         "id": result.Article.id,
                         "title": result.Article.title,
-                        "doi": result.Article.doi
+                        "doi": result.Article.doi,
                     },
                 ),
                 result.distance if self.embedding_function is not None else None,