Skip to content

Commit b3a2281

Browse files
Merge pull request #2398 from asamal4/OLS-1497d
enable periodic & rag tool calling e2e
2 parents 1f735e8 + 728badf commit b3a2281

File tree

2 files changed

+8
-9
lines changed

2 files changed

+8
-9
lines changed

tests/e2e/test_query_endpoint.py

+2
Original file line numberDiff line numberDiff line change
@@ -196,6 +196,7 @@ def test_too_long_question() -> None:
196196
assert json_response["detail"]["response"] == "Prompt is too long"
197197

198198

199+
@pytest.mark.introspection
199200
@pytest.mark.smoketest
200201
@pytest.mark.rag
201202
def test_valid_question() -> None:
@@ -347,6 +348,7 @@ def test_token_counters_for_query_call_with_improper_payload() -> None:
347348
response_utils.check_content_type(response, "application/json")
348349

349350

351+
@pytest.mark.introspection
350352
@pytest.mark.rag
351353
@retry(max_attempts=3, wait_between_runs=10)
352354
def test_rag_question() -> None:

tests/scripts/test-e2e-cluster-periodics.sh

+6-9
Original file line numberDiff line numberDiff line change
@@ -62,17 +62,14 @@ function run_suites() {
6262
run_suite "model_eval" "model_evaluation" "watsonx" "$WATSONX_PROVIDER_KEY_PATH" "ibm/granite-3-8b-instruct" "$OLS_IMAGE" "n"
6363
(( rc = rc || $? ))
6464

65-
# TODO: Enable below test cases once flag is added to operator CRD. Update flag name in CRD yaml (if different name is used)
6665
# TODO: Reduce execution time. Sequential execution will take more time. Parallel execution will have cluster claim issue.
6766
# Run tool calling - Enable introspection
68-
# run_suite "azure_openai_introspection" "introspection" "azure_openai" "$AZUREOPENAI_PROVIDER_KEY_PATH" "gpt-4o-mini" "$OLS_IMAGE" "y"
69-
# (( rc = rc || $? ))
70-
71-
# run_suite "openai_introspection" "introspection" "openai" "$OPENAI_PROVIDER_KEY_PATH" "gpt-4o-mini" "$OLS_IMAGE" "y"
72-
# (( rc = rc || $? ))
73-
74-
# run_suite "watsonx_introspection" "introspection" "watsonx" "$WATSONX_PROVIDER_KEY_PATH" "ibm/granite-3-8b-instruct" "$OLS_IMAGE" "y"
75-
# (( rc = rc || $? ))
67+
run_suite "azure_openai_introspection" "introspection" "azure_openai" "$AZUREOPENAI_PROVIDER_KEY_PATH" "gpt-4o-mini" "$OLS_IMAGE" "y"
68+
(( rc = rc || $? ))
69+
run_suite "openai_introspection" "introspection" "openai" "$OPENAI_PROVIDER_KEY_PATH" "gpt-4o-mini" "$OLS_IMAGE" "y"
70+
(( rc = rc || $? ))
71+
run_suite "watsonx_introspection" "introspection" "watsonx" "$WATSONX_PROVIDER_KEY_PATH" "ibm/granite-3-2-8b-instruct" "$OLS_IMAGE" "y"
72+
(( rc = rc || $? ))
7673

7774
set -e
7875

0 commit comments

Comments
 (0)