Skip to content

Commit a500ab7

Browse files
committed
feat: bot thinking in self check output rails
1 parent c841cc3 commit a500ab7

File tree

1 file changed

+2
-0
lines changed
  • nemoguardrails/library/self_check/output_check

1 file changed

+2
-0
lines changed

nemoguardrails/library/self_check/output_check/actions.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,6 +52,7 @@ async def self_check_output(
5252
_MAX_TOKENS = 3
5353
bot_response = context.get("bot_message")
5454
user_input = context.get("user_message")
55+
bot_thinking = context.get("bot_thinking")
5556

5657
task = Task.SELF_CHECK_OUTPUT
5758

@@ -61,6 +62,7 @@ async def self_check_output(
6162
context={
6263
"user_input": user_input,
6364
"bot_response": bot_response,
65+
"bot_thinking": bot_thinking,
6466
},
6567
)
6668
stop = llm_task_manager.get_stop_tokens(task=task)

0 commit comments

Comments
 (0)