feat: (ocr service)add confidence score and update package path

Author: mengqian

examples/handwriting_ocr_example.py

@@ -5,6 +5,7 @@
     api_key=""
 )
 
+
 def handwriting_ocr_example():
     """
     Full Example: Submit image for recognition and wait for the result to be returned.
@@ -14,15 +15,17 @@ def handwriting_ocr_example():
     file_path = 'Your image path'
     with open(file_path, 'rb') as f:
         print("Submitting a handwriting recognition task ...")
-        response = client.file_parser.handwriting_ocr(
+        response = client.ocr.handwriting_ocr(
             file=f,
             tool_type="hand_write",
+            probability=True
         )
         print("Task created successfully. Response:")
         print(response)
 
     print("Handwriting OCR demo completed.")
 
+
 if __name__ == "__main__":
     print("=== Handwriting recognition quick demo ===\n")
-    handwriting_ocr_example()
+    handwriting_ocr_example()

src/__init__.py

@@ -24,7 +24,7 @@
     from zai.api_resource.web_search import WebSearchApi
     from zai.api_resource.web_reader import WebReaderApi
     from zai.api_resource.file_parser import FileParser
-    from zai.api_resource.file_parser import HandwritingOCR
+    from zai.api_resource.ocr import HandwritingOCR
 
 from .core import (
     NOT_GIVEN,
@@ -202,8 +202,8 @@ def file_parser(self) -> FileParser:
         return FileParser(self)
 
     @cached_property
-    def file_parser(self) -> HandwritingOCR:
-        from zai.api_resource.file_parser import HandwritingOCR
+    def ocr(self) -> HandwritingOCR:
+        from zai.api_resource.ocr import HandwritingOCR
         return HandwritingOCR(self)
 
     @property
@@ -214,7 +214,7 @@ def auth_headers(self) -> dict[str, str]:
         if self.disable_token_cache:
             return {
                 'Authorization': f'Bearer {api_key}',
-                 'x-source-channel': source_channel,
+                'x-source-channel': source_channel,
             }
         else:
             return {

src/zai/_client.py

@@ -10,34 +10,35 @@
 	Completions,
 )
 from .embeddings import Embeddings
+from .file_parser import FileParser
 from .files import Files, FilesWithRawResponse
 from .images import Images
 from .moderations import Moderations
+from .ocr import HandwritingOCR
 from .tools import Tools
 from .videos import (
 	Videos,
 )
-from .web_search import WebSearchApi
 from .web_reader import WebReaderApi
-from .file_parser import FileParser, HandwritingOCR
+from .web_search import WebSearchApi
 
 __all__ = [
-	'Videos',
-	'AsyncCompletions',
-	'Chat',
-	'Completions',
-	'Images',
-	'Embeddings',
-	'Files',
-	'FilesWithRawResponse',
-	'Batches',
-	'Tools',
-	'Assistant',
-	'Audio',
-	'Moderations',
-	'WebSearchApi',
-	'WebReaderApi',
-	'Agents',
-	'FileParser',
-	'HandwritingOCR'
+    'Videos',
+    'AsyncCompletions',
+    'Chat',
+    'Completions',
+    'Images',
+    'Embeddings',
+    'Files',
+    'FilesWithRawResponse',
+    'Batches',
+    'Tools',
+    'Assistant',
+    'Audio',
+    'Moderations',
+    'WebSearchApi',
+    'WebReaderApi',
+    'Agents',
+    'FileParser',
+    'HandwritingOCR'
 ]

src/zai/api_resource/__init__.py

@@ -1,4 +1,3 @@
 from .file_parser import FileParser
-from .handwriting_ocr import HandwritingOCR
 
-__all__ = ['FileParser', "HandwritingOCR"]
+__all__ = ['FileParser']

src/zai/api_resource/file_parser/__init__.py

@@ -0,0 +1,3 @@
+from src.zai.api_resource.ocr.handwriting_ocr import HandwritingOCR
+
+__all__ = ["HandwritingOCR"]

src/zai/api_resource/ocr/__init__.py

@@ -16,8 +16,8 @@
     extract_files,
     make_request_options
 )
-from zai.types.file_parser.handwriting_ocr_params import HandwritingOCRParams
-from zai.types.file_parser.handwriting_ocr_resp import HandwritingOCRResp
+from zai.types.ocr.handwriting_ocr_params import HandwritingOCRParams
+from zai.types.ocr.handwriting_ocr_resp import HandwritingOCRResp
 
 if TYPE_CHECKING:
     from zai._client import ZaiClient
@@ -35,7 +35,8 @@ def handwriting_ocr(
             *,
             file: FileTypes,
             tool_type: Literal["hand_write"],
-            language_type: str = None,  # optional
+            language_type: str = None,  # optional,
+            probability: bool = None,
             extra_headers: Headers | None = None,
             extra_body: Body | None = None,
             timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
@@ -49,6 +50,7 @@ def handwriting_ocr(
                 "file": file,
                 "tool_type": tool_type,
                 "language_type": language_type,
+                "probability": probability
             }
         )
         files = extract_files(cast(Mapping[str, object], body), paths=[["file"]])

src/zai/api_resource/file_parser/handwriting_ocr.py renamed to src/zai/api_resource/ocr/handwriting_ocr.py

@@ -10,3 +10,4 @@ class HandwritingOCRParams(TypedDict, total=False):
     file: FileTypes  # Required
     tool_type: Literal["hand_write"]  # Required
     language_type: str  # Optional
+    probability: bool

src/zai/types/ocr/__init__.py

@@ -13,9 +13,16 @@ class Location(BaseModel):
     height: int
 
 
+class Probability(BaseModel):
+    average: float
+    variance: float
+    min: float
+
+
 class WordsResult(BaseModel):
     location: Location
     words: str
+    probability: Probability
 
 
 class HandwritingOCRResp(BaseModel):