llama-index-llms-openai 0.3.10__py3-none-any.whl → 0.3.12__py3-none-any.whl

llama_index/llms/openai/utils.py

@@ -34,12 +34,21 @@ DEFAULT_OPENAI_API_BASE = "https://api.openai.com/v1"
 DEFAULT_OPENAI_API_VERSION = ""
 
 O1_MODELS: Dict[str, int] = {
+    "o1": 200000,
+    "o1-2024-12-17": 200000,
     "o1-preview": 128000,
     "o1-preview-2024-09-12": 128000,
     "o1-mini": 128000,
     "o1-mini-2024-09-12": 128000,
 }
 
+O1_MODELS_WITHOUT_FUNCTION_CALLING = {
+    "o1-preview",
+    "o1-preview-2024-09-12",
+    "o1-mini",
+    "o1-mini-2024-09-12",
+}
+
 GPT4_MODELS: Dict[str, int] = {
     # stable model names:
     #   resolves to gpt-4-0314 before 2023-06-27,
@@ -245,9 +254,7 @@ def is_function_calling_model(model: str) -> bool:
 
     is_chat_model_ = is_chat_model(model)
     is_old = "0314" in model or "0301" in model
-
-    # TODO: This is temporary for openai's beta
-    is_o1_beta = "o1" in model
+    is_o1_beta = model in O1_MODELS_WITHOUT_FUNCTION_CALLING
 
     return is_chat_model_ and not is_old and not is_o1_beta
 
@@ -304,7 +311,7 @@ def to_openai_message_dict(
     # TODO: O1 models do not support system prompts
     if model is not None and model in O1_MODELS:
         if message_dict["role"] == "system":
-            message_dict["role"] = "user"
+            message_dict["role"] = "developer"
 
     # NOTE: openai messages have additional arguments:
     # - function messages have `name`

{llama_index_llms_openai-0.3.10.dist-info → llama_index_llms_openai-0.3.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llama-index-llms-openai
-Version: 0.3.10
+Version: 0.3.12
 Summary: llama-index llms openai integration
 License: MIT
 Author: llama-index
@@ -12,7 +12,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: llama-index-core (>=0.12.4,<0.13.0)
-Requires-Dist: openai (>=1.57.1,<2.0.0)
+Requires-Dist: openai (>=1.58.1,<2.0.0)
 Description-Content-Type: text/markdown
 
 # LlamaIndex Llms Integration: Openai

llama_index_llms_openai-0.3.12.dist-info/RECORD

@@ -0,0 +1,7 @@
+llama_index/llms/openai/__init__.py,sha256=vm3cIBSGkBFlE77GyfyN0EhpJcnJZN95QMhPN53EkbE,148
+llama_index/llms/openai/base.py,sha256=_kW22tTABTKQ-GFRN00P5OOcJS4IeVAI3vPqbjw1D8o,35777
+llama_index/llms/openai/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+llama_index/llms/openai/utils.py,sha256=xGmeX1TbJaLjdtryETTGHR3-OSdrBm-79UxC9FlrrdE,18563
+llama_index_llms_openai-0.3.12.dist-info/METADATA,sha256=n4knQjg7oDyUJSS8nL5YUNJrLbmNd9cdG-dMEEmWe7I,3321
+llama_index_llms_openai-0.3.12.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+llama_index_llms_openai-0.3.12.dist-info/RECORD,,

llama_index_llms_openai-0.3.10.dist-info/RECORD

@@ -1,6 +0,0 @@
-llama_index/llms/openai/__init__.py,sha256=vm3cIBSGkBFlE77GyfyN0EhpJcnJZN95QMhPN53EkbE,148
-llama_index/llms/openai/base.py,sha256=_kW22tTABTKQ-GFRN00P5OOcJS4IeVAI3vPqbjw1D8o,35777
-llama_index/llms/openai/utils.py,sha256=q4qfub-sbjdfszucQCIXUyxc54_SKNM5xt84GXjKaXY,18400
-llama_index_llms_openai-0.3.10.dist-info/METADATA,sha256=cDAY7NIsXdQDMDThNsifONnFKcsmWsbCaqeJw8TwXH0,3321
-llama_index_llms_openai-0.3.10.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-llama_index_llms_openai-0.3.10.dist-info/RECORD,,