Skip to content

Commit e5018f0

Browse files
committed
move qwen name constant to common constants.py
1 parent 19357c1 commit e5018f0

File tree

6 files changed

+5
-9
lines changed

6 files changed

+5
-9
lines changed

tests/llama_stack/core/test_llamastack_core.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,7 @@
11
import pytest
22

33
from tests.llama_stack.constants import LlamaStackProviders
4-
from tests.model_explainability.constants import QWEN_MODEL_NAME
5-
from utilities.constants import MinIo
4+
from utilities.constants import MinIo, QWEN_MODEL_NAME
65

76

87
@pytest.mark.parametrize(

tests/llama_stack/eval/test_lmeval_provider.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,7 @@
11
import pytest
22

33
from tests.llama_stack.constants import LlamaStackProviders
4-
from tests.model_explainability.constants import QWEN_MODEL_NAME
5-
from utilities.constants import Timeout, MinIo
4+
from utilities.constants import Timeout, MinIo, QWEN_MODEL_NAME
65
from timeout_sampler import TimeoutSampler
76

87

tests/llama_stack/safety/test_trustyai_fms_provider.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,8 @@
33
from simple_logger.logger import get_logger
44

55
from tests.llama_stack.constants import LlamaStackProviders
6-
from tests.model_explainability.constants import QWEN_MODEL_NAME
76

8-
from utilities.constants import MinIo, CHAT_GENERATION_CONFIG, BUILTIN_DETECTOR_CONFIG
7+
from utilities.constants import MinIo, CHAT_GENERATION_CONFIG, BUILTIN_DETECTOR_CONFIG, QWEN_MODEL_NAME
98

109
LOGGER = get_logger(name=__name__)
1110
SECURE_SHIELD_ID: str = "secure_shield"

tests/model_explainability/constants.py

Lines changed: 0 additions & 1 deletion
This file was deleted.

tests/model_explainability/guardrails/test_guardrails.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@
77
from simple_logger.logger import get_logger
88
from timeout_sampler import retry
99

10-
from tests.model_explainability.constants import QWEN_MODEL_NAME
1110
from tests.model_explainability.guardrails.constants import (
1211
QWEN_ISVC_NAME,
1312
PROMPT_WITH_PII,
@@ -23,7 +22,7 @@
2322
log_request_and_response,
2423
)
2524
from tests.model_explainability.utils import validate_tai_component_images
26-
from utilities.constants import Timeout, CHAT_GENERATION_CONFIG, BUILTIN_DETECTOR_CONFIG, MinIo
25+
from utilities.constants import Timeout, CHAT_GENERATION_CONFIG, BUILTIN_DETECTOR_CONFIG, MinIo, QWEN_MODEL_NAME
2726
from utilities.plugins.constant import OpenAIEnpoints
2827

2928
LOGGER = get_logger(name=__name__)

utilities/constants.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -373,3 +373,4 @@ class RunTimeConfig:
373373
"service": {"hostname": f"{QWEN_ISVC_NAME}-predictor", "port": 8032, "request_timeout": 600}
374374
}
375375
TRUSTYAI_SERVICE_NAME: str = "trustyai-service"
376+
QWEN_MODEL_NAME: str = "qwen2.5-0.5b-instruct"

0 commit comments

Comments
 (0)