vectorvein 0.2.85__py3-none-any.whl → 0.2.86__py3-none-any.whl

vectorvein/chat_clients/openai_compatible_client.py

@@ -99,7 +99,7 @@ class OpenAICompatibleChatClient(BaseChatClient):
             return AzureOpenAI(
                 azure_endpoint=self.endpoint.api_base,
                 api_key=self.endpoint.api_key,
-                api_version="2025-03-01-preview",
+                api_version="2025-04-01-preview",
                 http_client=self.http_client,
             )
         else:
@@ -327,9 +327,7 @@ class OpenAICompatibleChatClient(BaseChatClient):
         if not max_tokens and not max_completion_tokens:
             max_output_tokens = self.model_setting.max_output_tokens
             native_multimodal = self.model_setting.native_multimodal
-            token_counts = get_message_token_counts(
-                messages=messages, tools=tools, model=self.model, native_multimodal=native_multimodal
-            )
+            token_counts = get_message_token_counts(messages=messages, tools=tools, model=self.model, native_multimodal=native_multimodal)
             if max_output_tokens is not None:
                 max_tokens = self.model_setting.context_length - token_counts - 64
                 max_tokens = min(max(max_tokens, 1), max_output_tokens)
@@ -373,9 +371,7 @@ class OpenAICompatibleChatClient(BaseChatClient):
                 extra_body=extra_body,
                 timeout=timeout,
                 stream_options=stream_options,
-                response_format=response_format
-                if response_format and self.model_setting.response_format_available
-                else OPENAI_NOT_GIVEN,
+                response_format=response_format if response_format and self.model_setting.response_format_available else OPENAI_NOT_GIVEN,
                 **tools_params,  # type: ignore
             )
 
@@ -501,9 +497,7 @@ class OpenAICompatibleChatClient(BaseChatClient):
                 extra_body=extra_body,
                 timeout=timeout,
                 stream_options=stream_options,
-                response_format=response_format
-                if response_format and self.model_setting.response_format_available
-                else OPENAI_NOT_GIVEN,
+                response_format=response_format if response_format and self.model_setting.response_format_available else OPENAI_NOT_GIVEN,
                 **tools_params,  # type: ignore
             )
 
@@ -524,10 +518,7 @@ class OpenAICompatibleChatClient(BaseChatClient):
 
             if tools:
                 if self.model_setting.function_call_available and response.choices[0].message.tool_calls:
-                    result["tool_calls"] = [
-                        {**tool_call.model_dump(), "type": "function"}
-                        for tool_call in response.choices[0].message.tool_calls
-                    ]
+                    result["tool_calls"] = [{**tool_call.model_dump(), "type": "function"} for tool_call in response.choices[0].message.tool_calls]
                 else:
                     tool_call_content_processor = ToolCallContentProcessor(result["content"])
                     tool_call_data = tool_call_content_processor.tool_calls
@@ -581,7 +572,7 @@ class AsyncOpenAICompatibleChatClient(BaseAsyncChatClient):
             return AsyncAzureOpenAI(
                 azure_endpoint=self.endpoint.api_base,
                 api_key=self.endpoint.api_key,
-                api_version="2025-03-01-preview",
+                api_version="2025-04-01-preview",
                 http_client=self.http_client,
             )
         else:
@@ -809,9 +800,7 @@ class AsyncOpenAICompatibleChatClient(BaseAsyncChatClient):
         if not max_tokens and not max_completion_tokens:
             max_output_tokens = self.model_setting.max_output_tokens
             native_multimodal = self.model_setting.native_multimodal
-            token_counts = get_message_token_counts(
-                messages=messages, tools=tools, model=self.model, native_multimodal=native_multimodal
-            )
+            token_counts = get_message_token_counts(messages=messages, tools=tools, model=self.model, native_multimodal=native_multimodal)
             if max_output_tokens is not None:
                 max_tokens = self.model_setting.context_length - token_counts - 64
                 max_tokens = min(max(max_tokens, 1), max_output_tokens)
@@ -855,9 +844,7 @@ class AsyncOpenAICompatibleChatClient(BaseAsyncChatClient):
                 extra_body=extra_body,
                 timeout=timeout,
                 stream_options=stream_options,
