vectorvein 0.2.89__tar.gz → 0.2.91__tar.gz

{vectorvein-0.2.89 → vectorvein-0.2.91}/PKG-INFO

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

{vectorvein-0.2.89 → vectorvein-0.2.91}/pyproject.toml

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

{vectorvein-0.2.89 → vectorvein-0.2.91}/src/vectorvein/chat_clients/utils.py

@@ -19,6 +19,8 @@ from ..types.llm_parameters import (
     NOT_GIVEN,
     ToolParam,
     VectorVeinMessage,
+    VectorVeinTextMessage,
+    VectorVeinWorkflowMessage,
 )
 
 
@@ -436,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: VectorVeinMessage, content: str, backend: BackendType):
+def format_workflow_messages(message: VectorVeinWorkflowMessage, content: str, backend: BackendType):
     formatted_messages = []
 
     # 工具调用消息
@@ -654,13 +656,22 @@ 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 格式的消息

{vectorvein-0.2.89 → vectorvein-0.2.91}/src/vectorvein/types/__init__.py

@@ -56,6 +56,8 @@ from .llm_parameters import (
     OPENAI_NOT_GIVEN,
     ANTHROPIC_NOT_GIVEN,
     VectorVeinMessage,
+    VectorVeinTextMessage,
+    VectorVeinWorkflowMessage,
 )
 from .settings import (
     RedisConfigDict,
@@ -129,6 +131,8 @@ __all__ = [
     "OPENAI_NOT_GIVEN",
     "ANTHROPIC_NOT_GIVEN",
     "VectorVeinMessage",
+    "VectorVeinTextMessage",
+    "VectorVeinWorkflowMessage",
     "NotGiven",
     "NOT_GIVEN",
     "OpenAIToolParam",

{vectorvein-0.2.89 → vectorvein-0.2.91}/src/vectorvein/types/llm_parameters.py

@@ -1,7 +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
+from typing_extensions import TypedDict, NotRequired
 
 import httpx
 from pydantic import BaseModel, Field
@@ -27,16 +27,31 @@ from . import defaults as defs
 from .settings import EndpointOptionDict
 
 
-class VectorVeinMessage(TypedDict):
-    mid: str
-    author_type: str  # "U" for user, "A" for assistant
-    content_type: str  # "TXT" for text, "WKF" for workflow
+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: int
-    update_time: int
+    create_time: NotRequired[int]
+    update_time: NotRequired[int]
     metadata: dict
     content: Dict[str, str]
-    attachments: List[str]
+    attachments: NotRequired[List[str]]
+
+
+VectorVeinMessage = Union[VectorVeinTextMessage, VectorVeinWorkflowMessage]
 
 
 class EndpointSetting(BaseModel):
@@ -229,4 +244,6 @@ __all__ = [
     "ThinkingConfigParam",
     "ThinkingConfigEnabledParam",
     "VectorVeinMessage",
+    "VectorVeinTextMessage",
+    "VectorVeinWorkflowMessage",
 ]