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

Ykeremy/separate sonnet prompt (#263)

Browse Source
This commit is contained in:
Kerem Yilmaz
2024-05-06 13:07:52 -07:00
committed by GitHub
parent ccd435dd82
commit 25ac637b40
3 changed files with 94 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
skyvern/forge/agent.py
View File

@@ -636,8 +636,14 @@ class ForgeAgent:
current_url = (
await browser_state.page.evaluate("() => document.location.href") if browser_state.page else starting_url
)
prompt_template = "extract-action"
if app.EXPERIMENTATION_PROVIDER.is_feature_enabled_cached(
"USE_CLAUDE3_SONNET", task.workflow_run_id or task.task_id
):
LOG.info("Using Claude3 Sonnet prompt template for action extraction")
prompt_template = "extract-action-claude3-sonnet"
extract_action_prompt = prompt_engine.load_prompt(
"extract-action",
prompt_template,
navigation_goal=navigation_goal,
navigation_payload_str=json.dumps(task.navigation_payload),
starting_url=starting_url,