Only use json.dumps when workflow parameter type is json (#859)

This commit is contained in:
Kerem Yilmaz
2024-09-19 13:12:46 -07:00
committed by GitHub
parent 39c5f6233e
commit 30922a3f8c

View File

@@ -1123,11 +1123,16 @@ class AgentDB:
) -> WorkflowParameter: ) -> WorkflowParameter:
try: try:
async with self.Session() as session: 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_parameter = WorkflowParameterModel(
workflow_id=workflow_id, workflow_id=workflow_id,
workflow_parameter_type=workflow_parameter_type, workflow_parameter_type=workflow_parameter_type,
key=key, key=key,
default_value=json.dumps(default_value), default_value=default_value,
description=description, description=description,
) )
session.add(workflow_parameter) session.add(workflow_parameter)