runapi-grok-imagine 0.2.14 → 0.2.15

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9ab7cfd304ca617d40eba1ae9daf821f868c5f2f4bf1dd8b7093c527353cd2f2
-  data.tar.gz: 59af0fe8b7f046d2e7d4ce05ab3ae1f169feb8ffe737775c8e63f4655db6ad38
+  metadata.gz: 9977ae570c6a5ffe4608e156f1f8f752a5ef8f1ad3153eb9995d50d8544d072c
+  data.tar.gz: 6e860ff0abebbbc3afca07b44df6eeb24bbef7ca3138470d1fc13d3448bf0f5c
 SHA512:
-  metadata.gz: 70f90be7565d27d6f9e57f560ac33e6b4f67c36f5c6e6d3ba0c03d88912018dcfe6baf1c77f1a34d284efcbdad3a5d95917273d8964b506727582675e6069216
-  data.tar.gz: c59bab1a8968d004599a6ff0d89106f74647d04c811c7e578af2ac75f61af89a0255c6c212cb80cccbfc4e92a9d41bbfa78fe39e9f5f68956cad5ab912b4e3ca
+  metadata.gz: 88353615d590fe3779b74fd8d13afcc8fbe41037f390bbb4dd60c8a1de343af0d72179723886f0178adcca1f5755ce63b05209b145d44bd7f32676b45268fcf2
+  data.tar.gz: 745f3f6e1af60ef8a3d273cf09e87586028cfe7a6d1d257a6fe9cc064f86d0b2edcccf06cfc12dbe178b726d7234c4087dcb9bd09beae78a4beee7d6bd2bf0e8

data/lib/runapi/grok_imagine/client.rb

@@ -18,6 +18,8 @@ module RunApi
       attr_reader :image_to_video
       # @return [Resources::TextToImage] Text-to-image generation operations.
       attr_reader :text_to_image
+      # @return [Resources::SegmentMap] Segmentation map generation operations.
+      attr_reader :segment_map
       # @return [Resources::EditImage] Prompt-guided image editing operations.
       attr_reader :edit_image
       # @return [Resources::Extensions] Extend a previously generated video.
@@ -31,6 +33,7 @@ module RunApi
         @text_to_video = Resources::TextToVideo.new(http)
         @image_to_video = Resources::ImageToVideo.new(http)
         @text_to_image = Resources::TextToImage.new(http)
+        @segment_map = Resources::SegmentMap.new(http)
         @edit_image = Resources::EditImage.new(http)
         @extensions = Resources::Extensions.new(http)
         @upscales = Resources::Upscales.new(http)

data/lib/runapi/grok_imagine/contract_gen.rb

@@ -4,7 +4,7 @@ module RunApi
   module GrokImagine
     CONTRACT = {
       "edit-image" => {
-        "models" => ["grok-imagine-edit-image"],
+        "models" => ["grok-imagine-edit-image", "grok-imagine-image-2-0"],
         "fields_by_model" => {
           "grok-imagine-edit-image" => {
             "model" => {
@@ -13,8 +13,33 @@ module RunApi
             "source_image_url" => {
               "required" => true
             }
+          },
+          "grok-imagine-image-2-0" => {
+            "mask_indices" => {
+              "min_items" => 1
+            },
+            "model" => {
+              "required" => true
+            },
+            "prompt" => {
+              "required" => true
+            },
+            "source_task_id" => {
+              "required" => true
+            }
           }
-        }
+        },
+        "rules" => [{
+          "when" => {
+            "model" => "grok-imagine-edit-image"
+          },
+          "forbidden" => ["source_task_id", "mask_indices"]
+        }, {
+          "when" => {
+            "model" => "grok-imagine-image-2-0"
+          },
+          "forbidden" => ["source_image_url", "enable_safety_checker"]
+        }]
       },
       "extend" => {
         "models" => [],
@@ -146,9 +171,34 @@ module RunApi
           "forbidden" => ["source_task_id", "index", "motion_style", "enable_safety_checker"]
         }]
       },
