update

userpj · Dec 31, 2024 · 57e070e · 57e070e
1 parent 1d222c7
commit 57e070e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/java/src/main/java/com/baidubce/appbuilder/console/dataset/Dataset.java b/java/src/main/java/com/baidubce/appbuilder/console/dataset/Dataset.java
@@ -36,6 +36,10 @@ public Dataset(String secretKey, String datasetId) {
         this.datasetId = datasetId;
     }
 
+    public void setDatasetId(String datasetId) {
+        this.datasetId = datasetId;
+    }
+
     /**
      * 创建数据集
      *

diff --git a/java/src/test/java/com/baidubce/appbuilder/DatasetTest.java b/java/src/test/java/com/baidubce/appbuilder/DatasetTest.java
@@ -29,6 +29,7 @@ public void testCreateDataset() throws IOException, AppBuilderServerException {
             assertNotNull(datasetId);
         } catch (Exception e) {
             datasetId = System.getenv("DATASET_ID_V3");
+            dataset.setDatasetId(datasetId);
         }
 
         String filePath = "src/test/java/com/baidubce/appbuilder/files/test.pdf";