fix: delete-file adapt to ws://

gptscript-ai · Oct 29, 2024 · 17ea088 · 17ea088
1 parent 7b089d1
commit 17ea088
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/delete-file.gpt b/delete-file.gpt
@@ -1,4 +1,6 @@
 Name: Knowledge File Deletion
 Description: Delete a file.
+Params: File: The file to delete
+Params: Dataset: The dataset to delete the file from
 
-#!${GPTSCRIPT_TOOL_DIR}/bin/gptscript-go-tool delete-file "${GPTSCRIPT_INPUT}"
+#!${GPTSCRIPT_TOOL_DIR}/bin/gptscript-go-tool delete-file --dataset "${DATASET}" "ws://${FILE}"
diff --git a/pkg/cmd/delete_file.go b/pkg/cmd/delete_file.go
@@ -37,7 +37,7 @@ func (s *ClientDeleteFile) Run(cmd *cobra.Command, args []string) error {
 		Dataset: s.Dataset,
 	}
 
-	if strings.HasPrefix(fileRef, "/") {
+	if strings.HasPrefix(fileRef, "/") || strings.HasPrefix(fileRef, "ws://") {
 		searchFile.AbsolutePath = fileRef
 	} else if _, err := uuid.Parse(fileRef); err == nil {
 		searchFile.ID = fileRef