clean: ruff

chat_history.py

@@ -35,7 +35,10 @@ def insert_chat_history_articles(
     with Session(conn) as conn:
         conn.execute(
             text(
-                "INSERT INTO chat_history_articles (chat_history_id, article_id) VALUES (:chat_history_id, :article_id) ON CONFLICT DO NOTHING;"
+                """
+                    INSERT INTO chat_history_articles (chat_history_id, article_id)
+                    VALUES (:chat_history_id, :article_id) ON CONFLICT DO NOTHING;
+                """
             ),
             [
                 {

custom_pgvector.py

@@ -207,13 +207,10 @@ class CustomPGVector(VectorStore):
             metadatas = [{} for _ in texts]
 
         with Session(self._conn) as session:
-            # collection = self.get_collection(session)
-            # if not collection:
-            #     raise ValueError("Collection not found")
-            for text, metadata, embedding, id in zip(texts, metadatas, embeddings, ids):
+            for txt, metadata, embedding, id in zip(texts, metadatas, embeddings, ids):
                 embedding_store = self.EmbeddingStore(
                     embedding=embedding,
-                    document=text,
+                    document=txt,
                     cmetadata=metadata,
                     custom_id=id,
                 )
@@ -405,7 +402,7 @@ class CustomPGVector(VectorStore):
 
     @classmethod
     def from_texts(
-        cls: Type[PGVector],
+        cls: Type[CustomPGVector],
         texts: List[str],
         embedding: Embeddings,
         metadatas: Optional[List[dict]] = None,
@@ -414,7 +411,7 @@ class CustomPGVector(VectorStore):
         ids: Optional[List[str]] = None,
         pre_delete_collection: bool = False,
         **kwargs: Any,
-    ) -> PGVector:
+    ) -> CustomPGVector:
         """
         Return VectorStore initialized from texts and embeddings.
         Postgres connection string is required
@@ -446,7 +443,7 @@ class CustomPGVector(VectorStore):
         ids: Optional[List[str]] = None,
         pre_delete_collection: bool = False,
         **kwargs: Any,
-    ) -> PGVector:
+    ) -> CustomPGVector:
         """Construct PGVector wrapper from raw documents and pre-
         generated embeddings.
 
@@ -482,13 +479,13 @@ class CustomPGVector(VectorStore):
 
     @classmethod
     def from_existing_index(
-        cls: Type[PGVector],
+        cls: Type[CustomPGVector],
         embedding: Embeddings,
         table_name: str = _LANGCHAIN_DEFAULT_COLLECTION_NAME,
         distance_strategy: DistanceStrategy = DEFAULT_DISTANCE_STRATEGY,
         pre_delete_collection: bool = False,
         **kwargs: Any,
-    ) -> PGVector:
+    ) -> CustomPGVector:
         """
         Get intsance of an existing PGVector store.This method will
         return the instance of the store without inserting any new