runapi-core 0.2.5 → 0.2.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 43b27ac4a9c29db093d7efcad4686efe7730aa9ddb906a956c0f2a4b353e0ccd
-  data.tar.gz: a42105c4a366fa9a17642f519392ae6dd50978a32ae73fae6c62f38f8b02cb28
+  metadata.gz: b0f298f55ee5728667cb83a8db61391cd20bdc64582faec2207d5db9ce79c384
+  data.tar.gz: fe0c644ac7dd1377e0c4b0c0fb4b545afc0ff175ac83c3c3886b5219d93aed4a
 SHA512:
-  metadata.gz: 9bee66f588446e79bbbf8eea48964bb2505bfee41c4dc9c68ce1622359e0a536e498a895c065ef8efb54c2ee41b5df19b8d7c55b8707d181c3585d2887cb29b5
-  data.tar.gz: bb4dc1ff3c4f564c89bc20e4838263864704d54a80f315f093dba9d77e6efabce80bc4087748b816aec20ac0674141e30b1d3ee632e6d92b1f006bb4c07d2b3b
+  metadata.gz: 5d473dc453e4dba6a452c54901ad2274453464de990d34c3c2c685a92d24cdb8fb9b07d6e57086913e560f034f11e5a5deea40f030bf1c1909a02724907ac4af
+  data.tar.gz: 47ef6424ffd81307922ef8adfc7c42286b6b3cd69527478b28f0674ac23332f212c10682564da8ad23bdd3b5dc0728ae7f0ce678f8814ae4f7e40736392ffdc8

data/README.md

@@ -12,6 +12,18 @@ gem install runapi-core
 
 Use the core gem for common client options, error classes, request helpers, and task polling behavior that model SDKs share. Public SDK docs live at https://runapi.ai/docs#runapi-sdks and the model catalog lives at https://runapi.ai/models.
 
+## File Upload
+
+```ruby
+client = RunApi::NanoBanana::Client.new(api_key: ENV["RUNAPI_API_KEY"])
+
+upload = client.files.create(source: {type: "url", url: "https://example.com/photo.jpg"})
+puts upload.url
+```
+
+> [!IMPORTANT]
+> Uploaded file URLs expire 1 hour after creation. Pass them to a model promptly rather than storing them for later use.
+
 ## License
 
 Licensed under the Apache License, Version 2.0.

data/lib/runapi/core/account.rb

@@ -0,0 +1,44 @@
+# frozen_string_literal: true
+
+module RunApi
+  module Core
+    class Account
+      include RunApi::Core::ResourceHelpers
+
+      INFO_ENDPOINT = "/api/v1/me"
+      BALANCE_ENDPOINT = "/api/v1/me/balance"
+
+      class AccountRecord < RunApi::Core::BaseModel
+        required :id, Integer
+        required :name, String
+      end
+
+      class InfoResponse < RunApi::Core::BaseModel
+        required :id, Integer
+        required :name, String
+        required :email, String
+        required :account, AccountRecord
+      end
+
+      class BalanceResponse < RunApi::Core::BaseModel
+        required :balance_cents, Integer
+        required :paid_balance_cents, Integer
+        required :bonus_balance_cents, Integer
+        required :spent_cents_today, Integer
+        required :spent_cents_total, Integer
+      end
+
+      def initialize(http)
+        @http = http
+      end
+
+      def info(options: nil)
+        request(:get, INFO_ENDPOINT, options:, response_class: InfoResponse)
+      end
+
+      def balance(options: nil)
+        request(:get, BALANCE_ENDPOINT, options:, response_class: BalanceResponse)
+      end
+    end
+  end
+end

data/lib/runapi/core/client.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+module RunApi
+  module Core
+    # Base class for every RunAPI client. Resolves the API key, builds the shared
+    # HTTP client, and exposes the Universal Resources (file upload, account) that
+    # are available on any client regardless of which model gem was required.
+    #
+    # Provider clients inherit from this and build their model resources from the
+    # protected +http+ reader.
+    class Client
+      # @return [Files] Temporary file upload operations.
+      attr_reader :files
+      # @return [Account] Account info and balance operations.
+      attr_reader :account
+
+      def initialize(api_key: nil, **options)
+        @api_key = Core::Auth.resolve_api_key(api_key)
+        client_options = Core::ClientOptions.new(api_key: @api_key, **options)
+        @http = client_options.http_client || Core::HttpClient.new(client_options)
+        @files = Files.new(@http)
+        @account = Account.new(@http)
+      end
+
+      protected
+
+      attr_reader :http
+    end
+  end
+end

data/lib/runapi/core/files.rb