-                response_format=response_format
-                if response_format and self.model_setting.response_format_available
-                else OPENAI_NOT_GIVEN,
+                response_format=response_format if response_format and self.model_setting.response_format_available else OPENAI_NOT_GIVEN,
                 **tools_params,  # type: ignore
             )
 
@@ -983,9 +970,7 @@ class AsyncOpenAICompatibleChatClient(BaseAsyncChatClient):
                 extra_body=extra_body,
                 timeout=timeout,
                 stream_options=stream_options,
-                response_format=response_format
-                if response_format and self.model_setting.response_format_available
-                else OPENAI_NOT_GIVEN,
+                response_format=response_format if response_format and self.model_setting.response_format_available else OPENAI_NOT_GIVEN,
                 **tools_params,  # type: ignore
             )
 
@@ -1006,10 +991,7 @@ class AsyncOpenAICompatibleChatClient(BaseAsyncChatClient):
 
             if tools:
                 if self.model_setting.function_call_available and response.choices[0].message.tool_calls:
-                    result["tool_calls"] = [
-                        {**tool_call.model_dump(), "type": "function"}
-                        for tool_call in response.choices[0].message.tool_calls
-                    ]
+                    result["tool_calls"] = [{**tool_call.model_dump(), "type": "function"} for tool_call in response.choices[0].message.tool_calls]
                 else:
                     tool_call_content_processor = ToolCallContentProcessor(result["content"])
                     tool_call_data = tool_call_content_processor.tool_calls

vectorvein/utilities/media_processing.py

@@ -3,6 +3,7 @@
 import base64
 from io import BytesIO
 from pathlib import Path
+from typing import Literal
 from functools import cached_property
 
 import httpx
@@ -93,6 +94,66 @@ class ImageProcessor:
             if scale_factor < 0.1:
                 return img_bytes_resized
 
+    def _clear_cache(self):
+        self._cached_bytes = None
+        self._cached_base64_image = None
+
+    def convert_format(self, target_format: Literal["JPEG", "PNG", "GIF", "BMP", "WEBP", "TIFF"]):
+        """
+        转换图片的格式。
+
+        此方法将图片转换为指定的格式。支持的格式包括：'JPEG', 'PNG', 'GIF', 'BMP', 'WEBP' 等。
+
+        参数:
+        -----
+        target_format : str
+            目标格式的字符串表示，不区分大小写。例如 'png', 'jpeg', 'gif' 等。
+
+        返回:
+        -----
+        self
+            返回类的实例，支持方法链式调用。
+
+        异常:
+        -----
+        ValueError
+            如果指定的格式不被支持。
+
+        示例:
+        -----
+        >>> img.convert_format('png')  # 将图片转换为PNG格式
+        >>> img.convert_format('JPEG') # 将图片转换为JPEG格式
+        """
+
+        # 检查格式是否支持
+        supported_formats = ["JPEG", "PNG", "GIF", "BMP", "WEBP", "TIFF"]
+        if target_format not in supported_formats:
+            raise ValueError(f"不支持的格式: {target_format}。支持的格式有: {', '.join(supported_formats)}")
+
+        # 如果当前格式与目标格式相同，不执行转换
+        if self._image.format == target_format:
+            return self
+
+        # 创建一个新的字节流
+        img_bytes = BytesIO()
+
+        # 处理透明通道
+        if target_format == "JPEG" and self._image.mode in ("RGBA", "LA"):
+            # JPEG不支持透明通道，转换为RGB
+            self._image = self._image.convert("RGB")
+
+        # 保存为新格式
+        self._image.save(img_bytes, format=target_format, optimize=True)
+        img_bytes.seek(0)
+
+        # 加载新格式的图片
+        self._image = Image.open(img_bytes)
+        self._image_format = target_format
+
+        # 清除缓存
+        self._clear_cache()
+        return self
+
     @property
     def bytes(self):
         if self._cached_bytes is not None:
@@ -102,9 +163,7 @@ class ImageProcessor:
                 img_bytes = BytesIO()
 
                 # 检查图像是否有透明通道
-                has_transparency = self._image.mode in ("RGBA", "LA") or (
-                    self._image.mode == "P" and "transparency" in self._image.info
-                )
+                has_transparency = self._image.mode in ("RGBA", "LA") or (self._image.mode == "P" and "transparency" in self._image.info)
 
                 if has_transparency:
                     # 如果有透明通道，使用PNG格式

