zuno 1.0.0 → 1.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d1dea919bc404d8bafb4e970ffd4d2d418d5ea38123bc78abc4668fc72ce985d
-  data.tar.gz: 4953724ce6061eb7641d7e4c5ee7564204d4647968822fd8d88303b481cf09c1
+  metadata.gz: 7d6d4d8dd97c7ac8743ddc66545aeda41ed824357965c693ac244fcf46de5df2
+  data.tar.gz: d7e0a1a6029482af54dbf7bc6e5196b39dd60e424b1fb18b4003543cfbe79867
 SHA512:
-  metadata.gz: 16d264d3143fca0a55bb4a2ed1ad4cc08e0f8dbbb6effdf8ed75c7425b5087fd1a8763135117bcc3b1ea8cf9061a579fa95b21e813b81d90c7de1076978d3a32
-  data.tar.gz: 8f19a868f736ec65ddf5484affe0a538a54690fc420c234afa7ead82f6943040441bce2336b334becb14fedad7babaaec03025854bf7e638ee31330f4eadddbb
+  metadata.gz: 4862131a38d657f175bbc488599d5cbcd97899b43bdac7a26117a437c081bfdaa365e4a2e5255e7e48464aa5e5d129e14106364727b19bae4b8166ff779cd621
+  data.tar.gz: 96340e7d0b4a2153d58b328cbdcbec5836ddd76724941fd8efa39343de14c052a51c2031e916562e0b66de5044832a429f2332c92e4325f40d6eefc2d1e67454

data/lib/zuno/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Zuno
-  VERSION = "1.0.0"
+  VERSION = "1.0.1"
 end

data/lib/zuno.rb

@@ -76,6 +76,7 @@ module Zuno
   end
 
   OPENROUTER_ADAPTER_CONFIG_KEYS = %i[api_key app_url title timeout].freeze
+  AI_GATEWAY_ADAPTER_CONFIG_KEYS = %i[api_key timeout base_url].freeze
   REPLICATE_ADAPTER_CONFIG_KEYS = %i[api_key timeout].freeze
   DEFAULT_MAX_ITERATIONS = 1
   REPLICATE_PREFER_WAIT_SECONDS = 60
@@ -113,6 +114,18 @@ module Zuno
     )
   end
 
+  def ai_gateway(
+    api_key: nil,
+    timeout: Providers::AIGateway::DEFAULT_TIMEOUT,
+    base_url: Providers::AIGateway::DEFAULT_BASE_URL
+  )
+    Providers::AIGateway.new(
+      api_key: api_key,
+      timeout: timeout,
+      base_url: base_url
+    )
+  end
+
   def tool(name:, description:, input_schema:, &execute)
     raise ToolError, "A block is required for tool '#{name}'" unless block_given?
 
@@ -176,7 +189,7 @@ module Zuno
 
     result =
       case provider
-      when :openrouter
+      when :openrouter, :ai_gateway
         adapter = provider_adapter(provider, resolved_provider_options)
         generate_openrouter_single(
           model_descriptor: model_descriptor,
@@ -246,7 +259,9 @@ module Zuno
   )
     callback_control = nil
     model_descriptor = normalize_model(model)
