vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-09-13 00:39:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Add AI suggestion endpoints (#1519)

Browse source
This commit is contained in:
Shuchang Zheng 2025-01-08 21:45:38 -08:00 • committed by GitHub
parent f0ae840452
commit 5796de73d1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
14 changed files with 248 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
skyvern/forge/sdk/api/llm/models.py
View file

@ -4,6 +4,7 @@ from typing import Any, Awaitable, Literal, Optional, Protocol, TypedDict
from litellm import AllowedFailsPolicy
from skyvern.forge.sdk.models import Step
from skyvern.forge.sdk.schemas.ai_suggestions import AISuggestion
from skyvern.forge.sdk.schemas.observers import ObserverCruise, ObserverThought
from skyvern.forge.sdk.settings_manager import SettingsManager
@ -81,6 +82,7 @@ class LLMAPIHandler(Protocol):
step: Step | None = None,
observer_cruise: ObserverCruise | None = None,
observer_thought: ObserverThought | None = None,
ai_suggestion: AISuggestion | None = None,
screenshots: list[bytes] | None = None,
parameters: dict[str, Any] | None = None,
) -> Awaitable[dict[str, Any]]: ...