diff --git a/backend/core/consts.py b/backend/core/consts.py index 9a2c1d5..39fc2cf 100644 --- a/backend/core/consts.py +++ b/backend/core/consts.py @@ -12,11 +12,6 @@ 'id': 'gemini', 'label': 'Google Gemini', 'base_url': 'https://generativelanguage.googleapis.com/v1beta' - }, - { - 'id': 'custom', - 'label': 'Custom Provider', - 'base_url': '' } ] diff --git a/backend/core/tests/test_ai_provider.py b/backend/core/tests/test_ai_provider.py index 2922049..faab4bd 100644 --- a/backend/core/tests/test_ai_provider.py +++ b/backend/core/tests/test_ai_provider.py @@ -137,7 +137,7 @@ def test_delete_other_users_provider(self): def test_create_ai_provider(self, mock_validate): """Test that authenticated user can create their own AI provider.""" mock_validate.return_value = (True, ['gemini-1.5-pro', 'gemini-1.5-flash']) - + user = UserFactory() self.client.force_authenticate(user=user) @@ -169,7 +169,7 @@ def test_create_ai_provider(self, mock_validate): def test_update_own_provider(self, mock_validate): """Test that authenticated user can update their own AI provider.""" mock_validate.return_value = (True, ['gemini-1.5-pro', 'gemini-1.5-flash']) - + user = UserFactory() self.client.force_authenticate(user=user) @@ -249,7 +249,7 @@ def test_update_without_api_key_does_not_change_api_key(self): def test_update_with_api_key_changes_api_key(self, mock_validate): """Test that update request with provider api key updates the provider api key.""" mock_validate.return_value = (True, ['gemini-1.5-pro', 'gemini-1.5-flash']) - + user = UserFactory() self.client.force_authenticate(user=user) @@ -312,7 +312,7 @@ def test_update_with_whitespace_api_key_does_not_change_api_key(self): def test_api_key_is_encrypted_in_database(self, mock_validate): """Test that provider_api_key is encrypted when stored in the database.""" mock_validate.return_value = (True, ['gemini-1.5-pro', 'gemini-1.5-flash']) - + user = UserFactory() self.client.force_authenticate(user=user) @@ -343,7 +343,7 @@ def test_api_key_is_encrypted_in_database(self, mock_validate): def test_create_with_supported_provider_gemini(self, mock_validate): """Test that AI provider can be created with supported 'gemini' provider.""" mock_validate.return_value = (True, ['gemini-1.5-pro', 'gemini-1.5-flash']) - + user = UserFactory() data = { 'name': 'My Google Gemini Provider', @@ -362,29 +362,6 @@ def test_create_with_supported_provider_gemini(self, mock_validate): assert provider.metadata['base_url'] == 'https://generativelanguage.googleapis.com' assert provider.creator == user - @patch('core.services.factories.ai_provider_factory.AIProviderFactory.validate_provider') - def test_create_with_supported_provider_custom(self, mock_validate): - """Test that AI provider can be created with supported 'custom' provider.""" - mock_validate.return_value = (True, ['custom-model-1', 'custom-model-2']) - - user = UserFactory() - data = { - 'name': 'My Custom Provider', - 'provider': 'custom', - 'base_url': 'https://my-custom-api.com', - 'provider_api_key': 'test-api-key-67890' - } - - serializer = AIProviderCreateSerializer(data=data, context={'request': type('MockRequest', (), {'user': user})()}) - - assert serializer.is_valid() - provider = serializer.save() - - assert provider.name == 'My Custom Provider' - assert provider.provider == 'custom' - assert provider.metadata['base_url'] == 'https://my-custom-api.com' - assert provider.creator == user - def test_create_with_unsupported_provider_fails(self): """Test that AI provider creation fails with unsupported provider.""" user = UserFactory()