swarm_sdk 2.7.14 → 2.7.15

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 22ea4e05517149f4cb6f6f18e896e9493e9ed75f69e82bd6891a9b0b471e314e
-  data.tar.gz: 94f2f1b7c28fe626889aa4f1fbb611bb36ac4d73ab7de778f76e553538589cbb
+  metadata.gz: e774808cdf4817c1cd483b3b09972cc1bf8b99d9226308baac8df60315c85ea2
+  data.tar.gz: 13a214280887d48dafdfe8b36da95caca7b7788120d92efc71ed3e9cb6da0293
 SHA512:
-  metadata.gz: 2360f50b71a7c2342f2301a8dc5362b0e92402f16bdaee19dc7932b395a33f37e12a385fe752b69411e481ed10418396b7245a5d61669f7d41d12b6f7d0c3f40
-  data.tar.gz: 076b1ca993b017aeabfe5e35daa08be81e4cd815e314b2e28b594f298ae2b5b430101a0c9ea01d5263689b21ad888e4fb17d06c4a59904aa38734308a21ad130
+  metadata.gz: b0c7654413b29e405fec0fa66ef6e992c2f84a5fdeaf6be6ce52412ca2f3f00cca0d24865141c97d922a7cd73b12ed00fbd3c0118fa48bee78a5846e0bb89fb7
+  data.tar.gz: fa812ed602e16954a23f72dec7bd9c1783afe4df852a93aa29db0358db1b329ede8303d7f59b5bdfe0d7847074ed515c534d0e32d5c2e523d3cef67886f61f56

data/lib/swarm_sdk/agent/chat.rb

@@ -1140,6 +1140,7 @@ module SwarmSDK
           error_message: error.message,
           status_code: error.respond_to?(:response) ? error.response&.status : nil,
           retryable: false,
+          backtrace: error.backtrace&.first(20),
         )
       end
 

data/lib/swarm_sdk/agent/chat_helpers/llm_configuration.rb

@@ -155,6 +155,10 @@ module SwarmSDK
           when "ollama"
             config.ollama_api_base = base_url
             # Ollama doesn't need an API key
+          when "gemini"
+            config.gemini_api_base = base_url
+            api_key = SwarmSDK.config.gemini_api_key
+            config.gemini_api_key = api_key if api_key
           when "gpustack"
             config.gpustack_api_base = base_url
             api_key = SwarmSDK.config.gpustack_api_key

data/lib/swarm_sdk/ruby_llm_patches/init.rb

@@ -40,5 +40,11 @@ require_relative "message_management_patch"
 # 7. Responses API patch (depends on configuration, uses error classes)
 require_relative "responses_api_patch"
 
-# 8. MCP SSL patch (configures SSL for HTTPX connections in ruby_llm-mcp)
+# 8. Streaming error patch (hardens error parsing for non-standard proxy responses)
+require_relative "streaming_error_patch"
+
+# 9. OpenAI thought_signature patch (preserves Gemini thought signatures through OpenAI provider)
+require_relative "openai_thought_signature_patch"
+
+# 10. MCP SSL patch (configures SSL for HTTPX connections in ruby_llm-mcp)
 require_relative "mcp_ssl_patch"

data/lib/swarm_sdk/ruby_llm_patches/openai_thought_signature_patch.rb

