FIX: Return structured output on non-streaming mode (#1318)

Author: romanrizzi

lib/completions/endpoints/base.rb

@@ -114,8 +114,7 @@ def perform_completion!(
               model_params[:response_format].dig(:json_schema, :schema, :properties)
 
             if schema_properties.present?
-              structured_output =
-                DiscourseAi::Completions::StructuredOutput.new(schema_properties)
+              structured_output = DiscourseAi::Completions::StructuredOutput.new(schema_properties)
             end
           end
 
@@ -430,23 +429,9 @@ def non_streaming_response(
           response_data.reject!(&:blank?)
 
           if structured_output.present?
-            has_string_response = false
-
-            response_data =
-              response_data.reduce([]) do |memo, data|
-                if data.is_a?(String)
-                  structured_output << data
-                  has_string_response = true
-                  next(memo)
-                else
-                  memo << data
-                end
-
-                memo
-              end
+            response_data.each { |data| structured_output << data if data.is_a?(String) }
 
-            # We only include the structured output if there was actually a structured response
-            response_data << structured_output if has_string_response
+            return structured_output
           end
 
           # this is to keep stuff backwards compatible

lib/completions/endpoints/canned_response.rb

@@ -68,6 +68,7 @@ def perform_completion!(
           end
 
           response = response.first if response.is_a?(Array) && response.length == 1
+
           response
         end