File size: 1,192 Bytes
c59d808 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
import unittest
import os
from unittest.mock import patch
from backend.config.settings import settings
class TestValidLLMProviderSettings(unittest.TestCase):
@patch.dict(os.environ, {
'GOOGLE_API_KEY': 'ivq',
'GOOGLE_MAX_TOKENS': '1200',
'OPENAI_API_KEY': 'xyz',
'OPENAI_MAX_TOKENS': '3800',
'LLM_PROVIDER': 'openai'
})
def test_valid_llm_provider_settings(self):
"""
Test settings object provides correct config when LLM_PROVIDER is set
"""
# reinitialize settings instance
settings.__init__()
llm_config = settings.get_llm_config()
self.assertEqual(llm_config['api_key'], 'xyz')
self.assertEqual(llm_config['max_tokens'], 3800)
class TestInvalidLLMProviderSettings(unittest.TestCase):
@patch.dict(os.environ, {
'LLM_PROVIDER': 'microsoft'
})
def test_invalid_llm_provider_settings(self):
"""
Test that improper provider config raises the right error
"""
# reinitialize settings instance
settings.__init__()
with self.assertRaisesRegex(ValueError, "Unsupported"):
settings.get_llm_config() |