-    raise Error, "loop only supports openrouter provider" unless model_descriptor.provider.to_sym == :openrouter
+    unless %i[openrouter ai_gateway].include?(model_descriptor.provider.to_sym)
+      raise Error, "loop only supports openrouter or ai_gateway provider"
+    end
 
     resolved_provider_options = merge_provider_options(
       model_descriptor.provider_options,
@@ -320,6 +335,7 @@ module Zuno
 
       payload = build_payload(
         model_id: model_descriptor.id,
+        provider: model_descriptor.provider,
         messages: llm_messages,
         tools: tool_map,
         tool_choice: resolved_tool_choice,
@@ -497,7 +513,9 @@ module Zuno
     raise ArgumentError, "stream requires a block callback" unless block_given?
 
     model_descriptor = normalize_model(model)
-    raise ProviderError, "stream only supports openrouter provider" unless model_descriptor.provider.to_sym == :openrouter
+    unless %i[openrouter ai_gateway].include?(model_descriptor.provider.to_sym)
+      raise ProviderError, "stream only supports openrouter or ai_gateway provider"
+    end
 
     resolved_provider_options = merge_provider_options(
       model_descriptor.provider_options,
@@ -508,6 +526,7 @@ module Zuno
 
     payload = build_payload(
       model_id: model_descriptor.id,
+      provider: model_descriptor.provider,
       messages: llm_messages,
       tools: {},
       tool_choice: nil,
@@ -608,6 +627,7 @@ module Zuno
 
     payload = build_payload(
       model_id: model_descriptor.id,
+      provider: model_descriptor.provider,
       messages: llm_messages,
       tools: tool_map,
       tool_choice: resolved_tool_choice,
@@ -795,7 +815,7 @@ module Zuno
   end
   private_class_method :normalize_messages
 
-  def build_payload(model_id:, messages:, tools:, tool_choice:, temperature:, max_tokens:, provider_options:)
+  def build_payload(model_id:, provider:, messages:, tools:, tool_choice:, temperature:, max_tokens:, provider_options:)
     payload = {
       "model" => model_id,
       "messages" => messages
@@ -806,12 +826,26 @@ module Zuno
     payload["tools"] = tools.values.map(&:as_provider_tool) unless tools.empty?
     payload["tool_choice"] = deep_stringify(tool_choice) unless tool_choice.nil?
 
-    request_options = reject_keys(provider_options, OPENROUTER_ADAPTER_CONFIG_KEYS + [ :tool_choice ])
+    request_options = reject_keys(provider_options, provider_adapter_config_keys(provider) + [ :tool_choice ])
     payload.merge!(deep_stringify(request_options)) if request_options.is_a?(Hash)
     payload
   end
   private_class_method :build_payload
 
+  def provider_adapter_config_keys(provider)
+    case provider.to_sym
+    when :openrouter
+      OPENROUTER_ADAPTER_CONFIG_KEYS
+    when :ai_gateway
+      AI_GATEWAY_ADAPTER_CONFIG_KEYS
+    when :replicate
+      REPLICATE_ADAPTER_CONFIG_KEYS
+    else
+      []
+    end
+  end
+  private_class_method :provider_adapter_config_keys
+
   def merge_provider_options(model_provider_options, call_provider_options)
     merged = {}
     merged.merge!(default_provider_options) if default_provider_options.is_a?(Hash)
@@ -826,6 +860,9 @@ module Zuno
     when :openrouter
       config = pick_keys(provider_options, OPENROUTER_ADAPTER_CONFIG_KEYS)
       Providers::OpenRouter.new(**config)
+    when :ai_gateway
+      config = pick_keys(provider_options, AI_GATEWAY_ADAPTER_CONFIG_KEYS)
+      Providers::AIGateway.new(**config)
     when :replicate
       config = pick_keys(provider_options, REPLICATE_ADAPTER_CONFIG_KEYS)
       Providers::Replicate.new(**config)
@@ -1243,6 +1280,105 @@ module Zuno
 
     end
 
+    class AIGateway
+      DEFAULT_BASE_URL = "https://ai-gateway.vercel.sh/v1".freeze
+      DEFAULT_TIMEOUT = 120_000
+
+      def initialize(api_key: nil, timeout: DEFAULT_TIMEOUT, base_url: DEFAULT_BASE_URL)
+        @api_key = api_key
+        raise ProviderError, "Vercel Gateway API key not configured" if @api_key.nil? || @api_key.to_s.empty?
+
+        @timeout = timeout
+        @base_url = base_url.to_s.empty? ? DEFAULT_BASE_URL : base_url.to_s
+      end
+
+      def model(model_id)
+        ModelDescriptor.new(
+          id: model_id,
+          provider: :ai_gateway,
+          provider_options: provider_options
+        )
+      end
+
+      def chat(payload)
+        response = Typhoeus.post(
+          chat_completions_url,
+          headers: headers,
+          body: JSON.generate(payload),
+          timeout: @timeout
+        )
+
+        validate_response!(response)
+        parsed = JSON.parse(response.body)
+        raise ProviderError, "Vercel Gateway returned invalid JSON" unless parsed.is_a?(Hash)
+
+        parsed
+      rescue JSON::ParserError => e
+        raise ProviderError, "Failed to parse Vercel Gateway response: #{e.message}"
+      end
+
+      def stream(payload)
+        raise ArgumentError, "stream requires a block callback" unless block_given?
+
+        request = Typhoeus::Request.new(
+          chat_completions_url,
+          method: :post,
+          headers: headers,
+          body: JSON.generate(payload),
+          timeout: @timeout
+        )
+
+        parser = SseParser.new { |data| yield(data) }
+        request.on_body do |chunk|
+          parser.push(chunk)
+          nil
+        end
+
+        request.run
+        validate_response!(request.response)
+        parser.flush
+      end
+
+      private
+
+      def provider_options
+        {
+          api_key: @api_key,
+          timeout: @timeout,
+          base_url: @base_url
+        }
+      end
+
+      def chat_completions_url
+        "#{@base_url}/chat/completions"
+      end
+
+      def headers
+        {
+          "Authorization" => "Bearer #{@api_key}",
+          "Content-Type" => "application/json"
+        }
+      end
+
+      def validate_response!(response)
+        raise ProviderError, "No response returned from Vercel Gateway" if response.nil?
+        raise ProviderError, "Vercel Gateway request timed out" if response.timed_out?
+
+        status = response.code.to_i
+        body = response.body.to_s
+        message = body.length > 300 ? "#{body[0, 300]}..." : body
+
+        return if status >= 200 && status < 300
+
+        if status.positive?
+          raise ProviderError, "Vercel Gateway responded with HTTP #{status}: #{message}"
+        end
+
+        suffix = message.empty? ? "" : ": #{message}"
+        raise ProviderError, "Vercel Gateway request failed: #{response.return_code}#{suffix}"
+      end
+    end
+
     class Replicate
       API_BASE_URL = "https://api.replicate.com/v1".freeze
       DEFAULT_TIMEOUT = 120_000

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: zuno
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.1
 platform: ruby
 authors:
 - Hyperaide