diff --git a/lib/completions/endpoints/open_ai.rb b/lib/completions/endpoints/open_ai.rb index 3e7f54ee..6c6fffdc 100644 --- a/lib/completions/endpoints/open_ai.rb +++ b/lib/completions/endpoints/open_ai.rb @@ -107,7 +107,10 @@ module DiscourseAi if @streaming_mode payload[:stream] = true - payload[:stream_options] = { include_usage: true } + + # Usage is not available in Azure yet. + # We'll fallback to guess this using the tokenizer. + payload[:stream_options] = { include_usage: true } if model_uri.host.exclude?("azure") end payload[:tools] = dialect.tools if dialect.tools.present?