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

Add datetime library to code block (#823)

Browse Source
This commit is contained in:
Kerem Yilmaz
2024-09-12 11:24:44 -07:00
committed by GitHub
parent 068adccb87
commit 9c4c3b76b6
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
skyvern/forge/sdk/workflow/models/block.py
View File

@@ -1,5 +1,4 @@
import abc import abc
import asyncio
import csv import csv
import json import json
import os import os
@@ -537,7 +536,7 @@ class CodeBlock(Block):
maybe_browser_state = await app.BROWSER_MANAGER.get_for_workflow_run(workflow_run_id) maybe_browser_state = await app.BROWSER_MANAGER.get_for_workflow_run(workflow_run_id)
if maybe_browser_state: if maybe_browser_state:
if page := await maybe_browser_state.get_working_page(): if page := await maybe_browser_state.get_working_page():
parameter_values["skyvern_page"] = await page parameter_values["skyvern_page"] = page
for parameter in self.parameters: for parameter in self.parameters:
value = workflow_run_context.get_value(parameter.key) value = workflow_run_context.get_value(parameter.key)
@@ -547,9 +546,13 @@ class CodeBlock(Block):
else: else:
parameter_values[parameter.key] = value parameter_values[parameter.key] = value
# Add asyncio and the current event loop to the parameter_values # Import builtins and other modules that might be useful in the user code and add them to the parameter_values
parameter_values["asyncio"] = asyncio import asyncio
import datetime
parameter_values["__builtins__"] = __builtins__ # Include builtins for exec context parameter_values["__builtins__"] = __builtins__ # Include builtins for exec context
parameter_values["asyncio"] = asyncio
parameter_values["datetime"] = datetime
local_variables: dict[str, Any] = {} local_variables: dict[str, Any] = {}
result_container: dict[str, Any] = {} result_container: dict[str, Any] = {}