feat: refactor tongyi models (#3496)

This commit is contained in:
takatost
2024-04-15 22:28:32 +08:00
committed by GitHub
parent fd90d99cd0
commit 5b447d61a6
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'