diff --git a/sdk/python/ragflow/ragflow.py b/sdk/python/ragflow/ragflow.py index a3aabc5f631..4aefaf3b6c7 100644 --- 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 \ No newline at end of file diff --git a/sdk/python/test/test_basic.py b/sdk/python/test/test_basic.py index f784279f03a..9bae8e4cf71 100644 --- 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' \ No newline at end of file