support get_run, run_task in SkyvernAgent (#2047)

2025-03-30 23:09:11 -07:00
parent ea67502161
commit 0107054a75
6 changed files with 229 additions and 28 deletions
--- a/integrations/llama_index/skyvern_llamaindex/client.py
+++ b/integrations/llama_index/skyvern_llamaindex/client.py
@@ -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,