Merge pull request #1001 from parea-ai/PAI-1387-add-4-o-mini

feat(mini): add gpt mini
parea-ai · Jul 18, 2024 · bbca12c · bbca12c
2 parents 04cde9f + 6931255
commit bbca12c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/parea/constants.py b/parea/constants.py
@@ -145,6 +145,16 @@ def str2bool(v):
         "completion": 15.0,
         "token_limit": {"max_completion_tokens": 4096, "max_prompt_tokens": 128000},
     },
+    "gpt-4o-mini": {
+        "prompt": 0.15,
+        "completion": 0.6,
+        "token_limit": {"max_completion_tokens": 4096, "max_prompt_tokens": 128000},
+    },
+    "gpt-4o-mini-2024-07-18": {
+        "prompt": 0.15,
+        "completion": 0.6,
+        "token_limit": {"max_completion_tokens": 4096, "max_prompt_tokens": 128000},
+    },
 }
 AZURE_MODEL_INFO: Dict[str, Dict[str, Union[float, int, Dict[str, int]]]] = {
     "gpt-35-turbo": {

diff --git a/pyproject.toml b/pyproject.toml
@@ -6,7 +6,7 @@ build-backend = "poetry.core.masonry.api"
 [tool.poetry]
 name = "parea-ai"
 packages = [{ include = "parea" }]
-version = "0.2.186"
+version = "0.2.187"
 description = "Parea python sdk"
 readme = "README.md"
 authors = ["joel-parea-ai <[email protected]>"]