mirror of
https://github.com/microsoft/autogen.git
synced 2025-07-03 23:19:33 +00:00
69c0b2b5ef
Introduce validation for the ModelInfo dictionary to ensure required fields are present. Resolves #5501
23 lines
527 B
Python
23 lines
527 B
Python
import pytest
|
|
from autogen_core.models import ModelInfo, validate_model_info
|
|
|
|
|
|
def test_model_info() -> None:
|
|
# Valid model info.
|
|
info: ModelInfo = {
|
|
"family": "gpt-4o",
|
|
"vision": True,
|
|
"function_calling": True,
|
|
"json_output": True,
|
|
}
|
|
validate_model_info(info)
|
|
|
|
# Invalid model info.
|
|
info = {
|
|
"family": "gpt-4o",
|
|
"vision": True,
|
|
"function_calling": True,
|
|
} # type: ignore
|
|
with pytest.raises(ValueError):
|
|
validate_model_info(info)
|