@@ -0,0 +1,65 @@
+# frozen_string_literal: true
+
+module RunApi
+  module Core
+    class Files
+      include RunApi::Core::ResourceHelpers
+
+      ENDPOINT = "/api/v1/files"
+
+      class UploadResponse < RunApi::Core::BaseModel
+        required :file_name, String
+        required :url, String
+        required :size_bytes, Integer
+        required :mime_type, String
+        required :created_at, String
+        required :expires_at, String
+      end
+
+      RESPONSE_CLASS = UploadResponse
+
+      def initialize(http)
+        @http = http
+      end
+
+      def create(file: nil, source: nil, file_name: nil, options: nil)
+        validate_source!(file:, source:)
+
+        body = if file
+          multipart_body(file, file_name:)
+        else
+          compact_params(source:, file_name:)
+        end
+
+        request(:post, ENDPOINT, body:, options:)
+      end
+
+      private
+
+      def validate_source!(file:, source:)
+        source_count = [file, source].count { |value| !value.nil? }
+        return if source_count == 1
+
+        raise ArgumentError, "Exactly one source is required: file or source"
+      end
+
+      def multipart_body(file, file_name:)
+        path = file_path(file)
+        filename = file_name || File.basename(path)
+        Core::MultipartBody.new(
+          fields: compact_params(file_name: file_name),
+          files: {
+            file: Core::MultipartFile.new(path:, filename:)
+          }
+        )
+      end
+
+      def file_path(file)
+        return file if file.is_a?(String)
+        return file.path if file.respond_to?(:path)
+
+        raise ArgumentError, "file must be a file path or respond to :path"
+      end
+    end
+  end
+end

data/lib/runapi/core/http_client.rb

@@ -50,6 +50,8 @@ module RunApi
 
           raise error
         end
+      ensure
+        close_multipart_files(req)
       end
 
       private
@@ -69,16 +71,45 @@ module RunApi
         req = klass.new(uri.request_uri)
 
         req["Authorization"] = "Bearer #{@options.api_key}"
-        req["Content-Type"] = "application/json"
         req["Accept"] = "application/json"
         req["User-Agent"] = Constants::SDK_USER_AGENT
 
         options&.headers&.each { |k, v| req[k.to_s] = v }
 
-        req.body = JSON.generate(body) if body
+        if body.is_a?(MultipartBody)
+          req.set_form(multipart_parts(body), "multipart/form-data")
+        elsif body
+          req["Content-Type"] = "application/json"
+          req.body = JSON.generate(body)
+        end
         req
       end
 
+      def multipart_parts(body)
+        opened_files = []
+        field_parts = body.fields.map { |key, value| [key, value.to_s] }
+        file_parts = body.files.map do |key, file|
+          options = {filename: file.filename}
+          options[:content_type] = file.content_type if file.content_type
+          opened_files << File.open(file.path, "rb")
+          [key, opened_files.last, options]
+        end
+        field_parts + file_parts
+      rescue
+        opened_files.each { |file| file.close unless file.closed? }
+        raise
+      end
+
+      def close_multipart_files(request)
+        body_data = request&.instance_variable_get(:@body_data)
+        return unless body_data
+
+        body_data.each do |part|
+          file = part[1]
+          file.close if file.is_a?(File) && !file.closed?
+        end
+      end
+
       def retryable?(method, status)
         Constants::IDEMPOTENT_METHODS.include?(method.to_s.upcase) &&
           Constants::RETRYABLE_STATUS_CODES.include?(status)

data/lib/runapi/core/multipart_body.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+module RunApi
+  module Core
+    MultipartFile = Struct.new(:path, :filename, :content_type, keyword_init: true)
+
+    class MultipartBody
+      attr_reader :fields, :files
+
+      def initialize(fields: {}, files: {})
+        @fields = stringify_keys(fields)
+        @files = stringify_keys(files)
+      end
+
+      private
+
+      def stringify_keys(hash)
+        hash.each_with_object({}) do |(key, value), result|
+          result[key.to_s] = value
+        end
+      end
+    end
+  end
+end

data/lib/runapi/core/version.rb

@@ -2,7 +2,7 @@
 
 module RunApi
   module Core
-    VERSION = "0.2.5"
+    VERSION = "0.2.6"
   end
 
   VERSION = Core::VERSION

data/lib/runapi/core.rb

@@ -14,6 +14,10 @@ require_relative "core/base_model"
 require_relative "core/types"
 require_relative "core/errors"
 require_relative "core/auth"
+require_relative "core/multipart_body"
 require_relative "core/http_client"
 require_relative "core/polling"
 require_relative "core/resource_helpers"
+require_relative "core/files"
+require_relative "core/account"
+require_relative "core/client"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: runapi-core
 version: !ruby/object:Gem::Version
-  version: 0.2.5
+  version: 0.2.6
 platform: ruby
 authors:
 - RunAPI
@@ -38,12 +38,16 @@ files:
 - README.md
 - lib/runapi-core.rb
 - lib/runapi/core.rb
+- lib/runapi/core/account.rb
 - lib/runapi/core/auth.rb
 - lib/runapi/core/base_model.rb
+- lib/runapi/core/client.rb
 - lib/runapi/core/configuration.rb
 - lib/runapi/core/constants.rb
 - lib/runapi/core/errors.rb
+- lib/runapi/core/files.rb
 - lib/runapi/core/http_client.rb
+- lib/runapi/core/multipart_body.rb
 - lib/runapi/core/polling.rb
 - lib/runapi/core/resource_helpers.rb
 - lib/runapi/core/types.rb