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

Clean up alembic migrations: fix HTTP request block columns and add parameters column (#2848)

Browse Source 
Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>
This commit is contained in:
Prakash Maheshwaran
2025-07-01 14:10:58 -04:00
committed by GitHub
parent d23944bca7
commit 47f9f1dabb
11 changed files with 536 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -2765,6 +2765,14 @@ class AgentDB:
description: str | None = None,
block_workflow_run_id: str | None = None,
engine: str | None = None,
# HTTP request block parameters
http_request_method: str | None = None,
http_request_url: str | None = None,
http_request_headers: dict[str, str] | None = None,
http_request_body: dict[str, Any] | None = None,
http_request_parameters: dict[str, Any] | None = None,
http_request_timeout: int | None = None,
http_request_follow_redirects: bool | None = None,
) -> WorkflowRunBlock:
async with self.Session() as session:
workflow_run_block = (
@@ -2807,6 +2815,21 @@ class AgentDB:
workflow_run_block.block_workflow_run_id = block_workflow_run_id
if engine:
workflow_run_block.engine = engine
# HTTP request block fields
if http_request_method:
workflow_run_block.http_request_method = http_request_method
if http_request_url:
workflow_run_block.http_request_url = http_request_url
if http_request_headers:
workflow_run_block.http_request_headers = http_request_headers
if http_request_body:
workflow_run_block.http_request_body = http_request_body
if http_request_parameters:
workflow_run_block.http_request_parameters = http_request_parameters
if http_request_timeout:
workflow_run_block.http_request_timeout = http_request_timeout
if http_request_follow_redirects is not None:
workflow_run_block.http_request_follow_redirects = http_request_follow_redirects
await session.commit()
await session.refresh(workflow_run_block)
else:

9
skyvern/forge/sdk/db/models.py
View File

@@ -606,6 +606,15 @@ class WorkflowRunBlockModel(Base):
# wait block
wait_sec = Column(Integer, nullable=True)
# http request block
http_request_method = Column(String(10), nullable=True)
http_request_url = Column(String, nullable=True)
http_request_headers = Column(JSON, nullable=True)
http_request_body = Column(JSON, nullable=True)
http_request_parameters = Column(JSON, nullable=True)
http_request_timeout = Column(Integer, nullable=True)
http_request_follow_redirects = Column(Boolean, nullable=True)
created_at = Column(DateTime, default=datetime.datetime.utcnow, nullable=False)
modified_at = Column(DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow, nullable=False)