vodorod/Dorod-Sky
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix task v2 data schema (#1967)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-03-18 15:36:42 -07:00
committed by GitHub
parent 0b8e688d7b
commit be8f3eb8ec
4 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
skyvern/forge/sdk/db/client.py
View File

@@ -2251,6 +2251,8 @@ class AgentDB:
totp_identifier: str | None = None,
totp_verification_url: str | None = None,
webhook_callback_url: str | None = None,
extracted_information_schema: dict | list | str | None = None,
error_code_mapping: dict | None = None,
) -> TaskV2:
async with self.Session() as session:
new_task_v2 = TaskV2Model(
@@ -2263,6 +2265,8 @@ class AgentDB:
totp_identifier=totp_identifier,
totp_verification_url=totp_verification_url,
webhook_callback_url=webhook_callback_url,
extracted_information_schema=extracted_information_schema,
error_code_mapping=error_code_mapping,
organization_id=organization_id,
)
session.add(new_task_v2)