+      "segment-map" => {
+        "models" => ["grok-imagine-image-2-0"],
+        "fields_by_model" => {
+          "grok-imagine-image-2-0" => {
+            "model" => {
+              "required" => true
+            },
+            "source_task_id" => {
+              "required" => true
+            }
+          }
+        }
+      },
       "text-to-image" => {
-        "models" => ["grok-imagine-text-to-image"],
+        "models" => ["grok-imagine-image-2-0", "grok-imagine-text-to-image"],
         "fields_by_model" => {
+          "grok-imagine-image-2-0" => {
+            "aspect_ratio" => {
+              "enum" => ["1:1", "2:3", "3:2", "16:9", "9:16"],
+              "required" => true
+            },
+            "model" => {
+              "required" => true
+            },
+            "prompt" => {
+              "required" => true
+            }
+          },
           "grok-imagine-text-to-image" => {
             "aspect_ratio" => {
               "enum" => ["2:3", "3:2", "1:1", "16:9", "9:16"]
@@ -162,7 +212,13 @@ module RunApi
               "length" => true
             }
           }
-        }
+        },
+        "rules" => [{
+          "when" => {
+            "model" => "grok-imagine-image-2-0"
+          },
+          "forbidden" => ["enable_safety_checker", "enable_pro"]
+        }]
       },
       "text-to-video" => {
         "models" => ["grok-imagine-text-to-video", "grok-imagine-video-1.5-fast", "grok-imagine-video-1.5-preview"],

data/lib/runapi/grok_imagine/resources/edit_image.rb

@@ -35,7 +35,6 @@ module RunApi
 
         def validate_params!(params)
           validate_contract!(CONTRACT["edit-image"], params)
-          raise Core::ValidationError, "source_image_url is required" unless param(params, :source_image_url)
         end
       end
     end

data/lib/runapi/grok_imagine/resources/segment_map.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+module RunApi
+  module GrokImagine
+    module Resources
+      # Produces an editable segmentation map from an Image 2.0 text-to-image task.
+      class SegmentMap
+        include RunApi::Core::ResourceHelpers
+
+        ENDPOINT = "/api/v1/grok_imagine/segment_map"
+
+        RESPONSE_CLASS = Types::SegmentMapTaskResponse
+        COMPLETED_RESPONSE_CLASS = Types::CompletedSegmentMapTaskResponse
+
+        def initialize(http)
+          @http = http
+        end
+
+        def run(options: nil, **params)
+          task = create(options: options, **params)
+          poll_until_complete { get(task.id, options: options) }
+        end
+
+        def create(options: nil, **params)
+          params = compact_params(params)
+          validate_params!(params)
+          request(:post, ENDPOINT, body: params, options: options)
+        end
+
+        def get(id, options: nil)
+          request(:get, "#{ENDPOINT}/#{id}", options: options)
+        end
+
+        private
+
+        def validate_params!(params)
+          validate_contract!(CONTRACT["segment-map"], params)
+        end
+      end
+    end
+  end
+end

data/lib/runapi/grok_imagine/types.rb

@@ -37,6 +37,21 @@ module RunApi
       class CompletedImageTaskResponse < ImageTaskResponse
         required :images, [-> { MediaUrl }]
       end
+
+      class Segment < RunApi::Core::BaseModel
+        required :url, String
+        optional :name, String
+        optional :index, Integer
+      end
+
+      class SegmentMapTaskResponse < AsyncTaskResponse
+        optional :segments, [-> { Segment }]
+        optional :error, String
+      end
+
+      class CompletedSegmentMapTaskResponse < SegmentMapTaskResponse
+        required :segments, [-> { Segment }]
+      end
     end
   end
 end

data/lib/runapi/grok_imagine.rb

@@ -6,6 +6,7 @@ require_relative "grok_imagine/contract_gen"
 require_relative "grok_imagine/resources/text_to_video"
 require_relative "grok_imagine/resources/image_to_video"
 require_relative "grok_imagine/resources/text_to_image"
+require_relative "grok_imagine/resources/segment_map"
 require_relative "grok_imagine/resources/edit_image"
 require_relative "grok_imagine/resources/extensions"
 require_relative "grok_imagine/resources/upscales"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: runapi-grok-imagine
 version: !ruby/object:Gem::Version
-  version: 0.2.14
+  version: 0.2.15
 platform: ruby
 authors:
 - RunAPI
@@ -43,6 +43,7 @@ files:
 - lib/runapi/grok_imagine/resources/edit_image.rb
 - lib/runapi/grok_imagine/resources/extensions.rb
 - lib/runapi/grok_imagine/resources/image_to_video.rb
+- lib/runapi/grok_imagine/resources/segment_map.rb
 - lib/runapi/grok_imagine/resources/text_to_image.rb
 - lib/runapi/grok_imagine/resources/text_to_video.rb
 - lib/runapi/grok_imagine/resources/upscales.rb
@@ -71,7 +72,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 4.0.10
+rubygems_version: 4.0.17
 specification_version: 4
 summary: Grok Imagine Ruby SDK for RunAPI
 test_files: []