vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2026-04-29 04:00:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

add model to Task and TaskV2; expose it to run_task endpoint; thread … (#2540)

Browse source
This commit is contained in:
Shuchang Zheng 2025-05-30 20:07:12 -07:00 • committed by GitHub
parent aee129a0a8
commit 2ed14f42e7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
14 changed files with 103 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

5
skyvern/forge/sdk/executor/async_executor.py
View file

@ -5,6 +5,7 @@ from fastapi import BackgroundTasks, Request
from skyvern.exceptions import OrganizationNotFound
from skyvern.forge import app
from skyvern.forge.sdk.api.llm.api_handler_factory import LLMCaller
from skyvern.forge.sdk.core import skyvern_context
from skyvern.forge.sdk.core.skyvern_context import SkyvernContext
from skyvern.forge.sdk.schemas.organizations import Organization
@ -105,6 +106,9 @@ class BackgroundTaskExecutor(AsyncExecutor):
context.organization_id = organization_id
context.max_steps_override = max_steps_override
llm_key = task.llm_key
llm_caller = LLMCaller(llm_key) if llm_key else None
if background_tasks:
background_tasks.add_task(
app.agent.execute_step,
@ -115,6 +119,7 @@ class BackgroundTaskExecutor(AsyncExecutor):
close_browser_on_completion=close_browser_on_completion,
browser_session_id=browser_session_id,
engine=engine,
llm_caller=llm_caller,
)
async def execute_workflow(