gateway/tests/serviceAi/test_allowed_models_whitelist.py
2026-04-29 23:12:46 +02:00

14 lines
420 B
Python

# Copyright (c) 2025 Patrick Motsch
# All rights reserved.
import pytest
from modules.datamodels.datamodelAi import AiCallOptions
def test_allowed_models_field_exists():
opts = AiCallOptions(allowedModels=["gpt-5-mini", "claude-4-7-opus"])
assert opts.allowedModels == ["gpt-5-mini", "claude-4-7-opus"]
def test_allowed_models_default_none():
opts = AiCallOptions()
assert opts.allowedModels is None