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

2
integrations/langchain/pyproject.toml
View File

@@ -1,6 +1,6 @@
[tool.poetry]
name = "skyvern-langchain"
version = "0.1.5"
version = "0.1.6"
description = ""
authors = ["lawyzheng <lawy@skyvern.com>"]
packages = [{ include = "skyvern_langchain" }]

0
integrations/langchain/skyvern_langchain/__init__.py Normal file
View File

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):

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

@@ -15,7 +15,7 @@ class SkyvernTaskBaseTool(BaseTool):
api_key: str = Field(default=settings.api_key)
base_url: str = Field(default=settings.base_url)
engine: Literal["TaskV1", "TaskV2"] = Field(default=settings.engine)
run_task_timeout_seconds: int = Field(default=settings.run_task_timeout)
run_task_timeout_seconds: int = Field(default=settings.run_task_timeout_seconds)
def get_client(self) -> AsyncSkyvern:
httpx_client = AsyncClient(
@@ -46,10 +46,10 @@ class RunTask(SkyvernTaskBaseTool):
prompt=user_prompt,
)
if url is not None:
task_generation.url = url
task_request = TaskRequest.model_validate(task_generation, from_attributes=True)
if url is not None:
task_request.url = url
return await self.get_client().agent.run_task(
timeout_seconds=self.run_task_timeout_seconds,
url=task_request.url,
@@ -89,10 +89,10 @@ class DispatchTask(SkyvernTaskBaseTool):
prompt=user_prompt,
)
if url is not None:
task_generation.url = url
task_request = TaskRequest.model_validate(task_generation, from_attributes=True)
if url is not None:
task_request.url = url
return await self.get_client().agent.create_task(
url=task_request.url,
title=task_request.title,