feat: support json schema

Author: 潘婉宁

setup.py

@@ -29,8 +29,7 @@
             "httpx>=0.23.0, <1",
             "anyio>=3.5.0, <5",
             "cached-property; python_version < '3.8'",
-            "cryptography>=43.0.3, <43.0.4",
-            "jiter>=0.4.0, <1"
+            "cryptography>=43.0.3, <43.0.4"
         ]
     },
 )

volcenginesdkarkruntime/common/streaming/chat/_completions.py

@@ -14,7 +14,20 @@
 )
 from typing_extensions import Self, Iterator, assert_never
 
-from jiter import from_json
+try:
+    from jiter import from_json
+except ImportError:
+    import json
+    from typing import Dict, Any
+
+    def from_json(json_str: bytes, partial_mode: bool) -> Dict[str, Any]:
+        try:
+            return json.loads(json_str)
+        except json.JSONDecodeError as e:
+            if partial_mode:
+                return {}
+            raise e
+
 
 from ._types import (
     ParsedChoiceSnapshot,