feat: refactor tongyi models (#3496)

This commit is contained in:
takatost 2024-04-15 22:28:32 +08:00 • committed by GitHub
commit 5b447d61a6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
29 changed files with 639 additions and 177 deletions

View file

@ -0,0 +1,24 @@
import pytest
from pydantic.error_wrappers import ValidationError
from core.rag.datasource.vdb.milvus.milvus_vector import MilvusConfig
def test_default_value():
valid_config = {
'host': 'localhost',
'port': 19530,
'user': 'root',
'password': 'Milvus'
}
for key in valid_config:
config = valid_config.copy()
del config[key]
with pytest.raises(ValidationError) as e:
MilvusConfig(**config)
assert e.value.errors()[1]['msg'] == f'config MILVUS_{key.upper()} is required'
config = MilvusConfig(**valid_config)
assert config.secure is False
assert config.database == 'default'