vectorvein 0.1.32__tar.gz → 0.1.33__tar.gz

{vectorvein-0.1.32 → vectorvein-0.1.33}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vectorvein
-Version: 0.1.32
+Version: 0.1.33
 Summary: Default template for PDM package
 Author-Email: Anderson <andersonby@163.com>
 License: MIT

{vectorvein-0.1.32 → vectorvein-0.1.33}/pyproject.toml

@@ -16,7 +16,7 @@ description = "Default template for PDM package"
 name = "vectorvein"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "0.1.32"
+version = "0.1.33"
 
 [project.license]
 text = "MIT"

{vectorvein-0.1.32 → vectorvein-0.1.33}/src/vectorvein/types/defaults.py

@@ -1,17 +1,19 @@
 # @Author: Bi Ying
 # @Date:   2024-07-27 00:02:34
+from typing import Final, Dict, Any
+
 from .enums import ContextLengthControlType
 
-CONTEXT_LENGTH_CONTROL = ContextLengthControlType.Latest
+CONTEXT_LENGTH_CONTROL: Final[ContextLengthControlType] = ContextLengthControlType.Latest
 
-ENDPOINT_CONCURRENT_REQUESTS = 20
-ENDPOINT_RPM = 60
-ENDPOINT_TPM = 300000
+ENDPOINT_CONCURRENT_REQUESTS: Final[int] = 20
+ENDPOINT_RPM: Final[int] = 60
+ENDPOINT_TPM: Final[int] = 300000
 
-MODEL_CONTEXT_LENGTH = 32768
+MODEL_CONTEXT_LENGTH: Final[int] = 32768
 
 # Moonshot models
