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/langchain/skyvern_langchain/agent.py
View File

@@ -7,7 +7,7 @@ from skyvern_langchain.schema import CreateTaskInput, GetTaskInput
from skyvern_langchain.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
@@ -56,5 +56,5 @@ class GetTask(SkyvernTaskBaseTool):
description: str = """Use Skyvern agent to get a task."""
args_schema: Type[BaseModel] = GetTaskInput
async def _arun(self, task_id: str) -> AgentGetRunResponse | None:
async def _arun(self, task_id: str) -> GetRunResponse | None:
return await self.agent.get_run(run_id=task_id)

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

@@ -6,7 +6,7 @@ from skyvern_langchain.schema import CreateTaskInput, GetTaskInput
from skyvern_langchain.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
@@ -59,5 +59,5 @@ class GetTask(SkyvernTaskBaseTool):
description: str = """Use Skyvern client to get a task."""
args_schema: Type[BaseModel] = GetTaskInput
async def _arun(self, task_id: str) -> AgentGetRunResponse | None:
async def _arun(self, task_id: str) -> GetRunResponse | None:
return await self.get_client().get_run(run_id=task_id)

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.