报文解析错误

Author: glide-the

.gitignore

@@ -15,3 +15,4 @@ eggs/
 *.egg-info/
 test/
 examples/
+.pypirc

tests/test_chat.py

@@ -53,7 +53,7 @@ def encode_image(image_path):
 
 def test_completions_vis():
     client = ZhipuAI()  # 填写您自己的APIKey
-    base64_image  = encode_image("img/MetaGLM.png")
+    base64_image = encode_image("img/MetaGLM.png")
     response = client.chat.completions.create(
         model="glm-4v",  # 填写需要调用的模型名称
         extra_body={"temperature": 0.5, "max_tokens": 50},
@@ -116,6 +116,5 @@ def test_retrieve_completion_result():
     print(response)
 
 
-
 if __name__ == "__main__":
     test_completions_vis()

zhipuai/api_resource/chat/completions.py

@@ -99,17 +99,19 @@ def create(
             stream_cls=_stream_cls,
         )
 
-    def _drop_prefix_image_data(self, content: List) -> List:
+    def _drop_prefix_image_data(self, content: Union[str,List[dict]]) -> Union[str,List[dict]]:
         """
         删除 ;base64, 前缀
         :param image_data:
         :return:
         """
-        for data in content:
-            if data.get('type') == 'image_url':
-                image_data = data.get("image_url").get("url")
-                if image_data.startswith("data:image/"):
-                    image_data = image_data.split("base64,")[-1]
-                    data["image_url"]["url"] = image_data
+        if isinstance(content, List):
+            for data in content:
+                if data.get('type') == 'image_url':
+                    image_data = data.get("image_url").get("url")
+                    if image_data.startswith("data:image/"):
+                        image_data = image_data.split("base64,")[-1]
+                        data["image_url"]["url"] = image_data
 
         return content
+