whoosh 1.9.1 → 1.10.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 035c0a8c4cfc32ee16cba0eba7e84e23ec1c510f41bd541a8409fa59dc5dbbd2
-  data.tar.gz: aca768fe55956b76f30b9296fe9ee156b3547ce2089a49a5edcf18ab7ce33d07
+  metadata.gz: 0c8ecbe08e5ecf50f7f1233da3767074a79d9e4164a719e9397be1ac9637efb3
+  data.tar.gz: 622f3f889c550ecdff4cfab4f38120f2ce3f14b7f2021be99dae9790c3cfc5f7
 SHA512:
-  metadata.gz: f3f877e2e128bd4b16ffece2ebf7a41b87678185906688cac6d514e70c28d0bb4c0bcebd1e466368a2fc591af83f8e89b6b15d5c6b59d493a8e5a2bc78c3da76
-  data.tar.gz: aca365000fb9776754d29038d60061dcf92f71ade132e0ea602f80f0defe47a2fe83d5c3ef435e6e241b9bbd2cbdcb9fd1f4c34d38f99f160befd1b0f322930c
+  metadata.gz: 70add9595209f2ff3a1c5c4a52e25a4428149694f4df9debc7cc58dace2d15b77d46dcf1591cb9cd78e7d02654f0402326c58c9e060f1c94897e0011fc2cfab2
+  data.tar.gz: 83fb93d35d3b1d446bc99d1f8ee0a21b16f062af81da4f4e737b98cd85acced622fe332ccefb785c5f114fa78131131bdea9f008e109c49291c793c117cea843

data/lib/whoosh/ai/llm.rb

@@ -90,18 +90,19 @@ module Whoosh
         end
       end
 
-      # Stream LLM response — yields chunks
+      # Stream an LLM response — yields each chunk as ruby_llm produces it.
+      # The block receives a RubyLLM::Chunk (a Message subclass with #content).
+      # Returns the final response message after the stream completes.
       def stream(message, model: nil, system: nil, &block)
         ensure_ruby_llm!
+        unless @ruby_llm
+          raise Errors::DependencyError, "No LLM provider available. Add 'ruby_llm' to your Gemfile."
+        end
 
-        messages = [{ role: "user", content: message }]
-        # Delegate to ruby_llm's streaming interface
-        if @ruby_llm
-          # ruby_llm streaming would go here
-          # For now, fall back to non-streaming
-          result = chat(message, model: model, system: system, cache: false)
-          yield result if block_given?
-          result
+        chat = RubyLLM.chat(model: model || @model || DEFAULT_MODEL)
+        chat.with_instructions(system) if system
+        chat.ask(message) do |chunk|
+          block.call(chunk) if block
         end
       end
 

data/lib/whoosh/streaming/llm_stream.rb

@@ -17,7 +17,8 @@ module Whoosh
 
       def <<(chunk)
         return if @closed
-        text = chunk.respond_to?(:text) ? chunk.text : chunk.to_s
+        text = extract_text(chunk)
+        return self if text.nil? || text.empty?
         payload = { choices: [{ delta: { content: text } }] }
         write("data: #{JSON.generate(payload)}\n\n")
         self
@@ -40,6 +41,21 @@ module Whoosh
 
       private
 
+      # ruby_llm chunks expose #content (Message subclass). Older code paths
+      # and plain-string yields are also supported.
+      def extract_text(chunk)
+        return chunk if chunk.is_a?(String)
+        if chunk.respond_to?(:content)
+          c = chunk.content
+          return "" if c.nil?
+          return c if c.is_a?(String)
+          return c.text if c.respond_to?(:text)
+          return c.to_s
+        end
+        return chunk.text if chunk.respond_to?(:text)
+        chunk.to_s
+      end
+
       def write(data)
         @io.write(data)
         @io.flush if @io.respond_to?(:flush)

data/lib/whoosh/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Whoosh
-  VERSION = "1.9.1"
+  VERSION = "1.10.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: whoosh
 version: !ruby/object:Gem::Version
-  version: 1.9.1
+  version: 1.10.0
 platform: ruby
 authors:
 - Johannes Dwi Cahyo