Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license
This commit is contained in:
@@ -0,0 +1,44 @@
|
||||
"""Tests for per-provider TLS settings in custom_providers config."""
|
||||
|
||||
from hermes_cli.config import (
|
||||
apply_custom_provider_tls_to_client_kwargs,
|
||||
get_custom_provider_tls_settings,
|
||||
)
|
||||
|
||||
|
||||
def test_get_custom_provider_tls_settings_matches_base_url():
|
||||
providers = [
|
||||
{
|
||||
"name": "Ollama",
|
||||
"base_url": "https://ollama.example.com/v1",
|
||||
"ssl_ca_cert": "/etc/ssl/mkcert-root.pem",
|
||||
}
|
||||
]
|
||||
tls = get_custom_provider_tls_settings(
|
||||
"https://ollama.example.com/v1/",
|
||||
custom_providers=providers,
|
||||
)
|
||||
assert tls == {"ssl_ca_cert": "/etc/ssl/mkcert-root.pem"}
|
||||
|
||||
|
||||
def test_apply_custom_provider_tls_to_client_kwargs():
|
||||
client_kwargs = {"api_key": "x", "base_url": "https://ollama.example.com/v1"}
|
||||
providers = [
|
||||
{
|
||||
"name": "Ollama",
|
||||
"base_url": "https://ollama.example.com/v1",
|
||||
"ssl_ca_cert": "/etc/ssl/mkcert-root.pem",
|
||||
"ssl_verify": True,
|
||||
}
|
||||
]
|
||||
apply_custom_provider_tls_to_client_kwargs(
|
||||
client_kwargs,
|
||||
"https://ollama.example.com/v1",
|
||||
custom_providers=providers,
|
||||
)
|
||||
assert client_kwargs["ssl_ca_cert"] == "/etc/ssl/mkcert-root.pem"
|
||||
assert client_kwargs["ssl_verify"] is True
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user