vectorvein 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

vectorvein/chat_clients/__init__.py

@@ -17,7 +17,7 @@ from .deepseek_client import DeepSeekChatClient, AsyncDeepSeekChatClient
 from ..types import defaults as defs
 from ..types.enums import BackendType, ContextLengthControlType
 from .anthropic_client import AnthropicChatClient, AsyncAnthropicChatClient
-from .utils import format_messages
+from .utils import format_messages, get_token_counts, ToolCallContentProcessor
 
 
 BackendMap = {
@@ -103,8 +103,10 @@ def create_async_chat_client(
 
 
 __all__ = [
+    "BackendType",
+    "format_messages",
+    "get_token_counts",
     "create_chat_client",
     "create_async_chat_client",
-    "format_messages",
-    "BackendType",
+    "ToolCallContentProcessor",
 ]

vectorvein/chat_clients/openai_compatible_client.py

@@ -11,9 +11,8 @@ from openai import OpenAI, AsyncOpenAI, AzureOpenAI, AsyncAzureOpenAI
 
 from .base_client import BaseChatClient, BaseAsyncChatClient
 from .utils import (
-    tool_use_re,
     cutoff_messages,
-    extract_tool_calls,
+    ToolCallContentProcessor,
     generate_tool_use_system_prompt,
 )
 from ..settings import settings
@@ -128,7 +127,7 @@ class OpenAICompatibleChatClient(BaseChatClient):
                         message = chunk.choices[0].delta.model_dump()
                         full_content += message["content"] if message["content"] else ""
                         if tools:
-                            tool_call_data = extract_tool_calls(full_content)
+                            tool_call_data = ToolCallContentProcessor(result["content"]).tool_calls
                             if tool_call_data:
                                 message["tool_calls"] = tool_call_data["tool_calls"]
                         if full_content in ("<", "<|", "<|▶", "<|▶|") or full_content.startswith("<|▶|>"):
@@ -151,10 +150,11 @@ class OpenAICompatibleChatClient(BaseChatClient):
                         tool_call.model_dump() for tool_call in response.choices[0].message.tool_calls
                     ]
                 else:
-                    tool_call_data = extract_tool_calls(result["content"])
+                    tool_call_content_processor = ToolCallContentProcessor(result["content"])
+                    tool_call_data = tool_call_content_processor.tool_calls
                     if tool_call_data:
                         result["tool_calls"] = tool_call_data["tool_calls"]
-                        result["content"] = tool_use_re.sub("", result["content"])
+                        result["content"] = tool_call_content_processor.non_tool_content
             return result
 
 
@@ -265,7 +265,7 @@ class AsyncOpenAICompatibleChatClient(BaseAsyncChatClient):
                         message = chunk.choices[0].delta.model_dump()
                         full_content += message["content"] if message["content"] else ""
                         if tools:
-                            tool_call_data = extract_tool_calls(full_content)
+                            tool_call_data = ToolCallContentProcessor(result["content"]).tool_calls
                             if tool_call_data:
                                 message["tool_calls"] = tool_call_data["tool_calls"]
                         if full_content in ("<", "<|", "<|▶", "<|▶|") or full_content.startswith("<|▶|>"):
@@ -288,8 +288,9 @@ class AsyncOpenAICompatibleChatClient(BaseAsyncChatClient):
                         tool_call.model_dump() for tool_call in response.choices[0].message.tool_calls
                     ]
                 else:
-                    tool_call_data = extract_tool_calls(result["content"])
+                    tool_call_content_processor = ToolCallContentProcessor(result["content"])
+                    tool_call_data = tool_call_content_processor.tool_calls
                     if tool_call_data:
                         result["tool_calls"] = tool_call_data["tool_calls"]
-                        result["content"] = tool_use_re.sub("", result["content"])
+                        result["content"] = tool_call_content_processor.non_tool_content
             return result

vectorvein/chat_clients/utils.py

@@ -12,7 +12,49 @@ from ..utilities.media_processing import ImageProcessor
 chatgpt_encoding = tiktoken.encoding_for_model("gpt-3.5-turbo")
 gpt_4o_encoding = tiktoken.encoding_for_model("gpt-4o")
 
