added delete_dataset method (#1051)

### What problem does this PR solve? Added delete_dataset method and test for it. ### Type of change - [x] New Feature (non-breaking change which adds functionality)
infiniflow · Jun 4, 2024 · 4fba042 · 4fba042
1 parent c74d4d6
commit 4fba042
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/sdk/python/ragflow/ragflow.py b/sdk/python/ragflow/ragflow.py
@@ -25,3 +25,5 @@ def __init__(self, user_key, base_url):
     def create_dataset(self, name):
         return name
 
+    def delete_dataset(self, name):
+        return name
diff --git a/sdk/python/test/test_basic.py b/sdk/python/test/test_basic.py
@@ -10,3 +10,6 @@ def test_version(self):
 
     def test_create_dataset(self):
         assert ragflow.ragflow.RAGFLow('123', 'url').create_dataset('abc') == 'abc'
+
+    def test_delete_dataset(self):
+        assert ragflow.ragflow.RAGFLow('123', 'url').delete_dataset('abc') == 'abc'