SDK: text prompt (#4214)

2025-12-05 18:13:25 -07:00
parent 0f495f458e
commit b7d08fe906
9 changed files with 156 additions and 0 deletions
--- a/skyvern/forge/sdk/routes/sdk.py
+++ b/skyvern/forge/sdk/routes/sdk.py
@@ -205,6 +205,13 @@ async def run_sdk_action(
                prompt=action.prompt,
            )
            result = xpath_result
+        elif action.type == "prompt":
+            prompt_result = await page_ai.ai_prompt(
+                prompt=action.prompt,
+                schema=action.schema,
+                model=action.model,
+            )
+            result = prompt_result
        await app.DATABASE.update_task(
            task_id=task.task_id,
            organization_id=organization_id,