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

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,