Added gpt-4o to model costs and model names (#671)

Co-authored-by: rstaneker <[email protected]>
instructor-ai · May 16, 2024 · d2648f5 · d2648f5
1 parent 28deb93
commit d2648f5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/instructor/_types/_alias.py b/instructor/_types/_alias.py
@@ -3,6 +3,7 @@
 from typing_extensions import TypeAlias
 
 ModelNames: TypeAlias = Literal[
+    "gpt-4o",
     "gpt-4-0125-preview",
     "gpt-4-turbo-preview",
     "gpt-4-1106-preview",

diff --git a/instructor/cli/usage.py b/instructor/cli/usage.py
@@ -49,6 +49,7 @@ async def get_usage_for_past_n_days(n_days: int) -> list[dict[str, Any]]:
 
 # Define the cost per unit for each model
 MODEL_COSTS = {
+    "gpt-4o": {"prompt": 0.005 / 1000, "completion": 0.015 / 1000},
     "gpt-4-0125-preview": {"prompt": 0.01 / 1000, "completion": 0.03 / 1000},
     "gpt-4-turbo-preview": {"prompt": 0.01 / 1000, "completion": 0.03 / 1000},
     "gpt-4-1106-preview": {"prompt": 0.01 / 1000, "completion": 0.03 / 1000},