support list data schema (#1149)

This commit is contained in:
Shuchang Zheng 2024-11-06 16:14:39 -08:00 committed by GitHub
parent b73602dad4
commit 5583dee658
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions

View file

@ -173,7 +173,7 @@ class TaskBlock(Block):
title: str = "Untitled Task"
navigation_goal: str | None = None
data_extraction_goal: str | None = None
data_schema: dict[str, Any] | None = None
data_schema: dict[str, Any] | list | None = None
# error code to error description for the LLM
error_code_mapping: dict[str, str] | None = None
max_retries: int = 0

View file

@ -120,7 +120,7 @@ class TaskBlockYAML(BlockYAML):
title: str = "Untitled Task"
navigation_goal: str | None = None
data_extraction_goal: str | None = None
data_schema: dict[str, Any] | None = None
data_schema: dict[str, Any] | list | None = None
error_code_mapping: dict[str, str] | None = None
max_retries: int = 0
max_steps_per_run: int | None = None