Use public import for ChatCompletionStreamState (#398)

* Bump openai req to 1.56.0 * Use public import for ChatCompletionStreamState
jackmpcollins · Jan 6, 2025 · fd0d03e · fd0d03e
1 parent 635fabc
commit fd0d03e
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -7,7 +7,7 @@ requires-python = ">=3.10"
 dependencies = [
     "filetype>=1.2.0",
     "logfire-api>=0.1.0",
-    "openai>=1.40.0",
+    "openai>=1.56.0",
     "pydantic>=2.10.0",
     "pydantic-settings>=2.0.0",
     "typing-extensions>=4.5.0",

diff --git a/src/magentic/chat_model/litellm_chat_model.py b/src/magentic/chat_model/litellm_chat_model.py
@@ -2,7 +2,7 @@
 from typing import Any, Literal, TypeVar, cast, overload
 
 import openai
-from openai.lib.streaming.chat._completions import ChatCompletionStreamState
+from openai.lib.streaming.chat import ChatCompletionStreamState
 from openai.types.chat import ChatCompletionNamedToolChoiceParam
 
 from magentic._parsing import contains_string_type

diff --git a/src/magentic/chat_model/openai_chat_model.py b/src/magentic/chat_model/openai_chat_model.py
@@ -4,7 +4,7 @@
 from typing import Any, Generic, Literal, TypeVar, cast, overload
 
 import openai
-from openai.lib.streaming.chat._completions import ChatCompletionStreamState
+from openai.lib.streaming.chat import ChatCompletionStreamState
 from openai.types.chat import (
     ChatCompletionChunk,
     ChatCompletionContentPartParam,

diff --git a/uv.lock b/uv.lock