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

support get_run, run_task in SkyvernAgent (#2049)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-03-31 08:26:53 -07:00
committed by GitHub
parent 83ad2adabd
commit 3bcd7db2bb
4 changed files with 227 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
integrations/langchain/skyvern_langchain/agent.py
View File

@@ -46,7 +46,7 @@ class RunTask(SkyvernTaskBaseTool):
if url is not None:
task_request.url = url
return await self.agent.run_task(task_request=task_request, timeout_seconds=self.run_task_timeout_seconds)
return await self.agent.run_task_v1(task_request=task_request, timeout_seconds=self.run_task_timeout_seconds)
async def _arun_task_v2(self, user_prompt: str, url: str | None = None) -> TaskV2:
task_request = TaskV2Request(user_prompt=user_prompt, url=url)
@@ -72,7 +72,7 @@ class DispatchTask(SkyvernTaskBaseTool):
if url is not None:
task_request.url = url
return await self.agent.create_task(task_request=task_request)
return await self.agent.create_task_v1(task_request=task_request)
async def _arun_task_v2(self, user_prompt: str, url: str | None = None) -> TaskV2:
task_request = TaskV2Request(user_prompt=user_prompt, url=url)