vectorvein 0.2.88__tar.gz → 0.2.89__tar.gz

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

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

{vectorvein-0.2.88 → vectorvein-0.2.89}/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.89"
 
 [project.license]
 text = "MIT"

{vectorvein-0.2.88 → vectorvein-0.2.89}/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,7 @@ from ..types.llm_parameters import (
     NotGiven,
     NOT_GIVEN,
     ToolParam,
+    VectorVeinMessage,
 )
 
 
@@ -435,7 +436,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: VectorVeinMessage, content: str, backend: BackendType):
     formatted_messages = []
 
     # 工具调用消息
@@ -626,7 +627,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,6 +653,7 @@ def format_messages(
 
     for message in messages:
         if is_vectorvein_message(message):
+            message = cast(VectorVeinMessage, message)
             # 处理 VectorVein 格式的消息
             content = message["content"]["text"]
             if message["content_type"] == "TXT":
@@ -662,6 +664,7 @@ def format_messages(
                 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.89}/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.89}/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
 
 import httpx
 from pydantic import BaseModel, Field
@@ -26,6 +27,18 @@ 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
+    status: str
+    create_time: int
+    update_time: int
+    metadata: dict
+    content: Dict[str, str]
+    attachments: List[str]
+
+
 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 +228,5 @@ __all__ = [
     "ResponseFormat",
     "ThinkingConfigParam",
     "ThinkingConfigEnabledParam",
+    "VectorVeinMessage",
 ]