Clean Versions

ChromaDBFlow.py

@@ -37,7 +37,14 @@ class ChromaDBFlow(AtomicFlow):
     default parameters of LiteLLMBackend (see aiflows.backends.LiteLLMBackend). Except for the following parameter whose default value is overwritten:
         - `api_infos` (List[Dict[str, Any]]): The list of api infos. Default: No default value, this parameter is required.
         - `model_name` (str): The name of the model. Default: "". In the current implementation, this parameter is not used.
-    - `n_results` (int): The number of results to retrieve when reading from the database. Default: 5
+    - `similarity_search_kwargs` (Dict[str, Any]): The parameters to pass to the similarity search method of the ChromaDB. Default:
+        - `k` (int): The number of documents to retrieve. Default: 2
+        - `filter` (str): The filter to apply to the documents. Default: null
+    - `paths_to_data` (List[str]): The paths to the data to store in the database at instantiation. Default: []
+    - `chunk_size` (int): The size of the chunks to split the documents into. Default: 700
+    - `chunk_overlap` (int): The overlap between the chunks. Default: 0
+    - `persist_directory` (str): The directory to persist the database. Default: "./demo_db_dir"
+    
     - Other parameters are inherited from the default configuration of AtomicFlow (see AtomicFlow)
     
     *Input Interface*:

ChromaDBFlow.yaml

@@ -13,8 +13,7 @@ input_interface:
 - content
 output_interface:
 - retrieved
-paths_to_data:
-- ./data/paul_graham_essay.txt
-chunk_size: 300
+paths_to_data: []
+chunk_size: 700
 chunk_overlap: 0
 persist_directory: ./demo_db_dir