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

overhual llm key override (#2677)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-06-11 08:23:44 -07:00
committed by GitHub
parent 9a42c2ad9f
commit 9a29d966ab
5 changed files with 72 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
skyvern/services/task_v2_service.py
View File

@@ -20,6 +20,7 @@ from skyvern.exceptions import (
)
from skyvern.forge import app
from skyvern.forge.prompts import prompt_engine
from skyvern.forge.sdk.api.llm.api_handler_factory import LLMAPIHandlerFactory
from skyvern.forge.sdk.artifact.models import ArtifactType
from skyvern.forge.sdk.core import skyvern_context
from skyvern.forge.sdk.core.hashing import generate_url_hash
@@ -617,12 +618,14 @@ async def run_task_v2_helper(
thought_type=ThoughtType.plan,
thought_scenario=ThoughtScenario.generate_plan,
)
task_v2_response = await app.LLM_API_HANDLER(
llm_api_handler = LLMAPIHandlerFactory.get_override_llm_api_handler(
task_v2.llm_key, default=app.LLM_API_HANDLER
)
task_v2_response = await llm_api_handler(
prompt=task_v2_prompt,
screenshots=scraped_page.screenshots,
thought=thought,
prompt_name="task_v2",
llm_key_override=task_v2.llm_key,
)
LOG.info(
"Task v2 response",