update langchain integration (#1815)

2025-03-03 10:38:00 +08:00
parent 889c8f1963
commit b3b548ed38
6 changed files with 369 additions and 211 deletions
--- a/integrations/langchain/skyvern_langchain/schema.py
+++ b/integrations/langchain/skyvern_langchain/schema.py
@@ -1,17 +1,9 @@
 from pydantic import BaseModel

-from skyvern.forge.sdk.schemas.observers import ObserverTaskRequest
-from skyvern.forge.sdk.schemas.tasks import TaskRequest

-
-class TaskV1Request(TaskRequest):
-    max_steps: int = 10
-    timeout_seconds: int = 60 * 60
-
-
-class TaskV2Request(ObserverTaskRequest):
-    max_iterations: int = 10
-    timeout_seconds: int = 60 * 60
+class CreateTaskInput(BaseModel):
+    user_prompt: str
+    url: str | None = None


 class GetTaskInput(BaseModel):