Support for allowed model restrictions per provider
Tool escalation added to `analyze` to a graceful switch over to codereview is made when absolutely necessary
This commit is contained in:
@@ -164,6 +164,18 @@ class TestGeminiProvider:
|
||||
class TestOpenAIProvider:
|
||||
"""Test OpenAI model provider"""
|
||||
|
||||
def setup_method(self):
|
||||
"""Clear restriction service cache before each test"""
|
||||
import utils.model_restrictions
|
||||
|
||||
utils.model_restrictions._restriction_service = None
|
||||
|
||||
def teardown_method(self):
|
||||
"""Clear restriction service cache after each test"""
|
||||
import utils.model_restrictions
|
||||
|
||||
utils.model_restrictions._restriction_service = None
|
||||
|
||||
def test_provider_initialization(self):
|
||||
"""Test provider initialization"""
|
||||
provider = OpenAIModelProvider(api_key="test-key", organization="test-org")
|
||||
|
||||
Reference in New Issue
Block a user