vectorvein 0.1.14__tar.gz → 0.1.15__tar.gz

{vectorvein-0.1.14 → vectorvein-0.1.15}/PKG-INFO

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

{vectorvein-0.1.14 → vectorvein-0.1.15}/pyproject.toml

@@ -16,7 +16,7 @@ description = "Default template for PDM package"
 name = "vectorvein"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "0.1.14"
+version = "0.1.15"
 
 [project.license]
 text = "MIT"
@@ -29,3 +29,8 @@ requires = [
 
 [tool.pdm]
 distribution = true
+
+[tool.pdm.build]
+excludes = [
+    "tests",
+]

{vectorvein-0.1.14 → vectorvein-0.1.15}/src/vectorvein/types/defaults.py

@@ -233,6 +233,7 @@ YI_MODELS = {
         "max_output_tokens": 2000,
         "function_call_available": False,
         "response_format_available": False,
+        "native_multimodal": True,
     },
 }
 
@@ -301,6 +302,7 @@ ZHIPUAI_MODELS = {
         "function_call_available": False,
         "response_format_available": False,
         "max_output_tokens": 1024,
+        "native_multimodal": True,
     },
     "glm-4v-plus": {
         "id": "glm-4v-plus",
@@ -308,6 +310,7 @@ ZHIPUAI_MODELS = {
         "function_call_available": False,
         "response_format_available": False,
         "max_output_tokens": 1024,
+        "native_multimodal": True,
     },
 }
 
@@ -394,6 +397,7 @@ OPENAI_MODELS = {
         "max_output_tokens": 4096,
         "function_call_available": True,
         "response_format_available": True,
+        "native_multimodal": True,
     },
     "gpt-4o-mini": {
         "id": "gpt-4o-mini",
@@ -401,6 +405,7 @@ OPENAI_MODELS = {
         "max_output_tokens": 16384,
         "function_call_available": True,
         "response_format_available": True,
+        "native_multimodal": True,
     },
     "gpt-4v": {
         "id": "gpt-4v",
@@ -420,12 +425,14 @@ ANTHROPIC_MODELS = {
         "max_output_tokens": 4096,
         "function_call_available": True,
         "response_format_available": True,
+        "native_multimodal": True,
     },
     "claude-3-sonnet-20240229": {
         "id": "claude-3-sonnet-20240229",
         "context_length": 200000,
         "max_output_tokens": 4096,
         "function_call_available": True,
+        "native_multimodal": True,
         "response_format_available": True,
     },
     "claude-3-haiku-20240307": {
@@ -434,6 +441,7 @@ ANTHROPIC_MODELS = {
         "max_output_tokens": 4096,
         "function_call_available": True,
         "response_format_available": True,
+        "native_multimodal": True,
     },
     "claude-3-5-sonnet-20240620": {
         "id": "claude-3-5-sonnet-20240620",
@@ -441,6 +449,7 @@ ANTHROPIC_MODELS = {
         "max_output_tokens": 4096,
         "function_call_available": True,
         "response_format_available": True,
+        "native_multimodal": True,
     },
 }
 
@@ -485,11 +494,13 @@ GEMINI_MODELS = {
         "context_length": 1048576,
         "function_call_available": True,
         "response_format_available": True,
+        "native_multimodal": True,
     },
     "gemini-1.5-flash": {
         "id": "gemini-1.5-flash",
         "context_length": 1048576,
         "function_call_available": True,
         "response_format_available": True,
+        "native_multimodal": True,
     },
 }

{vectorvein-0.1.14 → vectorvein-0.1.15}/src/vectorvein/types/llm_parameters.py

@@ -30,6 +30,7 @@ class ModelSetting(BaseModel):
     endpoints: List[str] = Field(default_factory=list, description="Available endpoints for the model.")
     function_call_available: bool = Field(False, description="Indicates if function call is available.")
     response_format_available: bool = Field(False, description="Indicates if response format is available.")
+    native_multimodal: bool = Field(False, description="Indicates if the model is a native multimodal model.")
     context_length: int = Field(32768, description="The context length for the model.")
     max_output_tokens: Optional[int] = Field(None, description="Maximum number of output tokens allowed.")