swarm_sdk 2.7.4 → 2.7.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b5c93c6e9a850d624b8ac0fe9d31e53a4a764a1caa23319efcfa5b34edfd33b8
-  data.tar.gz: 2d4ec3258101a45507cbacbd98eeb86d8448627e35faedd2a693c0a18d8622a1
+  metadata.gz: 5806264bb99cea0a71727c89c5839ff3027c17cd725e4261244e14ae9079f513
+  data.tar.gz: f3d489dd1d6045dec6b97b499b2d6e27d78440d1b2289a19c2f3bedb27440c8d
 SHA512:
-  metadata.gz: ac8f8c8dea846e5b0783b49dace5f2e01a0908ebed31eb377fdd4217480e99499fe9eb4bda14772cbfd71d52badc642687434a214f76cf136768c21967e1beff
-  data.tar.gz: 61a58b209810484c7bcf24727feaa8fb0e23aafc485dbb3ad8a510c095caeaa72778b0e3c97b22eb599a2d9153ef95093dc9aa6a3b34e717b23c4c5d9797b490
+  metadata.gz: a37edf046435540ba32aa47b5552ef48e1c599c69a0bf0540ccfa2199a01947006ffe98eb8df1848175e11b673bebbf787156815dc4e5ea9403ab0db984997bd
+  data.tar.gz: 2ca616d67f41b19fe596d86fdd36aa6255cb314480c34253aacb7d8276196fd2e2bbd4fb3057b42faec5b000474754df7cce56cddbeb26fcc3301e6fa2a89fcb

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

@@ -117,6 +117,9 @@ module SwarmSDK
         def build_custom_context(provider:, base_url:, timeout:)
           RubyLLM.context do |config|
             config.request_timeout = timeout
+            # Set read_timeout to match request_timeout for streaming support
+            # This ensures long gaps between chunks (model thinking) don't cause timeouts
+            config.read_timeout = SwarmSDK.config.llm_read_timeout || timeout
 
             configure_provider_base_url(config, provider, base_url) if base_url
           end

data/lib/swarm_sdk/config.rb

@@ -56,6 +56,15 @@ module SwarmSDK
       gpustack_api_key: [:gpustack_api_key, "GPUSTACK_API_KEY"],
     }.freeze
 
