chore: reformat

umbertogriffo · Dec 7, 2024 · 2aeca30 · 2aeca30
1 parent 8e9784a
commit 2aeca30
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/chatbot/bot/memory/vector_database/chroma.py b/chatbot/bot/memory/vector_database/chroma.py
@@ -52,7 +52,8 @@ def __query_collection(
         where_document: dict[str, str] | None = None,
         **kwargs: Any,
     ):
-        """Query the chroma collection.
+        """
+        Query the chroma collection.
 
         Args:
             query_texts: List of query texts.
@@ -86,7 +87,8 @@ def add_texts(
         metadatas: list[dict] | None = None,
         ids: list[str] | None = None,
     ) -> list[str]:
-        """Run more texts through the embeddings and add to the vectorstore.
+        """
+        Run more texts through the embeddings and add to the vectorstore.
 
         Args:
             texts (Iterable[str]): Texts to add to the vectorstore.
@@ -249,7 +251,8 @@ def similarity_search_with_threshold(
         return retrieved_contents, sources
 
     def similarity_search(self, query: str, k: int = 4, filter: dict[str, str] | None = None) -> list[Document]:
-        """Run similarity search with Chroma.
+        """
+        Run similarity search with Chroma.
 
         Args:
             query (str): Query text to search for.