Add defaultdict import and handle empty custom_fields

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-08 11:24:23 -03:00
commit 1f76ad728f

View file

@ -1,3 +1,5 @@
from collections import defaultdict
from fastapi import APIRouter, HTTPException
from loguru import logger
@ -42,6 +44,8 @@ def post_validate_prompt(prompt_request: ValidatePromptRequest):
input_variables=input_variables,
frontend_node=None,
)
if not prompt_request.custom_fields:
prompt_request.custom_fields = defaultdict(list)
old_custom_fields = get_old_custom_fields(
prompt_request.custom_fields, prompt_request.name
)