replace jiter by partial-json-parser

Author: CNSeniorious000

pyproject.toml

@@ -16,7 +16,7 @@ dependencies = [
     "sniffio",
     "cached-property; python_version < '3.8'",
     "tqdm > 4",
-    "jiter>=0.4.0, <1",
+    "partial-json-parser ~= 0.2.1.1.post4",
 ]
 requires-python = ">= 3.7.1"
 classifiers = [

src/openai/lib/streaming/chat/_completions.py

@@ -5,7 +5,7 @@
 from typing import TYPE_CHECKING, Any, Generic, Callable, Iterable, Awaitable, AsyncIterator, cast
 from typing_extensions import Self, Iterator, assert_never
 
-from jiter import from_json
+from partial_json_parser import parse_json  # type: ignore
 
 from ._types import ParsedChoiceSnapshot, ParsedChatCompletionSnapshot, ParsedChatCompletionMessageSnapshot
 from ._events import (
@@ -406,10 +406,7 @@ def _accumulate_chunk(self, chunk: ChatCompletionChunk) -> ParsedChatCompletionS
                 and not choice_snapshot.message.refusal
                 and is_given(self._rich_response_format)
             ):
-                choice_snapshot.message.parsed = from_json(
-                    bytes(choice_snapshot.message.content, "utf-8"),
-                    partial_mode=True,
-                )
+                choice_snapshot.message.parsed = parse_json(choice_snapshot.message.content)
 
             for tool_call_chunk in choice.delta.tool_calls or []:
                 tool_call_snapshot = (choice_snapshot.message.tool_calls or [])[tool_call_chunk.index]
@@ -424,10 +421,7 @@ def _accumulate_chunk(self, chunk: ChatCompletionChunk) -> ParsedChatCompletionS
                         and input_tool.get("function", {}).get("strict")
                         and tool_call_snapshot.function.arguments
                     ):
-                        tool_call_snapshot.function.parsed_arguments = from_json(
-                            bytes(tool_call_snapshot.function.arguments, "utf-8"),
-                            partial_mode=True,
-                        )
+                        tool_call_snapshot.function.parsed_arguments = parse_json(tool_call_snapshot.function.arguments)
                 elif TYPE_CHECKING:  # type: ignore[unreachable]
                     assert_never(tool_call_snapshot)