migrate observer to task v2 (#1564)

This commit is contained in:
Shuchang Zheng 2025-01-15 09:59:18 -08:00 committed by GitHub
parent 997b0adea7
commit c158ad3f21
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
13 changed files with 79 additions and 83 deletions

View file

@ -5,7 +5,7 @@ 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.schemas.observers import ObserverTask, ObserverThought
from skyvern.forge.sdk.settings_manager import SettingsManager
@ -80,7 +80,7 @@ class LLMAPIHandler(Protocol):
self,
prompt: str,
step: Step | None = None,
observer_cruise: ObserverCruise | None = None,
observer_cruise: ObserverTask | None = None,
observer_thought: ObserverThought | None = None,
ai_suggestion: AISuggestion | None = None,
screenshots: list[bytes] | None = None,