From 30922a3f8c7dbfb18da3ab16ed7be68d8935dad5 Mon Sep 17 00:00:00 2001 From: Kerem Yilmaz Date: Thu, 19 Sep 2024 13:12:46 -0700 Subject: [PATCH] Only use json.dumps when workflow parameter type is json (#859) --- skyvern/forge/sdk/db/client.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/skyvern/forge/sdk/db/client.py b/skyvern/forge/sdk/db/client.py index 9321cdcd..98c44ca9 100644 --- a/skyvern/forge/sdk/db/client.py +++ b/skyvern/forge/sdk/db/client.py @@ -1123,11 +1123,16 @@ class AgentDB: ) -> WorkflowParameter: try: async with self.Session() as session: + default_value = ( + json.dumps(default_value) + if workflow_parameter_type == WorkflowParameterType.JSON + else default_value + ) workflow_parameter = WorkflowParameterModel( workflow_id=workflow_id, workflow_parameter_type=workflow_parameter_type, key=key, - default_value=json.dumps(default_value), + default_value=default_value, description=description, ) session.add(workflow_parameter)