vectorvein 0.1.5__py3-none-any.whl → 0.1.7__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- vectorvein/chat_clients/anthropic_client.py +4 -0
- vectorvein/settings/__init__.py +26 -28
- {vectorvein-0.1.5.dist-info → vectorvein-0.1.7.dist-info}/METADATA +1 -1
- {vectorvein-0.1.5.dist-info → vectorvein-0.1.7.dist-info}/RECORD +5 -5
- {vectorvein-0.1.5.dist-info → vectorvein-0.1.7.dist-info}/WHEEL +0 -0
@@ -125,6 +125,8 @@ class AnthropicChatClient(BaseChatClient):
|
|
125
125
|
self.stream = stream
|
126
126
|
if temperature is not None:
|
127
127
|
self.temperature = temperature
|
128
|
+
if isinstance(tools, OpenAINotGiven):
|
129
|
+
tools = NOT_GIVEN
|
128
130
|
if isinstance(tool_choice, OpenAINotGiven):
|
129
131
|
tool_choice = NOT_GIVEN
|
130
132
|
|
@@ -311,6 +313,8 @@ class AsyncAnthropicChatClient(BaseAsyncChatClient):
|
|
311
313
|
self.stream = stream
|
312
314
|
if temperature is not None:
|
313
315
|
self.temperature = temperature
|
316
|
+
if isinstance(tools, OpenAINotGiven):
|
317
|
+
tools = NOT_GIVEN
|
314
318
|
if isinstance(tool_choice, OpenAINotGiven):
|
315
319
|
tool_choice = NOT_GIVEN
|
316
320
|
|
vectorvein/settings/__init__.py
CHANGED
@@ -14,41 +14,39 @@ class Settings(BaseModel):
|
|
14
14
|
default_factory=list, description="Available endpoints for the LLM service."
|
15
15
|
)
|
16
16
|
|
17
|
-
|
18
|
-
|
19
|
-
)
|
20
|
-
|
21
|
-
|
22
|
-
|
23
|
-
|
24
|
-
|
25
|
-
|
26
|
-
|
27
|
-
|
28
|
-
|
29
|
-
yi_models: BackendSettings = Field(default_factory=BackendSettings, description="Yi models settings.")
|
30
|
-
zhipuai_models: BackendSettings = Field(default_factory=BackendSettings, description="Zhipuai models settings.")
|
17
|
+
anthropic: BackendSettings = Field(default_factory=BackendSettings, description="Anthropic models settings.")
|
18
|
+
deepseek: BackendSettings = Field(default_factory=BackendSettings, description="Deepseek models settings.")
|
19
|
+
gemini: BackendSettings = Field(default_factory=BackendSettings, description="Gemini models settings.")
|
20
|
+
groq: BackendSettings = Field(default_factory=BackendSettings, description="Groq models settings.")
|
21
|
+
local: BackendSettings = Field(default_factory=BackendSettings, description="Local models settings.")
|
22
|
+
minimax: BackendSettings = Field(default_factory=BackendSettings, description="Minimax models settings.")
|
23
|
+
mistral: BackendSettings = Field(default_factory=BackendSettings, description="Mistral models settings.")
|
24
|
+
moonshot: BackendSettings = Field(default_factory=BackendSettings, description="Moonshot models settings.")
|
25
|
+
openai: BackendSettings = Field(default_factory=BackendSettings, description="OpenAI models settings.")
|
26
|
+
qwen: BackendSettings = Field(default_factory=BackendSettings, description="Qwen models settings.")
|
27
|
+
yi: BackendSettings = Field(default_factory=BackendSettings, description="Yi models settings.")
|
28
|
+
zhipuai: BackendSettings = Field(default_factory=BackendSettings, description="Zhipuai models settings.")
|
31
29
|
|
32
30
|
def __init__(self, **data):
|
33
31
|
model_types = {
|
34
|
-
"
|
35
|
-
"
|
36
|
-
"
|
37
|
-
"
|
38
|
-
"
|
39
|
-
"
|
40
|
-
"
|
41
|
-
"
|
42
|
-
"
|
43
|
-
"
|
44
|
-
"
|
45
|
-
"
|
32
|
+
"anthropic": defs.ANTHROPIC_MODELS,
|
33
|
+
"deepseek": defs.DEEPSEEK_MODELS,
|
34
|
+
"gemini": defs.GEMINI_MODELS,
|
35
|
+
"groq": defs.GROQ_MODELS,
|
36
|
+
"local": {},
|
37
|
+
"minimax": defs.MINIMAX_MODELS,
|
38
|
+
"mistral": defs.MISTRAL_MODELS,
|
39
|
+
"moonshot": defs.MOONSHOT_MODELS,
|
40
|
+
"openai": defs.OPENAI_MODELS,
|
41
|
+
"qwen": defs.QWEN_MODELS,
|
42
|
+
"yi": defs.YI_MODELS,
|
43
|
+
"zhipuai": defs.ZHIPUAI_MODELS,
|
46
44
|
}
|
47
45
|
|
48
46
|
for model_type, default_models in model_types.items():
|
49
47
|
if model_type in data:
|
50
48
|
model_settings = BackendSettings()
|
51
|
-
model_settings.update_models(default_models, data[model_type])
|
49
|
+
model_settings.update_models(default_models, data[model_type].get("models", {}))
|
52
50
|
data[model_type] = model_settings
|
53
51
|
else:
|
54
52
|
data[model_type] = BackendSettings(models=default_models)
|
@@ -65,7 +63,7 @@ class Settings(BaseModel):
|
|
65
63
|
return EndpointSetting()
|
66
64
|
|
67
65
|
def get_backend(self, backend: BackendType) -> BackendSettings:
|
68
|
-
return getattr(self,
|
66
|
+
return getattr(self, backend.value.lower())
|
69
67
|
|
70
68
|
|
71
69
|
settings = Settings()
|
@@ -1,8 +1,8 @@
|
|
1
|
-
vectorvein-0.1.
|
2
|
-
vectorvein-0.1.
|
1
|
+
vectorvein-0.1.7.dist-info/METADATA,sha256=C_3UCv_92cL58DnFKRMs-1GbEbzGIdDPlNt-X4aIKF4,423
|
2
|
+
vectorvein-0.1.7.dist-info/WHEEL,sha256=rSwsxJWe3vzyR5HCwjWXQruDgschpei4h_giTm0dJVE,90
|
3
3
|
vectorvein/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
4
4
|
vectorvein/chat_clients/__init__.py,sha256=5j7W--jr-l2cqDJp38uXYvkydDK0rnzm7MYGSACHKmU,3976
|
5
|
-
vectorvein/chat_clients/anthropic_client.py,sha256=
|
5
|
+
vectorvein/chat_clients/anthropic_client.py,sha256=pkk0yPH05WTOnKOAXkm60ZZw1kdT8pCydNkHx7QArh4,18707
|
6
6
|
vectorvein/chat_clients/base_client.py,sha256=GjPUAjgd_36-lHr8QXvz2X_-ApjbmAD3KZRjmntJ65U,4247
|
7
7
|
vectorvein/chat_clients/deepseek_client.py,sha256=3qWu01NlJAP2N-Ff62d5-CZXZitlizE1fzb20LNetig,526
|
8
8
|
vectorvein/chat_clients/gemini_client.py,sha256=IpkfcqVF38f4kAFnlHyisn4vkiMeM4cICyS4XDD0jJE,12787
|
@@ -17,9 +17,9 @@ vectorvein/chat_clients/qwen_client.py,sha256=-ryh-m9PgsO0fc4ulcCmPTy1155J8YUy15
|
|
17
17
|
vectorvein/chat_clients/utils.py,sha256=tAQwfydj46sMxSHeaeOWXrTUY2q0h7482NbvZjbNz9A,17637
|
18
18
|
vectorvein/chat_clients/yi_client.py,sha256=RNf4CRuPJfixrwLZ3-DEc3t25QDe1mvZeb9sku2f8Bc,484
|
19
19
|
vectorvein/chat_clients/zhipuai_client.py,sha256=Ys5DSeLCuedaDXr3PfG1EW2zKXopt-awO2IylWSwY0s,519
|
20
|
-
vectorvein/settings/__init__.py,sha256=
|
20
|
+
vectorvein/settings/__init__.py,sha256=4mpccT7eZC3yI1vVnVViW4wHBnDEH9D2R5EsIP34VgU,3218
|
21
21
|
vectorvein/types/defaults.py,sha256=Mg-Mj3_eBzKZn1N8x1V2GqyaYgLD13i-NdSYdQC28X4,11437
|
22
22
|
vectorvein/types/enums.py,sha256=vzOenCnRlFXBwPh-lfFhjGfM-6yfDj7wZColHODqocI,1550
|
23
23
|
vectorvein/types/llm_parameters.py,sha256=nBjStC2zndTY__yhD2WFXB09taxEhDLE3OHA6MICfgE,3494
|
24
24
|
vectorvein/utilities/media_processing.py,sha256=BujciRmw1GMmc3ELRvafL8STcy6r5b2rVnh27-uA7so,2256
|
25
|
-
vectorvein-0.1.
|
25
|
+
vectorvein-0.1.7.dist-info/RECORD,,
|
File without changes
|