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

jalexanderII · web-flow · commit 971b1180ec2d · 2024-08-13T22:27:29.000-04:00
add pricing
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": {
