llm log (#3414)

2025-09-11 18:10:05 -07:00
parent 7a8974f6c7
commit 0e2aecc75d
11 changed files with 137 additions and 113 deletions
--- a/skyvern/forge/sdk/api/llm/models.py
+++ b/skyvern/forge/sdk/api/llm/models.py
@@ -95,6 +95,7 @@ class LLMAPIHandler(Protocol):
        ai_suggestion: AISuggestion | None = None,
        screenshots: list[bytes] | None = None,
        parameters: dict[str, Any] | None = None,
+        organization_id: str | None = None,
    ) -> Awaitable[dict[str, Any]]: ...


@@ -107,5 +108,6 @@ async def dummy_llm_api_handler(
    ai_suggestion: AISuggestion | None = None,
    screenshots: list[bytes] | None = None,
    parameters: dict[str, Any] | None = None,
+    organization_id: str | None = None,
 ) -> dict[str, Any]:
    raise NotImplementedError("Your LLM provider is not configured. Please configure it in the .env file.")