use task type instead of prompt template (#1261)

skyvern/forge/sdk/db/client.py

@@ -10,7 +10,7 @@ from sqlalchemy.ext.asyncio import async_sessionmaker, create_async_engine
 from skyvern.config import settings
 from skyvern.exceptions import WorkflowParameterNotFound
 from skyvern.forge.sdk.artifact.models import Artifact, ArtifactType
-from skyvern.forge.sdk.db.enums import OrganizationAuthTokenType, TaskPromptTemplate
+from skyvern.forge.sdk.db.enums import OrganizationAuthTokenType, TaskType
 from skyvern.forge.sdk.db.exceptions import NotFoundError
 from skyvern.forge.sdk.db.models import (
     ActionModel,
@@ -113,13 +113,13 @@ class AgentDB:
         retry: int | None = None,
         max_steps_per_run: int | None = None,
         error_code_mapping: dict[str, str] | None = None,
-        prompt_template: str = TaskPromptTemplate.ExtractAction,
+        task_type: str = TaskType.general,
     ) -> Task:
         try:
             async with self.Session() as session:
                 new_task = TaskModel(
                     status="created",
-                    prompt_template=prompt_template,
+                    task_type=task_type,
                     url=url,
                     title=title,
                     webhook_callback_url=webhook_callback_url,

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"

skyvern/forge/sdk/db/models.py

@@ -17,7 +17,7 @@ from sqlalchemy import (
 from sqlalchemy.ext.asyncio import AsyncAttrs
 from sqlalchemy.orm import DeclarativeBase
 
-from skyvern.forge.sdk.db.enums import OrganizationAuthTokenType, TaskPromptTemplate
+from skyvern.forge.sdk.db.enums import OrganizationAuthTokenType, TaskType
 from skyvern.forge.sdk.db.id import (
     generate_action_id,
     generate_artifact_id,
@@ -54,7 +54,7 @@ class TaskModel(Base):
     totp_verification_url = Column(String)
     totp_identifier = Column(String)
     title = Column(String)
-    prompt_template = Column(String, default=TaskPromptTemplate.ExtractAction)
+    task_type = Column(String, default=TaskType.general)
     url = Column(String)
     navigation_goal = Column(String)
     data_extraction_goal = Column(String)

skyvern/forge/sdk/db/utils.py

@@ -60,7 +60,7 @@ def convert_to_task(task_obj: TaskModel, debug_enabled: bool = False) -> Task:
         status=TaskStatus(task_obj.status),
         created_at=task_obj.created_at,
         modified_at=task_obj.modified_at,
-        prompt_template=task_obj.prompt_template,
+        task_type=task_obj.task_type,
         title=task_obj.title,
         url=task_obj.url,
         complete_criterion=task_obj.complete_criterion,