From 95c0ac4debca1c892586a6659f19d46e795c431f Mon Sep 17 00:00:00 2001
From: Suchintan <suchintan@users.noreply.github.com>
Date: Mon, 21 Jul 2025 22:59:03 -0700
Subject: [PATCH] Add O3 as a model option (#3008)

Co-authored-by: Suchintan Singh <suchintan@skyvern.com>
---
 skyvern/config.py             | 4 ++--
 skyvern/forge/sdk/db/polls.py | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/skyvern/config.py b/skyvern/config.py
index f4bebfa9..58f91775 100644
--- a/skyvern/config.py
+++ b/skyvern/config.py
@@ -308,7 +308,7 @@ class Settings(BaseSettings):
                     "label": "Gemini 2.5 Flash",
                 },
                 "azure/gpt-4.1": {"llm_key": "AZURE_OPENAI_GPT4_1", "label": "GPT 4.1"},
-                "azure/o4-mini": {"llm_key": "AZURE_OPENAI_O4_MINI", "label": "GPT O4 Mini"},
+                "azure/o3": {"llm_key": "AZURE_OPENAI_O3", "label": "GPT O3"},
                 # "us.anthropic.claude-opus-4-20250514-v1:0": {
                 #     "llm_key": "BEDROCK_ANTHROPIC_CLAUDE4_OPUS_INFERENCE_PROFILE",
                 #     "label": "Anthropic Claude 4 Opus",
@@ -335,7 +335,7 @@ class Settings(BaseSettings):
                     "label": "Gemini 2.5 Flash",
                 },
                 "azure/gpt-4.1": {"llm_key": "AZURE_OPENAI_GPT4_1", "label": "GPT 4.1"},
-                "azure/o4-mini": {"llm_key": "AZURE_OPENAI_O4_MINI", "label": "GPT O4 Mini"},
+                "azure/o3": {"llm_key": "AZURE_OPENAI_O3", "label": "GPT O3"},
                 "us.anthropic.claude-opus-4-20250514-v1:0": {
                     "llm_key": "BEDROCK_ANTHROPIC_CLAUDE4_OPUS_INFERENCE_PROFILE",
                     "label": "Anthropic Claude 4 Opus",
diff --git a/skyvern/forge/sdk/db/polls.py b/skyvern/forge/sdk/db/polls.py
index 9f9634f3..addee714 100644
--- a/skyvern/forge/sdk/db/polls.py
+++ b/skyvern/forge/sdk/db/polls.py
@@ -13,7 +13,7 @@ async def wait_on_persistent_browser_address(
     session_id: str,
     organization_id: str,
     timeout: int = 600,
-    poll_interval: float = 2,
+    poll_interval: int = 2,
 ) -> str | None:
     persistent_browser_session = await await_browser_session(db, session_id, organization_id, timeout, poll_interval)
     return persistent_browser_session.browser_address if persistent_browser_session else None
@@ -24,7 +24,7 @@ async def await_browser_session(
     session_id: str,
     organization_id: str,
     timeout: int = 600,
-    poll_interval: float = 2,
+    poll_interval: int = 2,
 ) -> PersistentBrowserSession | None:
     try:
         async with asyncio.timeout(timeout):