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 (#2047)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-03-30 23:09:11 -07:00
committed by GitHub
parent ea67502161
commit 0107054a75
6 changed files with 229 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
integrations/llama_index/skyvern_llamaindex/agent.py
View File

@@ -104,7 +104,7 @@ class SkyvernTaskToolSpec(BaseToolSpec):
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 dispatch_task_v1(self, user_prompt: str, url: Optional[str] = None) -> CreateTaskResponse:
task_generation = await self._generate_v1_task_request(user_prompt=user_prompt)
@@ -112,7 +112,7 @@ class SkyvernTaskToolSpec(BaseToolSpec):
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 get_task_v1(self, task_id: str) -> TaskResponse | None:
return await self.agent.get_task(task_id=task_id)

2
integrations/llama_index/skyvern_llamaindex/client.py
View File

@@ -116,7 +116,7 @@ class SkyvernTaskToolSpec(BaseToolSpec):
if url is not None:
task_request.url = url
return await self.client.agent.run_task(
return await self.client.agent.run_task_v1(
timeout_seconds=self.run_task_timeout_seconds,
url=task_request.url,
title=task_request.title,