add:file parsing

Author: mengqian

examples/file_parsing_example.py

@@ -1,14 +1,15 @@
 from zai import ZaiClient
 import time
 import json
+import traceback
 
 client = ZaiClient(
     base_url="",
-    api_key="",
-    disable_token_cache=False
+    api_key=""
 )
 
-def file_parser_create_example(file_path,tool_type,file_type):
+
+def file_parser_create_example(file_path, tool_type, file_type):
     """
     Example: Create a file parsing task
     """
@@ -26,6 +27,7 @@ def file_parser_create_example(file_path,tool_type,file_type):
         task_id = getattr(response, "task_id", None)
         return task_id
 
+
 def file_parser_content_example(task_id, format_type="download_link"):
     """
     Example: Get file parsing result
@@ -39,28 +41,31 @@ def file_parser_content_example(task_id, format_type="download_link"):
         )
         return response
     except Exception as err:
-        print("Failed to get parsing result:", err)
+        print("Failed to get parsing result:", traceback.format_exc())
         return None
 
+
 def file_parser_complete_example():
     """
     Full Example: Submit file for parsing, then poll until result is ready
     """
     # 1. 创建解析任务
     # 请修改本地文件路径
     file_path = 'your file path'
-    task_id = file_parser_create_example(file_path=file_path,tool_type="lite",file_type="pdf")
+    task_id = file_parser_create_example(file_path=file_path, tool_type="lite", file_type="pdf")
     if not task_id:
         print("Could not submit file for parsing.")
         return
+
     ''
     # 2. 轮询获取结果
     max_wait = 60  # 最多等待1分钟
     wait_time = 0
     while wait_time < max_wait:
         print(f"Waiting {wait_time}/{max_wait} seconds before querying result...")
         # format_type = text / download_link
-        response = file_parser_content_example(task_id=task_id,format_type="download_link")
+        response = file_parser_content_example(task_id=task_id, format_type="download_link")
+
         result = response.json()
         if result.get("status") == "processing":
             print(result)
@@ -73,7 +78,6 @@ def file_parser_complete_example():
     print("File parser demo completed.")
 
 
-
 if __name__ == "__main__":
     print("=== File Parsing Quick Demo ===\n")
-    file_parser_complete_example()
+    file_parser_complete_example()

src/zai/api_resource/file_parser/file_parser.py

@@ -95,14 +95,11 @@ def content(
         if not task_id:
             raise ValueError(f"Expected a non-empty value for `task_id` but received {task_id!r}")
         extra_headers = {"Accept": "application/binary", **(extra_headers or {})}
-        _legacy_binary_response.HttpxBinaryResponseContent = self._get(
+        httpxBinaryResponseContent = self._get(
             f"/files/parser/result/{task_id}/{format_type}",
             options=make_request_options(
                 extra_headers=extra_headers, extra_body=extra_body, timeout=timeout
             ),
             cast_type=_legacy_binary_response.HttpxBinaryResponseContent,
         )
-        return _legacy_binary_response.HttpxBinaryResponseContent.response
-
-    # def json(self):
-    #     return json.loads(self.content.decode("utf-8"))
+        return httpxBinaryResponseContent.response