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

fix totp code integration for anthropic CU (#2998)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-07-21 10:15:18 -07:00
committed by GitHub
parent f0081f55f1
commit e7a2816470
1 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
skyvern/forge/agent.py
View File

@@ -1524,7 +1524,20 @@ class ForgeAgent:
window_dimension=window_dimension, window_dimension=window_dimension,
) )
else: else:
current_context = skyvern_context.ensure_context()
resp_content = None
if task.task_id in current_context.totp_codes:
verification_code = current_context.totp_codes[task.task_id]
current_context.totp_codes.pop(task.task_id)
LOG.info(
"Using verification code from context for anthropic CU call",
task_id=task.task_id,
verification_code=verification_code,
)
resp_content = f"Here is the verification code: {verification_code}"
llm_response = await llm_caller.call( llm_response = await llm_caller.call(
prompt=resp_content,
step=step, step=step,
screenshots=scraped_page.screenshots, screenshots=scraped_page.screenshots,
use_message_history=True, use_message_history=True,