swarm_sdk 2.7.4 → 2.7.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b5c93c6e9a850d624b8ac0fe9d31e53a4a764a1caa23319efcfa5b34edfd33b8
-  data.tar.gz: 2d4ec3258101a45507cbacbd98eeb86d8448627e35faedd2a693c0a18d8622a1
+  metadata.gz: 6aaba8cabb5957715e88ee02e40e9492fe3546791c806d1ddaff95e967f77aba
+  data.tar.gz: 752d15e6e69bb1a9941212d1c33622d678dcc0dc734b88b77a845c7fca4e982b
 SHA512:
-  metadata.gz: ac8f8c8dea846e5b0783b49dace5f2e01a0908ebed31eb377fdd4217480e99499fe9eb4bda14772cbfd71d52badc642687434a214f76cf136768c21967e1beff
-  data.tar.gz: 61a58b209810484c7bcf24727feaa8fb0e23aafc485dbb3ad8a510c095caeaa72778b0e3c97b22eb599a2d9153ef95093dc9aa6a3b34e717b23c4c5d9797b490
+  metadata.gz: 36532a008bbb328cddf60bca55d7d8e210b2c0701ca546190164a16067d3878e26c6c4a6d347e9f262669c34289bad7ac0ddd714062a6d4ab9807e0ef5fd9dc5
+  data.tar.gz: e7e512794d27d5bd378ea1c96e66f18683cbab8c95e3820672bd74174e0fd5963894958a61aa2e7cf40485c908a7a7adf6a7cbda9d7bfdd480ccd0c3156e4d15

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/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module SwarmSDK
-  VERSION = "2.7.4"
+  VERSION = "2.7.5"
 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.5
 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