+    # RubyLLM connection settings that proxy to RubyLLM.config
+    # Maps SwarmSDK config key => [RubyLLM config key, ENV variable, default value]
+    RUBYLLM_CONNECTION_MAPPINGS = {
+      llm_request_timeout: [:request_timeout, "SWARM_SDK_LLM_REQUEST_TIMEOUT", 300],
+      llm_read_timeout: [:read_timeout, "SWARM_SDK_LLM_READ_TIMEOUT", nil], # nil = use request_timeout
+      llm_open_timeout: [:open_timeout, "SWARM_SDK_LLM_OPEN_TIMEOUT", 30],
+      llm_write_timeout: [:write_timeout, "SWARM_SDK_LLM_WRITE_TIMEOUT", 30],
+    }.freeze
+
     # SwarmSDK defaults that can be overridden
     # Maps config key => [ENV variable, default proc]
     DEFAULTS_MAPPINGS = {
@@ -152,6 +161,46 @@ module SwarmSDK
       end
     end
 
+    # ========== RubyLLM Connection Accessors (with RubyLLM proxying) ==========
+
+    # @!method llm_request_timeout
+    #   Get the LLM request timeout (seconds)
+    #   @return [Integer] The timeout (default: 300)
+    #
+    # @!method llm_read_timeout
+    #   Get the LLM read timeout (seconds) - time to wait between chunks
+    #   @return [Integer, nil] The timeout (nil = use request_timeout)
+    #
+    # @!method llm_open_timeout
+    #   Get the LLM connection open timeout (seconds)
+    #   @return [Integer] The timeout (default: 30)
+    #
+    # @!method llm_write_timeout
+    #   Get the LLM write timeout (seconds)
+    #   @return [Integer] The timeout (default: 30)
+
+    RUBYLLM_CONNECTION_MAPPINGS.each_key do |config_key|
+      ruby_llm_key, _env_key, default_value = RUBYLLM_CONNECTION_MAPPINGS[config_key]
+
+      # Getter with default fallback
+      define_method(config_key) do
+        ensure_env_loaded!
+        if @explicit_values.key?(config_key)
+          @explicit_values[config_key]
+        elsif @env_values.key?(config_key)
+          @env_values[config_key]
+        else
+          default_value
+        end
+      end
+
+      # Setter with RubyLLM proxying
+      define_method("#{config_key}=") do |value|
+        @explicit_values[config_key] = value
+        RubyLLM.config.public_send("#{ruby_llm_key}=", value)
+      end
+    end
+
     # ========== Defaults Accessors (with module constant fallback) ==========
 
     # @!method default_model
@@ -256,6 +305,18 @@ module SwarmSDK
         RubyLLM.config.public_send("#{ruby_llm_key}=", value)
       end
 
+      # Load RubyLLM connection settings and proxy to RubyLLM
+      RUBYLLM_CONNECTION_MAPPINGS.each do |config_key, (ruby_llm_key, env_key, _default)|
+        next if @explicit_values.key?(config_key)
+        next unless ENV.key?(env_key)
+
+        value = parse_env_value(ENV[env_key], config_key)
+        @env_values[config_key] = value
+
+        # Proxy to RubyLLM
+        RubyLLM.config.public_send("#{ruby_llm_key}=", value)
+      end
+
       # Load defaults (no RubyLLM proxy)
       DEFAULTS_MAPPINGS.each do |config_key, (env_key, _default_proc)|
         next if @explicit_values.key?(config_key)

data/lib/swarm_sdk/swarm/mcp_configurator.rb

@@ -45,17 +45,23 @@ module SwarmSDK
         Swarm.apply_mcp_logging_configuration
 
         mcp_server_configs.each do |server_config|
+          tools_config = server_config[:tools]
+          mode = tools_config.nil? ? :discovery : :optimized
+
+          # Emit event before initialization
+          emit_mcp_init_start(agent_name, server_config, mode)
+
           client = initialize_mcp_client(server_config)
 
           # Store client for cleanup
           @mcp_clients[agent_name] << client
 
-          tools_config = server_config[:tools]
-
           if tools_config.nil?
             # Discovery mode: Fetch all tools from server (calls tools/list)
             # client.tools returns RubyLLM::Tool instances (already wrapped by internal Coordinator)
             all_tools = client.tools
+            tool_names = all_tools.map { |t| t.respond_to?(:name) ? t.name : t.to_s }
+
             all_tools.each do |tool|
               chat.tool_registry.register(
                 tool,
@@ -63,10 +69,15 @@ module SwarmSDK
                 metadata: { server_name: server_config[:name] },
               )
             end
+
+            # Emit completion event for discovery mode
+            emit_mcp_init_complete(agent_name, server_config, mode, all_tools.size, tool_names)
             RubyLLM.logger.debug("SwarmSDK: Discovered and registered #{all_tools.size} tools from MCP server '#{server_config[:name]}'")
           else
             # Optimized mode: Create tool stubs without tools/list RPC (Plan 025)
             # Use client directly (it has internal coordinator)
+            tool_names = tools_config.map(&:to_s)
+
             tools_config.each do |tool_name|
               stub = Tools::McpToolStub.new(
                 client: client,
@@ -78,6 +89,9 @@ module SwarmSDK
                 metadata: { server_name: server_config[:name] },
               )
             end
+
+            # Emit completion event for optimized mode
+            emit_mcp_init_complete(agent_name, server_config, mode, tools_config.size, tool_names)
             RubyLLM.logger.debug("SwarmSDK: Registered #{tools_config.size} tool stubs from MCP server '#{server_config[:name]}' (lazy schema)")
           end
         rescue StandardError => e
@@ -187,6 +201,42 @@ module SwarmSDK
 
         streamable_config
       end
+
+      # Emit MCP server initialization start event
+      #
+      # @param agent_name [Symbol] Agent name
+      # @param server_config [Hash] MCP server configuration
+      # @param mode [Symbol] Initialization mode (:discovery or :optimized)
+      # @return [void]
+      def emit_mcp_init_start(agent_name, server_config, mode)
+        LogStream.emit(
+          type: "mcp_server_init_start",
+          agent: agent_name,
+          server_name: server_config[:name],
+          transport_type: server_config[:type],
+          mode: mode,
+        )
+      end
+
+      # Emit MCP server initialization complete event
+      #
+      # @param agent_name [Symbol] Agent name
+      # @param server_config [Hash] MCP server configuration
+      # @param mode [Symbol] Initialization mode (:discovery or :optimized)
+      # @param tool_count [Integer] Number of tools registered
+      # @param tool_names [Array<String>] Names of registered tools
+      # @return [void]
+      def emit_mcp_init_complete(agent_name, server_config, mode, tool_count, tool_names)
+        LogStream.emit(
+          type: "mcp_server_init_complete",
+          agent: agent_name,
+          server_name: server_config[:name],
+          transport_type: server_config[:type],
+          mode: mode,
+          tool_count: tool_count,
+          tools: tool_names,
+        )
+      end
     end
   end
 end

data/lib/swarm_sdk/tools/image_extractors/pdf_image_extractor.rb

@@ -4,8 +4,10 @@ module SwarmSDK
   module Tools
     module ImageExtractors
       # Extracts images from PDF documents
-      # Supports JPEG (DCTDecode), FlateDecode, and LZWDecode formats
-      # Converts non-JPEG images to TIFF format
+      # Only extracts JPEG images (DCTDecode format) which are LLM API compatible
+      # Non-JPEG images (FlateDecode, LZWDecode) are skipped because they would
+      # require TIFF format which is not supported by LLM APIs
+      # Supported LLM image formats: ['png', 'jpeg', 'gif', 'webp']
       class PdfImageExtractor
         class << self
           # Extract all images from a PDF document
@@ -65,11 +67,13 @@ module SwarmSDK
 
             case filter
             when :DCTDecode
-              # JPEG images can be saved directly
+              # JPEG images can be saved directly - LLM API compatible
               save_jpeg(stream, page_number, name, temp_dir)
             when :FlateDecode, :LZWDecode, nil
-              # Raw or compressed formats - save as TIFF
-              save_as_tiff(stream, page_number, name, temp_dir)
+              # Skip non-JPEG images to avoid TIFF format (not supported by LLM APIs)
+              # LLM APIs only support: ['png', 'jpeg', 'gif', 'webp']
+              # These images would require TIFF conversion which causes API errors
+              nil
             end
             # Unsupported formats return nil
           rescue StandardError

data/lib/swarm_sdk/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: swarm_sdk
 version: !ruby/object:Gem::Version
-  version: 2.7.4
+  version: 2.7.6
 platform: ruby
 authors:
 - Paulo Arruda
@@ -71,14 +71,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.9.6
+        version: 1.9.7
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.9.6
+        version: 1.9.7
 - !ruby/object:Gem::Dependency
   name: ruby_llm_swarm-mcp
   requirement: !ruby/object:Gem::Requirement