-tool_use_re = re.compile(r"<\|▶\|>(.*?)<\|◀\|>", re.DOTALL)
+
+class ToolCallContentProcessor:
+    tool_use_re = re.compile(r"<\|▶\|>(.*?)<\|◀\|>", re.DOTALL)
+
+    def __init__(self, content: str):
+        self.content = content
+
+    @property
+    def non_tool_content(self):
+        return re.sub(self.tool_use_re, "", self.content).strip()
+
+    @property
+    def tool_calls(self):
+        if "<|▶|>" not in self.content or "<|◀|>" not in self.content:
+            return {}
+        tool_calls_matches = re.findall(self.tool_use_re, self.content)
+        if tool_calls_matches:
+            tool_call_data = {}
+            for match in tool_calls_matches:
+                try:
+                    tool_call_data = json.loads(match)
+                except json.JSONDecodeError:
+                    print(f"Failed to parse tool call data:\nContent: {self.content}\nMatch: {match}")
+
+            if not tool_call_data:
+                return {}
+
+            arguments = json.dumps(tool_call_data["arguments"], ensure_ascii=False)
+            return {
+                "tool_calls": [
+                    {
+                        "index": 0,
+                        "id": "fc1",
+                        "function": {
+                            "arguments": arguments,
+                            "name": tool_call_data["name"],
+                        },
+                        "type": "function",
+                    }
+                ]
+            }
+        else:
+            return {}
 
 
 def get_assistant_role_key(backend: BackendType) -> str:
@@ -48,7 +90,7 @@ def convert_type(value, value_type):
         return value  # 如果类型未知，返回原始值
 
 
-def get_token_counts(text: str, model: str = "") -> int:
+def get_token_counts(text: str | dict, model: str = "") -> int:
     if not isinstance(text, str):
         text = str(text)
     if model == "gpt-3.5-turbo":
@@ -178,7 +220,7 @@ def format_workflow_messages(message, content, backend):
     formatted_messages = []
 
     # 工具调用消息
