fix: Add gpt4.1, o3 to get_forecast

Author: robcaulk

asknews_sdk/api/chat.py

@@ -197,13 +197,14 @@ def get_forecast(
         method: Literal["nl", "kw", "both"] = "kw",
         model: Literal[
             "gpt-4o",
+            "gpt-4o-mini",
+            "gpt-4.1-2025-04-14",
+            "claude-3-5-sonnet-latest",
             "claude-3-5-sonnet-20240620",
-            "command-nightly",
-            "meta-llama/Meta-Llama-3.1-405B-Instruct",
-            "o1-mini",
-            "o3-mini",
-            "o1-preview",
-        ] = "claude-3-5-sonnet-20240620",
+            "claude-sonnet-4-20250514",
+            "claude-opus-4-20250514" "o3-mini",
+            "o3",
+        ] = "gpt-4.1-2025-04-14",
         cutoff_date: Optional[str] = None,
         use_reddit: bool = False,
         additional_context: Optional[str] = None,
@@ -735,12 +736,14 @@ async def get_forecast(
         method: Literal["nl", "kw", "both"] = "kw",
         model: Literal[
             "gpt-4o",
+            "gpt-4o-mini",
+            "gpt-4.1-2025-04-14",
+            "claude-3-5-sonnet-latest",
             "claude-3-5-sonnet-20240620",
-            "command-nightly",
-            "meta-llama/Meta-Llama-3.1-405B-Instruct",
-            "o1-mini",
-            "o1-preview",
-        ] = "claude-3-5-sonnet-20240620",
+            "claude-sonnet-4-20250514",
+            "claude-opus-4-20250514" "o3-mini",
+            "o3",
+        ] = "gpt-4.1-2025-04-14",
         cutoff_date: Optional[str] = None,
         use_reddit: bool = False,
         additional_context: Optional[str] = None,