Skip to content

Commit 9091259

Browse files
committed
Update unit tests to accommodate brainforge-llm-service config
1 parent 6a34688 commit 9091259

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/test_diana_utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -295,6 +295,7 @@ def test_get_unconfigured_backend_services(self):
295295
all_configured = {'keys': {'emails': {'configured': True},
296296
'track_my_brands': True},
297297
'api_services': {'configured': True},
298+
'brainforge_llm_service': {'configured': True},
298299
'LLM_VLLM': {'config': False},
299300
'LLM_CHAT_GPT': {'config': False},
300301
'LLM_CLAUDE': {'': ''},
@@ -312,7 +313,7 @@ def test_get_unconfigured_backend_services(self):
312313
'neon-email-proxy', 'neon-llm-chatgpt',
313314
'neon-llm-fastchat', 'neon-llm-claude',
314315
'neon-llm-palm', 'neon-llm-gemini',
315-
'neon-llm-vllm'})
316+
'neon-llm-vllm', 'brainforge-llm-service'})
316317

317318
def test_get_optional_http_backend(self):
318319
from neon_diana_utils.configuration import _get_optional_http_backend

0 commit comments

Comments
 (0)