-    if backend in (BackendType.OpenAI, BackendType.ZhiPuAI, BackendType.Mistral):
+    if backend in (BackendType.OpenAI, BackendType.ZhiPuAI, BackendType.Mistral, BackendType.Yi):
         tool_call_message = {
             "content": None,
             "role": "assistant",
@@ -235,7 +277,7 @@ def format_workflow_messages(message, content, backend):
     formatted_messages.append(tool_call_message)
 
     # 工具调用结果消息
-    if backend in (BackendType.OpenAI, BackendType.ZhiPuAI, BackendType.Mistral):
+    if backend in (BackendType.OpenAI, BackendType.ZhiPuAI, BackendType.Mistral, BackendType.Yi):
         tool_call_result_message = {
             "role": "tool",
             "tool_call_id": message["metadata"]["selected_workflow"]["tool_call_id"],
@@ -321,12 +363,14 @@ def format_openai_message(message, backend):
 
 
 def format_messages(
-    messages: list, backend: BackendType = BackendType.OpenAI, native_multimodal: bool = False
+    messages: list,
+    backend: BackendType = BackendType.OpenAI,
+    native_multimodal: bool = False,
 ) -> list:
     """将 VectorVein 和 OpenAI 的 Message 序列化后的格式转换为不同模型支持的格式
 
     Args:
-        messages (list): VectorVein 或 OpenAI messages list.
+        messages (list): VectorVein Or OpenAI messages list.
         backend (str, optional): Messages format target backend. Defaults to BackendType.OpenAI.
         native_multimodal (bool, optional): Use native multimodal ability. Defaults to False.
 
@@ -396,189 +440,6 @@ def format_text_message(content, role, attachments, backend, native_multimodal):
             return {"role": role, "content": content}
 
 
-def format_messages_v1(
-    messages: list, backend: BackendType = BackendType.OpenAI, native_multimodal: bool = False
-) -> list:
-    """将 VectorVein 的 Message 序列化后的格式转换为不同模型支持的格式
-
-    Args:
-        messages (list): VectorVein messages list.
-        backend (str, optional): Messages format target backend. Defaults to BackendType.OpenAI.
-        native_multimodal (bool, optional): Use native multimodal ability. Defaults to False.
-
-    Returns:
-        list: _description_
-    """
-
-    backend = backend.lower()
-    formatted_messages = []
-    for message in messages:
-        content = message["content"]["text"]
-        if message["content_type"] == "TXT":
-            role = "user" if message["author_type"] == "U" else get_assistant_role_key(backend)
-            if not message.get("attachments"):
-                if backend == BackendType.Gemini:
-                    formatted_message = {"role": role, "parts": [{"text": content}]}
-                else:
-                    formatted_message = {"role": role, "content": content}
-                formatted_messages.append(formatted_message)
-                continue
-
-            images_extensions = ("jpg", "jpeg", "png", "bmp")
-            has_images = any(attachment.lower().endswith(images_extensions) for attachment in message["attachments"])
-
-            content += "\n# Attachments:\n"
-            content += "\n".join([f"- {attachment}" for attachment in message["attachments"]])
-
-            if native_multimodal and has_images:
-                if backend == BackendType.Gemini:
-                    parts = [{"text": content}]
-                    for attachment in message["attachments"]:
-                        if attachment.lower().endswith(images_extensions):
-                            parts.append(format_image_message(image=attachment, backend=backend))
-                    formatted_message = {"role": role, "parts": parts}
-                else:
-                    formatted_message = {
-                        "role": role,
-                        "content": [
-                            {"type": "text", "text": content},
-                            *[
-                                format_image_message(image=attachment, backend=backend)
-                                for attachment in message["attachments"]
-                                if attachment.lower().endswith(images_extensions)
-                            ],
-                        ],
-                    }
-                formatted_messages.append(formatted_message)
-            else:
-                if backend == BackendType.Gemini:
-                    formatted_message = {"role": role, "parts": [{"text": content}]}
-                else:
-                    formatted_message = {"role": role, "content": content}
-                formatted_messages.append(formatted_message)
-        elif message["content_type"] == "WKF" and message["status"] in ("S", "R"):
-            # TODO: 目前只考虑单个 tool_call 的情况
-            if backend in (BackendType.OpenAI, BackendType.ZhiPuAI, BackendType.Mistral):
-                tool_call_message = {
-                    "content": None,
-                    "role": "assistant",
-                    "tool_calls": [
-                        {
-                            "id": message["metadata"]["selected_workflow"]["tool_call_id"],
-                            "type": "function",
-                            "function": {
-                                "name": message["metadata"]["selected_workflow"]["function_name"],
-                                "arguments": json.dumps(message["metadata"]["selected_workflow"]["params"]),
-                            },
-                        }
-                    ],
-                }
-            elif backend == BackendType.Anthropic:
-                tool_call_message = {
-                    "role": "assistant",
-                    "content": [
-                        {
-                            "type": "tool_use",
-                            "id": message["metadata"]["selected_workflow"]["tool_call_id"],
-                            "name": message["metadata"]["selected_workflow"]["function_name"],
-                            "input": message["metadata"]["selected_workflow"]["params"],
-                        },
-                    ],
-                }
-                if content:
-                    tool_call_message["content"].insert(
-                        0,
-                        {
-                            "type": "text",
-                            "text": content,
-                        },
-                    )
-            elif backend == BackendType.Gemini:
-                tool_call_message = {
-                    "role": "model",
-                    "parts": [
-                        {
-                            "functionCall": {
-                                "name": message["metadata"]["selected_workflow"]["function_name"],
-                                "args": message["metadata"]["selected_workflow"]["params"],
-                            }
-                        },
-                    ],
-                }
-                if content:
-                    tool_call_message["parts"].insert(
-                        0,
-                        {
-                            "text": content,
-                        },
-                    )
-            else:
-                tool_call_message = {
-                    "content": json.dumps(
-                        {
-                            "name": message["metadata"]["selected_workflow"]["function_name"],
-                            "arguments": json.dumps(message["metadata"]["selected_workflow"]["params"]),
-                        },
-                        ensure_ascii=False,
-                    ),
-                    "role": "assistant",
-                }
-            formatted_messages.append(tool_call_message)
-
-            if backend in (BackendType.OpenAI, BackendType.ZhiPuAI, BackendType.Mistral):
-                tool_call_result_message = {
-                    "role": "tool",
-                    "tool_call_id": message["metadata"]["selected_workflow"]["tool_call_id"],
-                    "name": message["metadata"]["selected_workflow"]["function_name"],
-                    "content": message["metadata"].get("workflow_result", ""),
-                }
-            elif backend == BackendType.Anthropic:
-                tool_call_result_message = {
-                    "role": "user",
-                    "content": [
-                        {
-                            "type": "tool_result",
-                            "tool_use_id": message["metadata"]["selected_workflow"]["tool_call_id"],
-                            "content": message["metadata"].get("workflow_result", ""),
-                        }
-                    ],
-                }
-            elif backend == BackendType.Gemini:
-                tool_call_result_message = {
-                    "role": "function",
-                    "parts": [
-                        {
-                            "functionResponse": {
-                                "name": message["metadata"]["selected_workflow"]["function_name"],
-                                "response": {
-                                    "name": message["metadata"]["selected_workflow"]["function_name"],
-                                    "content": message["metadata"].get("workflow_result", ""),
-                                },
-                            }
-                        }
-                    ],
-                }
-            else:
-                tool_call_result_message = {
-                    "role": "user",
-                    "content": json.dumps(
-                        {
-                            "function": message["metadata"]["selected_workflow"]["function_name"],
-                            "result": message["metadata"].get("workflow_result", ""),
-                        },
-                        ensure_ascii=False,
-                    ),
-                }
-            formatted_messages.append(tool_call_result_message)
-
-            if content and backend not in (BackendType.Mistral, BackendType.Anthropic, BackendType.Gemini):
-                formatted_messages.append({"role": "assistant", "content": content})
-        else:
-            continue
-
-    return formatted_messages
-
-
 def generate_tool_use_system_prompt(tools: list, format_type: str = "json") -> str:
     if format_type == "json":
         return (
@@ -600,36 +461,3 @@ def generate_tool_use_system_prompt(tools: list, format_type: str = "json") -> s
             "## Output format\n<|▶|><invoke><tool_name>[function name:str]</tool_name><parameters><parameter_1_name>[parameter_1_value]</parameter_1_name><parameter_2_name>[parameter_2_value]</parameter_2_name>...</parameters></invoke><|◀|>\n\n"
             "## Example output\n<|▶|><invoke><tool_name>calculator</tool_name><parameters><first_operand>1984135</first_operand><second_operand>9343116</second_operand><operator>*</operator></parameters></invoke><|◀|>"
         )
-
-
-def extract_tool_calls(content: str) -> dict:
-    if "<|▶|>" not in content or "<|◀|>" not in content:
-        return {}
-    tool_calls_matches = tool_use_re.findall(content)
-    if tool_calls_matches:
-        tool_call_data = {}
-        for match in tool_calls_matches:
-            try:
-                tool_call_data = json.loads(match)
-            except json.JSONDecodeError:
-                print(f"Failed to parse tool call data:\nContent: {content}\nMatch: {match}")
-
-        if not tool_call_data:
-            return {}
-
-        arguments = json.dumps(tool_call_data["arguments"], ensure_ascii=False)
-        return {
-            "tool_calls": [
-                {
-                    "index": 0,
-                    "id": "fc1",
-                    "function": {
-                        "arguments": arguments,
-                        "name": tool_call_data["name"],
-                    },
-                    "type": "function",
-                }
-            ]
-        }
-    else:
-        return {}

{vectorvein-0.1.1.dist-info → vectorvein-0.1.3.dist-info}/METADATA

@@ -1,10 +1,10 @@
 Metadata-Version: 2.1
 Name: vectorvein
-Version: 0.1.1
+Version: 0.1.3
 Summary: Default template for PDM package
 Author-Email: Anderson <andersonby@163.com>
 License: MIT
-Requires-Python: >=3.8
+Requires-Python: >=3.10
 Requires-Dist: openai>=1.37.1
 Requires-Dist: tiktoken>=0.7.0
 Requires-Dist: httpx>=0.27.0

{vectorvein-0.1.1.dist-info → vectorvein-0.1.3.dist-info}/RECORD

@@ -1,7 +1,7 @@
-vectorvein-0.1.1.dist-info/METADATA,sha256=pWX5z4EA3jmbcbo3DhmSx8JIGVKL2w0jKGaDb1XXdg0,422
-vectorvein-0.1.1.dist-info/WHEEL,sha256=rSwsxJWe3vzyR5HCwjWXQruDgschpei4h_giTm0dJVE,90
+vectorvein-0.1.3.dist-info/METADATA,sha256=C0hbnXUDDO_cTFk7pWq3BBEV3bYZVB96KO1ZvQYn0ac,423
+vectorvein-0.1.3.dist-info/WHEEL,sha256=rSwsxJWe3vzyR5HCwjWXQruDgschpei4h_giTm0dJVE,90
 vectorvein/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vectorvein/chat_clients/__init__.py,sha256=H7QU9Tv2mpJe1Sv5tcw3LM6ulC9tR4be7RmX3B7FECE,3874
+vectorvein/chat_clients/__init__.py,sha256=kjXCFj4tGvD-qpiurFdc6gJWJSaqClr08iWYY3GVdJQ,3974
 vectorvein/chat_clients/anthropic_client.py,sha256=sH6tZHYeZ4WncHtALvbx7axmY8YFOn8JDJKgTgIpqbw,18025
 vectorvein/chat_clients/base_client.py,sha256=za03oViP1hfAE9NAtuAMsms7Chp8-Nm0s8nR1rX9OGo,2705
 vectorvein/chat_clients/deepseek_client.py,sha256=3qWu01NlJAP2N-Ff62d5-CZXZitlizE1fzb20LNetig,526
@@ -12,9 +12,9 @@ vectorvein/chat_clients/minimax_client.py,sha256=x6PCSV8uYz_khfAL87q0IrbFEh6hWq5
 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=-cDw1Axf9wHYSj2EOw8vZnATm9yVOPS4_Z31U31d0BI,11490
+vectorvein/chat_clients/openai_compatible_client.py,sha256=wtFstRAbVuhxHLxpihaiAoJbDd5-h35yVnhurCMYZRM,11726
 vectorvein/chat_clients/qwen_client.py,sha256=-ryh-m9PgsO0fc4ulcCmPTy1155J8YUy15uPoJQOHA0,513
-vectorvein/chat_clients/utils.py,sha256=nz1hpK6uOFRdCcWrNrctkCqbeXSgOogDtOyDkS47jDU,25591
+vectorvein/chat_clients/utils.py,sha256=DGIh9RqRq61krLX4bLVlqUg7H7MmCyCzBdysRXvNcHY,17634
 vectorvein/chat_clients/yi_client.py,sha256=RNf4CRuPJfixrwLZ3-DEc3t25QDe1mvZeb9sku2f8Bc,484
 vectorvein/chat_clients/zhipuai_client.py,sha256=Ys5DSeLCuedaDXr3PfG1EW2zKXopt-awO2IylWSwY0s,519
 vectorvein/settings/__init__.py,sha256=1__ILDpBh25S43d32knqLBe05tF4AQwT0ZALdBHw1OY,3396
@@ -22,4 +22,4 @@ vectorvein/types/defaults.py,sha256=Mg-Mj3_eBzKZn1N8x1V2GqyaYgLD13i-NdSYdQC28X4,
 vectorvein/types/enums.py,sha256=vzOenCnRlFXBwPh-lfFhjGfM-6yfDj7wZColHODqocI,1550
 vectorvein/types/llm_parameters.py,sha256=88CwRQ3oCY35nPlCCcLk9T3jofKqT876CexXiB7ExQ8,3101
 vectorvein/utilities/media_processing.py,sha256=BujciRmw1GMmc3ELRvafL8STcy6r5b2rVnh27-uA7so,2256
-vectorvein-0.1.1.dist-info/RECORD,,
+vectorvein-0.1.3.dist-info/RECORD,,