feat(models): add new openai models

parea-ai · Sep 12, 2024 · feeec14 · feeec14
1 parent b1cb7be
commit feeec14
Show file tree

Hide file tree

Showing 3 changed files with 43 additions and 3 deletions.
diff --git a/cookbook/openai/tracing_with_openai_with_functions.py b/cookbook/openai/tracing_with_openai_with_functions.py
@@ -3,9 +3,9 @@
 import json
 import os
 
-import openai
 import requests
 from dotenv import load_dotenv
+from openai import OpenAI
 
 from parea import Parea, get_current_trace_id, trace
 from parea.schemas import FeedbackRequest
@@ -14,7 +14,7 @@
 
 API_KEY = os.getenv("GOOGLE_PLACES_API_KEY")
 PLACES_URL = "https://maps.googleapis.com/maps/api/place/nearbysearch/json"
-openai.api_key = os.getenv("OPENAI_API_KEY")
+openai = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 
 p = Parea(api_key=os.getenv("PAREA_API_KEY"))
 p.wrap_openai_client(openai)

diff --git a/parea/constants.py b/parea/constants.py
@@ -166,6 +166,26 @@ def str2bool(v):
         "completion": 10.0,
         "token_limit": {"max_completion_tokens": 16384, "max_prompt_tokens": 128000},
     },
+    "o1-preview": {
+        "prompt": 15.0,
+        "completion": 60.0,
+        "token_limit": {"max_completion_tokens": 16384, "max_prompt_tokens": 128000},
+    },
+    "o1-preview-2024-09-12": {
+        "prompt": 15.0,
+        "completion": 60.0,
+        "token_limit": {"max_completion_tokens": 16384, "max_prompt_tokens": 128000},
+    },
+    "o1-mini": {
+        "prompt": 3.0,
+        "completion": 12.0,
+        "token_limit": {"max_completion_tokens": 16384, "max_prompt_tokens": 128000},
+    },
+    "o1-mini-2024-09-12": {
+        "prompt": 3.0,
+        "completion": 12.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": {
@@ -268,6 +288,26 @@ def str2bool(v):
         "completion": 15.0,
         "token_limit": {"max_completion_tokens": 4096, "max_prompt_tokens": 128000},
     },
+    "o1-preview": {
+        "prompt": 15.0,
+        "completion": 60.0,
+        "token_limit": {"max_completion_tokens": 16384, "max_prompt_tokens": 128000},
+    },
+    "o1-preview-2024-09-12": {
+        "prompt": 15.0,
+        "completion": 60.0,
+        "token_limit": {"max_completion_tokens": 16384, "max_prompt_tokens": 128000},
+    },
+    "o1-mini": {
+        "prompt": 3.0,
+        "completion": 12.0,
+        "token_limit": {"max_completion_tokens": 16384, "max_prompt_tokens": 128000},
+    },
+    "o1-mini-2024-09-12": {
+        "prompt": 3.0,
+        "completion": 12.0,
+        "token_limit": {"max_completion_tokens": 16384, "max_prompt_tokens": 128000},
+    },
 }
 
 instant_pricing = {"prompt": 1.63, "completion": 5.51}

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.214"
+version = "0.2.215"
 description = "Parea python sdk"
 readme = "README.md"
 authors = ["joel-parea-ai <joel@parea.ai>"]