Merge pull request #1054 from parea-ai/new-openai

add pricing
parea-ai · Aug 14, 2024 · 971b118 · 971b118
2 parents 3777103 + daab9dc
commit 971b118
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/parea/constants.py b/parea/constants.py
@@ -161,6 +161,11 @@ def str2bool(v):
         "completion": 0.6,
         "token_limit": {"max_completion_tokens": 4096, "max_prompt_tokens": 128000},
     },
+    "chatgpt-4o-latest": {
+        "prompt": 5.0,
+        "completion": 10.0,
+        "token_limit": {"max_completion_tokens": 16384, "max_prompt_tokens": 128000},
+    },
 }
 AZURE_MODEL_INFO: Dict[str, Dict[str, Union[float, int, Dict[str, int]]]] = {
     "gpt-35-turbo": {