vectorvein 0.1.94__py3-none-any.whl → 0.1.95__py3-none-any.whl

vectorvein/chat_clients/utils.py

@@ -562,7 +562,10 @@ def transform_from_openai_message(message: ChatCompletionMessageParam, backend:
                 if isinstance(item, str):
                     formatted_content.append({"type": "text", "text": item})
                 elif isinstance(item, dict) and "type" in item:
-                    formatted_content.append(item)
+                    if item["type"] == "image_url":
+                        formatted_content.append(format_image_message(item["image_url"]["url"], backend))
+                    else:
+                        formatted_content.append(item)
             return {"role": role, "content": formatted_content}
         else:
             return {"role": role, "content": content}

vectorvein/utilities/media_processing.py

@@ -32,11 +32,18 @@ class ImageProcessor:
         self._cached_base64_image = None
 
     def _load_image(self):
-        if not self.is_local and isinstance(self.image_source, str):
-            image_url = self.image_source
-            response = httpx.get(image_url)
-            return Image.open(BytesIO(response.content))
-        elif isinstance(self.image_source, (Path, str)):
+        if isinstance(self.image_source, str):
+            if self.image_source.startswith(("data:image/", "data:application/octet-stream;base64,")):
+                base64_data = self.image_source.split(",")[1]
+                image_data = base64.b64decode(base64_data)
+                return Image.open(BytesIO(image_data))
+            elif not self.is_local:
+                image_url = self.image_source
+                response = httpx.get(image_url)
+                return Image.open(BytesIO(response.content))
+            else:
+                return Image.open(self.image_source)
+        elif isinstance(self.image_source, Path):
             return Image.open(self.image_source)
         elif isinstance(self.image_source, Image.Image):
             return self.image_source

{vectorvein-0.1.94.dist-info → vectorvein-0.1.95.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vectorvein
-Version: 0.1.94
+Version: 0.1.95
 Summary: VectorVein python SDK
 Author-Email: Anderson <andersonby@163.com>
 License: MIT

{vectorvein-0.1.94.dist-info → vectorvein-0.1.95.dist-info}/RECORD

@@ -1,6 +1,6 @@
-vectorvein-0.1.94.dist-info/METADATA,sha256=GOy1lUZfBXtI3RqRGmAfpT6ju1I1gG94tytmlTbyNt8,775
-vectorvein-0.1.94.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-vectorvein-0.1.94.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+vectorvein-0.1.95.dist-info/METADATA,sha256=9jaFqXZu5L1clFveApXDGA8IQEsC72Q7pyGibiFBAfU,775
+vectorvein-0.1.95.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+vectorvein-0.1.95.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
 vectorvein/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/chat_clients/__init__.py,sha256=omQuG4PRRPNflSAgtdU--rwsWG6vMpwMEyIGZyFVHVQ,18596
 vectorvein/chat_clients/anthropic_client.py,sha256=Zk6X1feIvv7Az5dgyipJXbm9TkgWgpFghSTxLiXKKA8,38405
@@ -18,7 +18,7 @@ vectorvein/chat_clients/openai_compatible_client.py,sha256=H20SXdvXalf7_jaashff_
 vectorvein/chat_clients/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/chat_clients/qwen_client.py,sha256=-ryh-m9PgsO0fc4ulcCmPTy1155J8YUy15uPoJQOHA0,513
 vectorvein/chat_clients/stepfun_client.py,sha256=zsD2W5ahmR4DD9cqQTXmJr3txrGuvxbRWhFlRdwNijI,519
-vectorvein/chat_clients/utils.py,sha256=Xt-9Dm564SY0jRO4N9ArDcFDKZIdfh9pYMxshLRMfUc,24853
+vectorvein/chat_clients/utils.py,sha256=r8tlt-suUjISSRjc_tCCcBYmQT3zng1YFVnQgeAG1Co,25041
 vectorvein/chat_clients/xai_client.py,sha256=eLFJJrNRJ-ni3DpshODcr3S1EJQLbhVwxyO1E54LaqM,491
 vectorvein/chat_clients/yi_client.py,sha256=RNf4CRuPJfixrwLZ3-DEc3t25QDe1mvZeb9sku2f8Bc,484
 vectorvein/chat_clients/zhipuai_client.py,sha256=Ys5DSeLCuedaDXr3PfG1EW2zKXopt-awO2IylWSwY0s,519
@@ -31,7 +31,7 @@ vectorvein/types/enums.py,sha256=7KTJSVtQueImmbr1fSwv3rQVtc0RyMWXJmoE2tDOaso,166
 vectorvein/types/exception.py,sha256=gnW4GnJ76jND6UGnodk9xmqkcbeS7Cz2rvncA2HpD5E,69
 vectorvein/types/llm_parameters.py,sha256=9KJ8z_7xVa0JsV0YPxRKz5n8Eb8PfvB9Y89e6ahjmjw,5989
 vectorvein/types/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vectorvein/utilities/media_processing.py,sha256=CTRq-lGlFkFgP_FSRhNwF_qUgmOrXPf2_1Ok9HY42_g,5887
+vectorvein/utilities/media_processing.py,sha256=7KtbLFzOYIn1e9QTN9G6C76NH8CBlV9kfAgiRKEIeXY,6263
 vectorvein/utilities/rate_limiter.py,sha256=dwolIUVw2wP83Odqpx0AAaE77de1GzxkYDGH4tM_u_4,10300
 vectorvein/utilities/retry.py,sha256=6KFS9R2HdhqM3_9jkjD4F36ZSpEx2YNFGOVlpOsUetM,2208
 vectorvein/workflow/graph/edge.py,sha256=xLZEJmBjAfVB53cd7CuRcKhgE6QqXv9nz32wJn8cmyk,1064
@@ -55,4 +55,4 @@ vectorvein/workflow/nodes/vector_db.py,sha256=t6I17q6iR3yQreiDHpRrksMdWDPIvgqJs0
 vectorvein/workflow/nodes/video_generation.py,sha256=qmdg-t_idpxq1veukd-jv_ChICMOoInKxprV9Z4Vi2w,4118
 vectorvein/workflow/nodes/web_crawlers.py,sha256=LsqomfXfqrXfHJDO1cl0Ox48f4St7X_SL12DSbAMSOw,5415
 vectorvein/workflow/utils/json_to_code.py,sha256=F7dhDy8kGc8ndOeihGLRLGFGlquoxVlb02ENtxnQ0C8,5914
-vectorvein-0.1.94.dist-info/RECORD,,
+vectorvein-0.1.95.dist-info/RECORD,,