From 8ac954000a9a70979d839901fe76ba92dee86767 Mon Sep 17 00:00:00 2001
From: thomashacker <EdwardSchmuhl@web.de>
Date: Wed, 4 Sep 2024 09:55:40 +0200
Subject: [PATCH] Add reset capabilities

---
 CHANGELOG.md              |  6 ++++--
 goldenverba/server/cli.py | 16 ++++++++++++----
 2 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index c6fc923ff..3860d4ed4 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -31,14 +31,16 @@ All notable changes to this project will be documented in this file.
 - Added Suggestion View
 - Reworked Suggestion logic
 - Added VoyageAI
+- Added custom metadata
 - Added DocumentExplorer with
+
   - Content View
   - Chunk View
   - Vector View
     - Visualize vectors of chunks of one or multiple documents
     - PCA
-    - t-SNE
-    - UMAP
+
+-
 
 ## [1.0.3]
 
diff --git a/goldenverba/server/cli.py b/goldenverba/server/cli.py
index d1ee8f8c6..266c051e7 100644
--- a/goldenverba/server/cli.py
+++ b/goldenverba/server/cli.py
@@ -64,8 +64,13 @@ def start(port, host, prod, workers):
     default="",
     help="Deployment (Local, Weaviate, Docker)",
 )
+@click.option(
+    "--full_reset",
+    default=False,
+    help="Full reset (True, False)",
+)
 @cli.command()
-def reset(url, api_key, deployment):
+def reset(url, api_key, deployment, full_reset):
     """
     Run the FastAPI application.
     """
@@ -93,9 +98,12 @@ async def async_reset():
             else:
                 raise ValueError("Invalid deployment")
 
-        await manager.reset_rag_config(client)
-        await manager.reset_theme_config(client)
-        await manager.reset_user_config(client)
+        if not full_reset:
+            await manager.reset_rag_config(client)
+            await manager.reset_theme_config(client)
+            await manager.reset_user_config(client)
+        else:
+            await manager.weaviate_manager.delete_all(client)
 
         await client.close()