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

langchain bugfix (#1870)

Browse Source
This commit is contained in:
LawyZheng
2025-03-03 23:05:35 +08:00
committed by GitHub
parent c6a733ff27
commit c7e6a5c84b
4 changed files with 13 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -18,7 +18,7 @@ agent = Agent()
class SkyvernTaskBaseTool(BaseTool):
engine: Literal["TaskV1", "TaskV2"] = Field(default=settings.engine)
timeout_seconds: int = Field(default=settings.run_task_timeout)
run_task_timeout_seconds: int = Field(default=settings.run_task_timeout_seconds)
agent: Agent = agent
def _run(self, *args: Any, **kwargs: Any) -> None:
@@ -48,11 +48,13 @@ 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.timeout_seconds)
return await self.agent.run_task(task_request=task_request, timeout_seconds=self.run_task_timeout_seconds)
async def _arun_task_v2(self, user_prompt: str, url: str | None = None) -> ObserverTask:
task_request = ObserverTaskRequest(user_prompt=user_prompt, url=url)
return await self.agent.run_observer_task_v_2(task_request=task_request, timeout_seconds=self.timeout_seconds)
return await self.agent.run_observer_task_v_2(
task_request=task_request, timeout_seconds=self.run_task_timeout_seconds
)
class DispatchTask(SkyvernTaskBaseTool):