use task type instead of prompt template (#1261)

2024-11-26 11:29:33 +08:00
parent 74a9fc70d6
commit 6b417d0e83
11 changed files with 124 additions and 69 deletions
--- a/skyvern/forge/sdk/db/enums.py
+++ b/skyvern/forge/sdk/db/enums.py
@@ -5,10 +5,7 @@ class OrganizationAuthTokenType(StrEnum):
    api = "api"


-class TaskPromptTemplate(StrEnum):
-    ExtractAction = "extract-action"
-    DecisiveCriterionValidate = "decisive-criterion-validate"
-    SingleClickAction = "single-click-action"
-    SingleInputAction = "single-input-action"
-    SingleUploadAction = "single-upload-action"
-    SingleSelectAction = "single-select-action"
+class TaskType(StrEnum):
+    general = "general"
+    validation = "validation"
+    action = "action"