{vectorvein-0.2.85.dist-info → vectorvein-0.2.86.dist-info}/METADATA

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

{vectorvein-0.2.85.dist-info → vectorvein-0.2.86.dist-info}/RECORD

@@ -1,6 +1,6 @@
-vectorvein-0.2.85.dist-info/METADATA,sha256=KMo3arhAIBZf_7vHG883CJNEwPFYVqz1gs-DF7m5sjY,4567
-vectorvein-0.2.85.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
-vectorvein-0.2.85.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+vectorvein-0.2.86.dist-info/METADATA,sha256=ZprAIAU9UT5xHzgCmYN1CAFWjddGwQXuoes1UmLnmU0,4567
+vectorvein-0.2.86.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
+vectorvein-0.2.86.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
 vectorvein/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/api/__init__.py,sha256=lfY-XA46fgD2iIZTU0VYP8i07AwA03Egj4Qua0vUKrQ,738
 vectorvein/api/client.py,sha256=xF-leKDQzVyyy9FnIRaz0k4eElYW1XbbzeRLcpnyk90,33047
@@ -19,7 +19,7 @@ vectorvein/chat_clients/minimax_client.py,sha256=YOILWcsHsN5tihLTMbKJIyJr9TJREMI
 vectorvein/chat_clients/mistral_client.py,sha256=1aKSylzBDaLYcFnaBIL4-sXSzWmXfBeON9Q0rq-ziWw,534
 vectorvein/chat_clients/moonshot_client.py,sha256=gbu-6nGxx8uM_U2WlI4Wus881rFRotzHtMSoYOcruGU,526
 vectorvein/chat_clients/openai_client.py,sha256=Nz6tV45pWcsOupxjnsRsGTicbQNJWIZyxuJoJ5DGMpg,527
-vectorvein/chat_clients/openai_compatible_client.py,sha256=3TBQaFZD5o8GUeOnLz_bQ4gzwGANEVu-8HW4iNNcXTo,49215
+vectorvein/chat_clients/openai_compatible_client.py,sha256=0nj5DwnY8llctoxbW-UXsqVjTRiuezAgkISnV_ygnA4,48869
 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
@@ -38,7 +38,7 @@ vectorvein/types/exception.py,sha256=KtnqZ-1DstHm95SZAyZdHhkGq1bJ4A9Aw3Zfdu-VIFo
 vectorvein/types/llm_parameters.py,sha256=Fo_-_rFkBvl_dUsCZxy2xNMC9Q4qVGLzDBL9jRVbXIQ,8489
 vectorvein/types/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/types/settings.py,sha256=5wINBiT9qj3ZwDiE6Kf4r6wD3yY1bKN40xxZ7F6jEfI,4844
-vectorvein/utilities/media_processing.py,sha256=7KtbLFzOYIn1e9QTN9G6C76NH8CBlV9kfAgiRKEIeXY,6263
+vectorvein/utilities/media_processing.py,sha256=TPm73MS_a1WVqpYv0N9BrZcOK-kxWa3QHSFUI_7WgmU,8257
 vectorvein/utilities/rate_limiter.py,sha256=dwolIUVw2wP83Odqpx0AAaE77de1GzxkYDGH4tM_u_4,10300
 vectorvein/utilities/retry.py,sha256=6KFS9R2HdhqM3_9jkjD4F36ZSpEx2YNFGOVlpOsUetM,2208
 vectorvein/workflow/graph/edge.py,sha256=1ckyyjCue_PLm7P1ItUfKOy6AKkemOpZ9m1WJ8UXIHQ,1072
@@ -65,4 +65,4 @@ vectorvein/workflow/utils/analyse.py,sha256=msmvyz35UTYTwqQR5sg9H0sm1vxmGDSmep9X
 vectorvein/workflow/utils/check.py,sha256=B_NdwqIqnc7Ko2HHqFpfOmWVaAu21tPITe0szKfiZKc,11414
 vectorvein/workflow/utils/json_to_code.py,sha256=P8dhhSNgKhTnW17qXNjLO2aLdb0rA8qMAWxhObol2TU,7295
 vectorvein/workflow/utils/layout.py,sha256=j0bRD3uaXu40xCS6U6BGahBI8FrHa5MiF55GbTrZ1LM,4565
-vectorvein-0.2.85.dist-info/RECORD,,
+vectorvein-0.2.86.dist-info/RECORD,,