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

move every interface to top level and get rid of sdk client grouping (#2490)

Browse Source
This commit is contained in:
Shuchang Zheng
2025-05-27 23:38:14 -07:00
committed by GitHub
parent 1e16559141
commit abab86619c
33 changed files with 3358 additions and 3563 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -5,7 +5,7 @@ from llama_index.core.tools.tool_spec.base import SPEC_FUNCTION_TYPE, BaseToolSp
from skyvern_llamaindex.settings import settings
from skyvern import Skyvern
from skyvern.client.agent.types.agent_get_run_response import AgentGetRunResponse
from skyvern.client.types.get_run_response import GetRunResponse
from skyvern.client.types.task_run_response import TaskRunResponse
from skyvern.schemas.runs import RunEngine
@@ -81,7 +81,7 @@ class SkyvernTaskToolSpec(BaseToolSpec):
wait_for_completion=False,
)
async def get_task(self, task_id: str) -> AgentGetRunResponse | None:
async def get_task(self, task_id: str) -> GetRunResponse | None:
"""
Use Skyvern agent to get a task.

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

@@ -6,7 +6,7 @@ from pydantic import BaseModel
from skyvern_llamaindex.settings import settings
from skyvern import Skyvern
from skyvern.client.agent.types.agent_get_run_response import AgentGetRunResponse
from skyvern.client.types.get_run_response import GetRunResponse
from skyvern.client.types.task_run_response import TaskRunResponse
from skyvern.schemas.runs import RunEngine
@@ -93,7 +93,7 @@ class SkyvernTaskToolSpec(BaseToolSpec):
wait_for_completion=False,
)
async def get_task(self, task_id: str) -> AgentGetRunResponse | None:
async def get_task(self, task_id: str) -> GetRunResponse | None:
"""
Use Skyvern client to get a task.