diff --git a/sdk/python/ragflow/ragflow.py b/sdk/python/ragflow/ragflow.py index a3aabc5f6..4aefaf3b6 100644 --- a/sdk/python/ragflow/ragflow.py +++ b/sdk/python/ragflow/ragflow.py @@ -25,3 +25,5 @@ class RAGFLow(ABC): def create_dataset(self, name): return name + def delete_dataset(self, name): + return name \ No newline at end of file diff --git a/sdk/python/test/test_basic.py b/sdk/python/test/test_basic.py index f784279f0..9bae8e4cf 100644 --- a/sdk/python/test/test_basic.py +++ b/sdk/python/test/test_basic.py @@ -10,3 +10,6 @@ class TestCase(TestSdk): 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' \ No newline at end of file