|
2 | 2 |
|
3 | 3 | import pytest |
4 | 4 |
|
5 | | -from models.config import Configuration, LLamaStackConfiguration, ServiceConfiguration |
| 5 | +from models.config import LLamaStackConfiguration, ServiceConfiguration |
6 | 6 |
|
7 | 7 |
|
8 | 8 | def test_service_configuration_constructor() -> None: |
@@ -35,21 +35,23 @@ def test_service_configuration_workers_value() -> None: |
35 | 35 |
|
36 | 36 | def test_llama_stack_configuration_constructor() -> None: |
37 | 37 | """Test the LLamaStackConfiguration constructor.""" |
38 | | - l = LLamaStackConfiguration( |
| 38 | + llama_stack_configuration = LLamaStackConfiguration( |
39 | 39 | use_as_library_client=True, library_client_config_path="foo" |
40 | 40 | ) |
41 | | - assert l is not None |
| 41 | + assert llama_stack_configuration is not None |
42 | 42 |
|
43 | | - l = LLamaStackConfiguration(use_as_library_client=False, url="http://localhost") |
44 | | - assert l is not None |
| 43 | + llama_stack_configuration = LLamaStackConfiguration( |
| 44 | + use_as_library_client=False, url="http://localhost" |
| 45 | + ) |
| 46 | + assert llama_stack_configuration is not None |
45 | 47 |
|
46 | | - l = LLamaStackConfiguration(url="http://localhost") |
47 | | - assert l is not None |
| 48 | + llama_stack_configuration = LLamaStackConfiguration(url="http://localhost") |
| 49 | + assert llama_stack_configuration is not None |
48 | 50 |
|
49 | | - l = LLamaStackConfiguration( |
| 51 | + llama_stack_configuration = LLamaStackConfiguration( |
50 | 52 | use_as_library_client=False, url="http://localhost", api_key="foo" |
51 | 53 | ) |
52 | | - assert l is not None |
| 54 | + assert llama_stack_configuration is not None |
53 | 55 |
|
54 | 56 |
|
55 | 57 | def test_llama_stack_wrong_configuration_constructor_no_url() -> None: |
|
0 commit comments