Merge pull request #123 from mindflowai/patch

Patch broken release

Author: steegecs

mindflow/__init__.py

@@ -1 +1 @@
-__version__ = "0.5.0"
+__version__ = "0.5.1"

mindflow/core/__init__.py

@@ -12,9 +12,18 @@
 
 def run_delete(document_paths: List[str]) -> str:
     """Delete documents from MindFlow index."""
-    document_references: List[DocumentReference] = resolve_paths_to_document_references(document_paths)
-
-    document_ids = [document_id for document_id in [get_document_id(document_reference.path, document_reference.document_type) for document_reference in document_references] if document_id is not None]
+    document_references: List[DocumentReference] = resolve_paths_to_document_references(
+        document_paths
+    )
+
+    document_ids = [
+        document_id
+        for document_id in [
+            get_document_id(document_reference.path, document_reference.document_type)
+            for document_reference in document_references
+        ]
+        if document_id is not None
+    ]
 
     if not document_ids:
         return "No document IDs resolved. Nothing to delete."
@@ -30,4 +39,4 @@ def run_delete(document_paths: List[str]) -> str:
     Document.delete_bulk(document_ids)
     DocumentChunk.delete_bulk(document_chunk_ids)
 
-    return True, "Documents and associated chunks deleted successfully."
+    return "Documents and associated chunks deleted successfully."

mindflow/core/commands/delete.py

@@ -39,7 +39,7 @@ def run_index(document_paths: List[str]) -> str:
         indexable_documents := get_indexable_documents(
             document_references, completion_model
         )
-    ):  
+    ):
         return "No documents to index"
 
     print_total_size_of_documents(indexable_documents)

mindflow/core/commands/index.py

@@ -52,7 +52,9 @@ def run_query(document_paths: List[str], query: str) -> str:
             top_k=100,
         )
     ):
-        return "No index for requested hashes. Please generate index for passed content."
+        return (
+            "No index for requested hashes. Please generate index for passed content."
+        )
 
     document_selection_batch: List[Tuple[str, DocumentChunk]] = [
         (document_hash_to_path[document_chunk.id.split("_")[0]], document_chunk)

mindflow/core/commands/query.py

@@ -29,6 +29,6 @@ def get_all_unignored_git_files_from_path(path: Union[str, os.PathLike]) -> List
         )
         git_files = output.stdout.decode().strip().split("\n")
         return [os.path.abspath(os.path.join(os.fspath(path), f)) for f in git_files]
-    except GitError as error:
+    except subprocess.CalledProcessError as error:
         logging.debug("Failed extract git files with 'git ls-files': %s", error)
         return []