Pass OutputRailsStreamingConfig into _run_output_rails_in_streaming as mandatory argument as all 2 calls check for None

tgasser-nv · tgasser-nv · commit 483c547dba6a · 2025-09-23T20:44:42.000-05:00
diff --git a/nemoguardrails/rails/llm/llmrails.py b/nemoguardrails/rails/llm/llmrails.py
@@ -1595,9 +1595,6 @@ def _prepare_params(
                 **action_params,
             }
 
-        output_rails_streaming_config = self.config.rails.output.streaming
-        if output_rails_streaming_config is None:
-            raise ValueError("Output rails streaming config is not available")
         buffer_strategy = get_buffer_strategy(output_rails_streaming_config)
         output_rails_flows_id = self.config.rails.output.flows
         stream_first = stream_first or output_rails_streaming_config.stream_first

Original file line number	Diff line number	Diff line change
`@@ -1595,9 +1595,6 @@ def _prepare_params(`
`1595`	`1595`	`**action_params,`
`1596`	`1596`	`}`
`1597`	`1597`
`1598`		`- output_rails_streaming_config = self.config.rails.output.streaming`
`1599`		`- if output_rails_streaming_config is None:`
`1600`		`- raise ValueError("Output rails streaming config is not available")`
`1601`	`1598`	`buffer_strategy = get_buffer_strategy(output_rails_streaming_config)`
`1602`	`1599`	`output_rails_flows_id = self.config.rails.output.flows`
`1603`	`1600`	`stream_first = stream_first or output_rails_streaming_config.stream_first`