vectorvein 0.2.88__tar.gz → 0.2.90__tar.gz

{vectorvein-0.2.88 → vectorvein-0.2.90}/PKG-INFO

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

{vectorvein-0.2.88 → vectorvein-0.2.90}/pyproject.toml

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

{vectorvein-0.2.88 → vectorvein-0.2.90}/src/vectorvein/chat_clients/utils.py

@@ -4,7 +4,7 @@ import re
 import json
 import warnings
 from math import ceil
-from typing import Iterable
+from typing import Iterable, cast
 
 import httpx
 import tiktoken
@@ -18,6 +18,9 @@ from ..types.llm_parameters import (
     NotGiven,
     NOT_GIVEN,
     ToolParam,
+    VectorVeinMessage,
+    VectorVeinTextMessage,
+    VectorVeinWorkflowMessage,
 )
 
 
@@ -435,7 +438,7 @@ def format_image_message(image: str, backend: BackendType = BackendType.OpenAI,
             return {"type": "image_url", "image_url": {"url": image}}
 
 
-def format_workflow_messages(message: dict, content: str, backend: BackendType):
+def format_workflow_messages(message: VectorVeinWorkflowMessage, content: str, backend: BackendType):
     formatted_messages = []
 
     # 工具调用消息
@@ -626,7 +629,7 @@ def transform_from_openai_message(
 
 
 def format_messages(
-    messages: list,
+    messages: Iterable[ChatCompletionMessageParam | VectorVeinMessage],
     backend: BackendType = BackendType.OpenAI,
     native_multimodal: bool = False,
     function_call_available: bool = False,
@@ -652,16 +655,27 @@ def format_messages(
 
     for message in messages:
         if is_vectorvein_message(message):
+            message = cast(VectorVeinMessage, message)
+            match message["content_type"]:
+                case "TXT":
+                    message = cast(VectorVeinTextMessage, message)
+                case "WKF":
+                    message = cast(VectorVeinWorkflowMessage, message)
+                case _:
+                    raise ValueError(f"Unsupported message type: {message['content_type']}")
             # 处理 VectorVein 格式的消息
             content = message["content"]["text"]
             if message["content_type"] == "TXT":
+                message = cast(VectorVeinTextMessage, message)
                 role = "user" if message["author_type"] == "U" else "assistant"
                 formatted_message = format_text_message(content, role, message.get("attachments", []), backend, native_multimodal, process_image)
                 formatted_messages.append(formatted_message)
             elif message["content_type"] == "WKF" and message["status"] in ("S", "R"):
+                message = cast(VectorVeinWorkflowMessage, message)
                 formatted_messages.extend(format_workflow_messages(message, content, backend))
         else:
             # 处理 OpenAI 格式的消息
+            message = cast(ChatCompletionMessageParam, message)
             formatted_message = transform_from_openai_message(message, backend, native_multimodal, function_call_available, process_image)
             formatted_messages.append(formatted_message)
 

{vectorvein-0.2.88 → vectorvein-0.2.90}/src/vectorvein/types/__init__.py

@@ -44,12 +44,18 @@ from .llm_parameters import (
     Usage,
     ChatCompletionMessage,
     ChatCompletionDeltaMessage,
+    ChatCompletionStreamOptionsParam,
     NotGiven,
     NOT_GIVEN,
     OpenAIToolParam,
     ToolParam,
     Tools,
     ToolChoice,
+    AnthropicToolParam,
+    AnthropicToolChoice,
+    OPENAI_NOT_GIVEN,
+    ANTHROPIC_NOT_GIVEN,
+    VectorVeinMessage,
 )
 from .settings import (
     RedisConfigDict,
@@ -117,6 +123,12 @@ __all__ = [
     "Usage",
     "ChatCompletionMessage",
     "ChatCompletionDeltaMessage",
+    "ChatCompletionStreamOptionsParam",
+    "AnthropicToolParam",
+    "AnthropicToolChoice",
+    "OPENAI_NOT_GIVEN",
+    "ANTHROPIC_NOT_GIVEN",
+    "VectorVeinMessage",
     "NotGiven",
     "NOT_GIVEN",
     "OpenAIToolParam",

{vectorvein-0.2.88 → vectorvein-0.2.90}/src/vectorvein/types/llm_parameters.py

@@ -1,6 +1,7 @@
 # @Author: Bi Ying
 # @Date:   2024-07-26 23:48:04
 from typing import List, Dict, Optional, Union, Iterable, Literal
+from typing_extensions import TypedDict, NotRequired
 
 import httpx
 from pydantic import BaseModel, Field
@@ -26,6 +27,33 @@ from . import defaults as defs
 from .settings import EndpointOptionDict
 
 
+class VectorVeinTextMessage(TypedDict):
+    mid: NotRequired[str]
+    author_type: Literal["U", "A"]  # "U" for user, "A" for assistant
+    content_type: Literal["TXT"]
+    status: NotRequired[str]
+    create_time: NotRequired[int]
+    update_time: NotRequired[int]
+    metadata: NotRequired[dict]
+    content: Dict[str, str]
+    attachments: NotRequired[List[str]]
+
+
+class VectorVeinWorkflowMessage(TypedDict):
+    mid: NotRequired[str]
+    author_type: Literal["U", "A"]  # "U" for user, "A" for assistant
+    content_type: Literal["WKF"]
+    status: str
+    create_time: NotRequired[int]
+    update_time: NotRequired[int]
+    metadata: dict
+    content: Dict[str, str]
+    attachments: NotRequired[List[str]]
+
+
+VectorVeinMessage = Union[VectorVeinTextMessage, VectorVeinWorkflowMessage]
+
+
 class EndpointSetting(BaseModel):
     id: str = Field(..., description="The id of the endpoint.")
     region: Optional[str] = Field(None, description="The region for the endpoint.")
@@ -215,4 +243,5 @@ __all__ = [
     "ResponseFormat",
     "ThinkingConfigParam",
     "ThinkingConfigEnabledParam",
+    "VectorVeinMessage",
 ]