vectorvein 0.1.94__tar.gz → 0.1.95__tar.gz

{vectorvein-0.1.94 → vectorvein-0.1.95}/PKG-INFO

@@ -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 → vectorvein-0.1.95}/pyproject.toml

@@ -17,7 +17,7 @@ description = "VectorVein python SDK"
 name = "vectorvein"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "0.1.94"
+version = "0.1.95"
 
 [project.license]
 text = "MIT"

{vectorvein-0.1.94 → vectorvein-0.1.95}/src/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-0.1.94 → vectorvein-0.1.95}/src/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