@@ -0,0 +1,98 @@
+# frozen_string_literal: true
+
+# Patches RubyLLM::Providers::OpenAI::Tools to preserve thought_signature
+# through the OpenAI-compatible streaming pipeline.
+#
+# Vertex AI Gemini 3 models with "thinking" enabled return a thought_signature
+# in tool call responses via extra_content.google.thought_signature. This must
+# be echoed back in subsequent requests or the API rejects the request with:
+#
+#   "function call is missing a thought_signature"
+#
+# The native Gemini provider handles this correctly, but the OpenAI provider
+# (used for OpenAI-compatible proxies) drops thought_signature in both
+# parse_tool_calls and format_tool_calls. The rest of the pipeline
+# (StreamAccumulator, ToolCall) already supports thought_signature.
+#
+# This patch:
+# - Extracts thought_signature from extra_content during tool call parsing
+# - Echoes thought_signature back in extra_content during serialization
+
+module RubyLLM
+  module Providers
+    class OpenAI
+      module Tools
+        # rubocop:disable Style/ModuleFunction -- required to replace singleton method copy
+
+        module_function
+
+        # Parse tool calls from OpenAI-format response data
+        #
+        # @param tool_calls [Array<Hash>] Raw tool call data from API response
+        # @param parse_arguments [Boolean] Whether to JSON-parse arguments (false during streaming)
+        # @return [Hash{String => ToolCall}, nil] Parsed tool calls keyed by ID
+        def parse_tool_calls(tool_calls, parse_arguments: true)
+          return unless tool_calls&.any?
+
+          tool_calls.to_h do |tc|
+            thought_sig = tc.dig("extra_content", "google", "thought_signature")
+
+            [
+              tc["id"],
+              ToolCall.new(
+                id: tc["id"],
+                name: tc.dig("function", "name"),
+                arguments: if parse_arguments
+                             parse_tool_call_arguments(tc)
+                           else
+                             tc.dig("function", "arguments")
+                           end,
+                thought_signature: thought_sig,
+              ),
+            ]
+          end
+        end
+
+        # Serialize tool calls into OpenAI-format request data
+        #
+        # @param tool_calls [Hash{String => ToolCall}] Tool calls to serialize
+        # @return [Array<Hash>, nil] Serialized tool calls for API request
+        def format_tool_calls(tool_calls)
+          return unless tool_calls&.any?
+
+          tool_calls.map do |_, tc|
+            entry = {
+              id: tc.id,
+              type: "function",
+              function: {
+                name: tc.name,
+                arguments: JSON.generate(tc.arguments),
+              },
+            }
+
+            if tc.thought_signature
+              entry[:extra_content] = { google: { thought_signature: tc.thought_signature } }
+            end
+
+            entry
+          end
+        end
+
+        # Parse tool call arguments from raw hash
+        #
+        # @param tool_call [Hash] Raw tool call hash
+        # @return [Hash] Parsed arguments
+        def parse_tool_call_arguments(tool_call)
+          arguments = tool_call.dig("function", "arguments")
+
+          if arguments.nil? || arguments.empty?
+            {}
+          else
+            JSON.parse(arguments)
+          end
+        end
+        # rubocop:enable Style/ModuleFunction
+      end
+    end
+  end
+end

data/lib/swarm_sdk/ruby_llm_patches/streaming_error_patch.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+# Hardens RubyLLM::Providers::OpenAI::Streaming#parse_streaming_error against
+# non-standard error response shapes returned by OpenAI-compatible proxies
+# (e.g. Gemini via Vertex AI).
+#
+# The upstream implementation assumes `error_data['error']` is always a Hash,
+# but some proxies return a bare String ({"error": "message"}) or an Array
+# top-level, causing TypeError: no implicit conversion of String into Integer.
+#
+# This patch adds type guards while preserving the exact original behavior
+# for well-formed OpenAI error responses.
+#
+# Upstream issue: https://github.com/crmne/ruby_llm/issues/XXX
+
+module RubyLLM
+  module Providers
+    class OpenAI
+      module Streaming
+        # rubocop:disable Style/ModuleFunction -- module_function is required here
+        # to replace both the singleton and instance method copies created by the
+        # original module_function call in upstream RubyLLM. extend self would only
+        # add a delegation layer and not override the existing singleton method.
+
+        module_function
+
+        def parse_streaming_error(data)
+          error_data = JSON.parse(data)
+          return unless error_data.is_a?(Hash)
+
+          error = error_data["error"]
+          return unless error
+
+          # Some proxies return {"error": "message"} instead of {"error": {"type": ..., "message": ...}}
+          return [500, error.to_s] unless error.is_a?(Hash)
+
+          case error["type"]
+          when "server_error"
+            [500, error["message"]]
+          when "rate_limit_exceeded", "insufficient_quota"
+            [429, error["message"]]
+          else
+            [400, error["message"]]
+          end
+        end
+        # rubocop:enable Style/ModuleFunction
+      end
+    end
+  end
+end

data/lib/swarm_sdk/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module SwarmSDK
-  VERSION = "2.7.14"
+  VERSION = "2.7.15"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: swarm_sdk
 version: !ruby/object:Gem::Version
-  version: 2.7.14
+  version: 2.7.15
 platform: ruby
 authors:
 - Paulo Arruda
@@ -183,7 +183,9 @@ files:
 - lib/swarm_sdk/ruby_llm_patches/io_endpoint_patch.rb
 - lib/swarm_sdk/ruby_llm_patches/mcp_ssl_patch.rb
 - lib/swarm_sdk/ruby_llm_patches/message_management_patch.rb
+- lib/swarm_sdk/ruby_llm_patches/openai_thought_signature_patch.rb
 - lib/swarm_sdk/ruby_llm_patches/responses_api_patch.rb
+- lib/swarm_sdk/ruby_llm_patches/streaming_error_patch.rb
 - lib/swarm_sdk/ruby_llm_patches/tool_concurrency_patch.rb
 - lib/swarm_sdk/snapshot.rb
 - lib/swarm_sdk/snapshot_from_events.rb