Skip to content

Commit f484f15

Browse files
committed
remove generate bot message task from dialog rails tasks
1 parent 59f67c4 commit f484f15

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

nemoguardrails/rails/llm/config.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1143,7 +1143,7 @@ def check_reasoning_traces_with_dialog_rails(cls, values):
11431143

11441144
# dialog rail tasks that should not have reasoning traces
11451145
dialog_rail_tasks = [
1146-
Task.GENERATE_BOT_MESSAGE,
1146+
# Task.GENERATE_BOT_MESSAGE,
11471147
Task.GENERATE_USER_INTENT,
11481148
Task.GENERATE_NEXT_STEPS,
11491149
Task.GENERATE_INTENT_STEPS_MESSAGE,

tests/test_config_validation.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -401,7 +401,7 @@ def test_reasoning_traces_with_dedicated_task_models():
401401
)
402402

403403
assert (
404-
"Model 'generate_bot_message' has reasoning traces enabled in config.yml"
404+
"Model 'generate_user_intent' has reasoning traces enabled in config.yml"
405405
in str(exc_info.value)
406406
)
407407
assert "Reasoning traces must be disabled for dialog rail tasks" in str(
@@ -489,7 +489,7 @@ def test_reasoning_traces_with_all_dialog_tasks():
489489
error_message = str(exc_info.value)
490490
assert (
491491
"Model 'generate_bot_message' has reasoning traces enabled in config.yml"
492-
in error_message
492+
not in error_message
493493
)
494494
assert (
495495
"Model 'generate_next_steps' has reasoning traces enabled in config.yml"
@@ -535,7 +535,7 @@ def test_reasoning_traces_with_implicit_dialog_rails_and_dedicated_models():
535535
- type: main
536536
engine: openai
537537
model: gpt-3.5-turbo-instruct
538-
- type: generate_bot_message
538+
- type: generate_user_intent
539539
engine: openai
540540
model: gpt-3.5-turbo-instruct
541541
reasoning_config:
@@ -556,7 +556,7 @@ def test_reasoning_traces_with_implicit_dialog_rails_and_dedicated_models():
556556
)
557557

558558
assert (
559-
"Model 'generate_bot_message' has reasoning traces enabled in config.yml"
559+
"Model 'generate_user_intent' has reasoning traces enabled in config.yml"
560560
in str(exc_info.value)
561561
)
562562
assert "Reasoning traces must be disabled for dialog rail tasks" in str(

0 commit comments

Comments
 (0)