-MOONSHOT_MODELS = {
+MOONSHOT_MODELS: Final[Dict[str, Dict[str, Any]]] = {
     "moonshot-v1-8k": {
         "id": "moonshot-v1-8k",
         "context_length": 8192,
@@ -31,10 +33,10 @@ MOONSHOT_MODELS = {
         "response_format_available": True,
     },
 }
-MOONSHOT_DEFAULT_MODEL = "moonshot-v1-8k"
+MOONSHOT_DEFAULT_MODEL: Final[str] = "moonshot-v1-8k"
 
 # Deepseek models
-DEEPSEEK_MODELS = {
+DEEPSEEK_MODELS: Final[Dict[str, Dict[str, Any]]] = {
     "deepseek-chat": {
         "id": "deepseek-chat",
         "context_length": 128000,
@@ -50,10 +52,10 @@ DEEPSEEK_MODELS = {
         "response_format_available": True,
     },
 }
-DEEPSEEK_DEFAULT_MODEL = "deepseek-chat"
+DEEPSEEK_DEFAULT_MODEL: Final[str] = "deepseek-chat"
 
 # Baichuan models
-BAICHUAN_MODELS = {
+BAICHUAN_MODELS: Final[Dict[str, Dict[str, Any]]] = {
     "Baichuan4": {
         "id": "Baichuan4",
         "context_length": 32768,
@@ -90,11 +92,11 @@ BAICHUAN_MODELS = {
         "response_format_available": False,
     },
 }
-BAICHUAN_DEFAULT_MODEL = "Baichuan3-Turbo"
+BAICHUAN_DEFAULT_MODEL: Final[str] = "Baichuan3-Turbo"
 
 # Groq models
-GROQ_DEFAULT_MODEL = "llama3-70b-8192"
-GROQ_MODELS = {
+GROQ_DEFAULT_MODEL: Final[str] = "llama3-70b-8192"
+GROQ_MODELS: Final[Dict[str, Dict[str, Any]]] = {
     "mixtral-8x7b-32768": {
         "id": "mixtral-8x7b-32768",
         "context_length": 32768,
@@ -150,8 +152,8 @@ GROQ_MODELS = {
 }
 
 # Qwen models
-QWEN_DEFAULT_MODEL = "qwen2.5-72b-instruct"
-QWEN_MODELS = {
+QWEN_DEFAULT_MODEL: Final[str] = "qwen2.5-72b-instruct"
+QWEN_MODELS: Final[Dict[str, Dict[str, Any]]] = {
     "qwen1.5-1.8b-chat": {
         "id": "qwen1.5-1.8b-chat",
         "context_length": 30000,
@@ -267,8 +269,8 @@ QWEN_MODELS = {
 }
 
 # Yi models
-YI_DEFAULT_MODEL = "yi-large-turbo"
-YI_MODELS = {
+YI_DEFAULT_MODEL: Final[str] = "yi-large-turbo"
+YI_MODELS: Final[Dict[str, Dict[str, Any]]] = {
     "yi-large": {
         "id": "yi-large",
         "context_length": 32000,
@@ -322,8 +324,8 @@ YI_MODELS = {
 }
 
 # ZhiPuAI models
-ZHIPUAI_DEFAULT_MODEL = "glm-4-air"
-ZHIPUAI_MODELS = {
+ZHIPUAI_DEFAULT_MODEL: Final[str] = "glm-4-air"
+ZHIPUAI_MODELS: Final[Dict[str, Dict[str, Any]]] = {
     "glm-3-turbo": {
         "id": "glm-3-turbo",
         "context_length": 128000,
@@ -399,8 +401,8 @@ ZHIPUAI_MODELS = {
 }
 
 # Mistral models
-MISTRAL_DEFAULT_MODEL = "mistral-small"
-MISTRAL_MODELS = {
+MISTRAL_DEFAULT_MODEL: Final[str] = "mistral-small"
+MISTRAL_MODELS: Final[Dict[str, Dict[str, Any]]] = {
     "open-mistral-7b": {
         "id": "open-mistral-7b",
         "context_length": 32000,
@@ -452,8 +454,8 @@ MISTRAL_MODELS = {
 }
 
 # OpenAI models
-OPENAI_DEFAULT_MODEL = "gpt-4o"
-OPENAI_MODELS = {
+OPENAI_DEFAULT_MODEL: Final[str] = "gpt-4o"
+OPENAI_MODELS: Final[Dict[str, Dict[str, Any]]] = {
     "gpt-35-turbo": {
         "id": "gpt-35-turbo",
         "context_length": 16385,
@@ -517,8 +519,8 @@ OPENAI_MODELS = {
 }
 
 # Anthropic models
-ANTHROPIC_DEFAULT_MODEL = "claude-3-5-sonnet-20240620"
-ANTHROPIC_MODELS = {
+ANTHROPIC_DEFAULT_MODEL: Final[str] = "claude-3-5-sonnet-20240620"
+ANTHROPIC_MODELS: Final[Dict[str, Dict[str, Any]]] = {
     "claude-3-opus-20240229": {
         "id": "claude-3-opus-20240229",
         "context_length": 200000,
@@ -554,8 +556,8 @@ ANTHROPIC_MODELS = {
 }
 
 # Minimax models
-MINIMAX_DEFAULT_MODEL = "abab6.5s-chat"
-MINIMAX_MODELS = {
+MINIMAX_DEFAULT_MODEL: Final[str] = "abab6.5s-chat"
+MINIMAX_MODELS: Final[Dict[str, Dict[str, Any]]] = {
     "abab5-chat": {
         "id": "abab5-chat",
         "context_length": 6144,
@@ -587,8 +589,8 @@ MINIMAX_MODELS = {
 }
 
 # Gemini models
-GEMINI_DEFAULT_MODEL = "gemini-1.5-pro"
-GEMINI_MODELS = {
+GEMINI_DEFAULT_MODEL: Final[str] = "gemini-1.5-pro"
+GEMINI_MODELS: Final[Dict[str, Dict[str, Any]]] = {
     "gemini-1.5-pro": {
         "id": "gemini-1.5-pro",
         "context_length": 1048576,
@@ -606,8 +608,8 @@ GEMINI_MODELS = {
 }
 
 # 百度文心一言 ERNIE 模型
-ERNIE_DEFAULT_MODEL = "ernie-lite"
-ERNIE_MODELS = {
+ERNIE_DEFAULT_MODEL: Final[str] = "ernie-lite"
+ERNIE_MODELS: Final[Dict[str, Dict[str, Any]]] = {
     "ernie-lite": {
         "id": "ernie-lite",
         "context_length": 6144,
@@ -646,8 +648,8 @@ ERNIE_MODELS = {
 }
 
 
-STEPFUN_DEFAULT_MODEL = "step-1-8k"
-STEPFUN_MODELS = {
+STEPFUN_DEFAULT_MODEL: Final[str] = "step-1-8k"
+STEPFUN_MODELS: Final[Dict[str, Dict[str, Any]]] = {
     "step-1-8k": {
         "id": "step-1-8k",
         "context_length": 8192,

{vectorvein-0.1.32 → vectorvein-0.1.33}/src/vectorvein/types/llm_parameters.py

@@ -9,7 +9,7 @@ from anthropic._types import NotGiven as AnthropicNotGiven
 from anthropic.types.message_create_params import ToolChoice as AnthropicToolChoice
 
 from openai._types import NotGiven as OpenAINotGiven
-from openai._types import NOT_GIVEN as OpenAINOT_GIVEN
+from openai._types import NOT_GIVEN as OPENAI_NOT_GIVEN
 from openai.types.chat.chat_completion_chunk import ChoiceDeltaToolCall
 from openai.types.chat.chat_completion_tool_param import ChatCompletionToolParam
 from openai.types.chat.chat_completion_message_tool_call import ChatCompletionMessageToolCall
@@ -93,7 +93,7 @@ class ChatCompletionDeltaMessage(BaseModel):
 
 NotGiven = Union[AnthropicNotGiven, OpenAINotGiven]
 
-NOT_GIVEN = OpenAINOT_GIVEN
+NOT_GIVEN = OPENAI_NOT_GIVEN
 
 OpenAIToolParam = ChatCompletionToolParam
 ToolParam = OpenAIToolParam
@@ -101,3 +101,21 @@ ToolParam = OpenAIToolParam
 Tools = Iterable[ToolParam]
 
 ToolChoice = ChatCompletionToolChoiceOptionParam
+
+
+__all__ = [
+    "EndpointSetting",
+    "ModelSetting",
+    "BackendSettings",
+    "Usage",
+    "ChatCompletionMessage",
+    "ChatCompletionDeltaMessage",
+    "NotGiven",
+    "NOT_GIVEN",
+    "OpenAIToolParam",
+    "ToolParam",
+    "Tools",
+    "ToolChoice",
+    "AnthropicToolParam",
+    "AnthropicToolChoice",
+]