Workflow Copilot: backend side of the first version (#4401)

skyvern/forge/sdk/db/utils.py

@@ -21,6 +21,7 @@ from skyvern.forge.sdk.db.models import (
     StepModel,
     TaskModel,
     TaskV2Model,
+    WorkflowCopilotChatMessageModel,
     WorkflowModel,
     WorkflowParameterModel,
     WorkflowRunBlockModel,
@@ -39,6 +40,7 @@ from skyvern.forge.sdk.schemas.organizations import (
 )
 from skyvern.forge.sdk.schemas.task_v2 import TaskV2
 from skyvern.forge.sdk.schemas.tasks import Task, TaskStatus
+from skyvern.forge.sdk.schemas.workflow_copilot import WorkflowCopilotChatMessage as WorkflowCopilotChatMessageSchema
 from skyvern.forge.sdk.schemas.workflow_runs import WorkflowRunBlock
 from skyvern.forge.sdk.workflow.models.parameter import (
     AWSSecretParameter,
@@ -217,6 +219,17 @@ def convert_to_task_v2(task_v2_model: TaskV2Model, debug_enabled: bool = False)
     return TaskV2.model_validate(task_v2_data)
 
 
+def convert_to_workflow_copilot_chat_message(
+    message_model: WorkflowCopilotChatMessageModel, debug_enabled: bool = False
+) -> WorkflowCopilotChatMessageSchema:
+    if debug_enabled:
+        LOG.debug(
+            "Converting WorkflowCopilotChatMessage to WorkflowCopilotChatMessageSchema",
+            workflow_copilot_chat_message_id=message_model.workflow_copilot_chat_message_id,
+        )
+    return WorkflowCopilotChatMessageSchema.model_validate(message_model)
+
+
 def convert_to_step(step_model: StepModel, debug_enabled: bool = False) -> Step:
     if debug_enabled:
         LOG.debug("Converting StepModel to Step", step_id=step_model.step_id)