runapi-midjourney 0.2.0 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 06e85132c76348835ae96514c74883406ea74865d9d11525d7577e17c88f8e6f
-  data.tar.gz: 84922384c8d30f48b9798a0b0299e6ff4be448f44819cd955fe3aba2b54ef398
+  metadata.gz: 4d95f63c07b630fb2078c85e3f41ae11a1485b73790c8de5c50a9e8412e51930
+  data.tar.gz: 571d993e64f17ea0da765284d3f0b7d82790c1dc060a2d17e24914d793799237
 SHA512:
-  metadata.gz: 3e364a667e6b92db66d620c123d2fcf645d5a6c1af7342c8e209e5599129e720fb181603853158081cd4c45ebcfd26565a9ef11855c44b710b4cc0748c612572
-  data.tar.gz: 9651d5a10d8eb4d6024dca341b307445bf57a4864e8ebdd62a9b43995a55eadb64a138f8363ff0ef6047b7518a9021e8accae2a90325c6296cfc2b3e48a58a28
+  metadata.gz: 11d53601405856537b50c13b5132de40f4ce91cd94258b347152dcce9b0905ba69eab26611b362c573a5ec9e79859e9411e0a3b1ad89c276dd7ef3163cb95a08
+  data.tar.gz: 8d7ff802f36621b71c98bff61fe2199098899229d4ffa9a01efa397a46e81845fb535c8be2edf5d65259c9060e966da6d92f13dcbe4b46b55034a565c764cc0f

data/README.md

@@ -1,6 +1,6 @@
 # Midjourney Ruby SDK for RunAPI
 
-Use `runapi-midjourney` for Midjourney image generation, editing, image-to-video, image-to-prompt, prompt shortening, and seed lookup in Ruby applications and workers.
+Use `runapi-midjourney` for Midjourney image generation, editing, image-to-video, first-video extension, image-to-prompt, prompt shortening, and seed lookup in Ruby applications and workers.
 
 ## Install
 
@@ -21,7 +21,7 @@ result = client.text_to_image.run(
 puts result.images.first.url
 ```
 
-Use `create`, `get`, and `run` for `text_to_image`, `edit_image`, and `image_to_video`. Use `run` directly for `image_to_prompt`, `shorten_prompt`, and `get_seed`. Generated media URLs are temporary and should be stored in durable storage.
+Use `create`, `get`, and `run` for `text_to_image`, `edit_image`, `image_to_video`, and `extend_video`. `extend_video` accepts the task ID of an account-owned completed direct image-to-video task and continues its first video; extension tasks cannot be chained. Use `run` directly for `image_to_prompt`, `shorten_prompt`, and `get_seed`. Generated media URLs are temporary and should be stored in durable storage.
 
 ## Links
 

data/lib/runapi/midjourney/client.rb

@@ -4,13 +4,14 @@ module RunApi
   module Midjourney
     # Midjourney image generation, editing, image-to-video, and helper client.
     class Client < RunApi::Core::Client
-      attr_reader :text_to_image, :edit_image, :image_to_video, :image_to_prompt, :shorten_prompt, :get_seed
+      attr_reader :text_to_image, :edit_image, :image_to_video, :extend_video, :image_to_prompt, :shorten_prompt, :get_seed
 
       def initialize(api_key: nil, **options)
         super
         @text_to_image = Resources::TextToImage.new(http)
         @edit_image = Resources::EditImage.new(http)
         @image_to_video = Resources::ImageToVideo.new(http)
+        @extend_video = Resources::ExtendVideo.new(http)
         @image_to_prompt = Resources::ImageToPrompt.new(http)
         @shorten_prompt = Resources::ShortenPrompt.new(http)
         @get_seed = Resources::GetSeed.new(http)

data/lib/runapi/midjourney/contract_gen.rb

@@ -16,6 +16,16 @@ module RunApi
           }
         }
       },
+      "extend-video" => {
+        "models" => ["midjourney-image-to-video"],
+        "fields_by_model" => {
+          "midjourney-image-to-video" => {
+            "source_task_id" => {
+              "required" => true
+            }
+          }
+        }
+      },
       "get-seed" => {
         "models" => [],
         "fields_by_model" => {

data/lib/runapi/midjourney/resources/extend_video.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+module RunApi
+  module Midjourney
+    module Resources
+      # Extends the first video from a completed Midjourney image-to-video task.
+      class ExtendVideo
+        include RunApi::Core::ResourceHelpers
+
+        ENDPOINT = "/api/v1/midjourney/extend_video"
+        MODEL = "midjourney-image-to-video"
+        RESPONSE_CLASS = Types::VideoTaskResponse
+        COMPLETED_RESPONSE_CLASS = Types::CompletedVideoTaskResponse
+
+        def initialize(http) = (@http = http)
+
+        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_contract!(CONTRACT["extend-video"], params.merge(model: MODEL))
+          request(:post, ENDPOINT, body: params, options: options)
+        end
+
+        def get(id, options: nil)
+          request(:get, "#{ENDPOINT}/#{id}", options: options)
+        end
+      end
+    end
+  end
+end

data/lib/runapi/midjourney/types.rb

@@ -43,18 +43,21 @@ module RunApi
       class ImageToPromptResponse < RunApi::Core::BaseModel
         required :prompts, [String]
         optional :error, String
+        optional :billing, RunApi::Core::TaskBillingFacts
       end
 
       # Concise suggestions derived from a prompt.
       class ShortenPromptResponse < RunApi::Core::BaseModel
         required :prompts, [String]
         optional :error, String
+        optional :billing, RunApi::Core::TaskBillingFacts
       end
 
       # Seed associated with a generated image.
       class GetSeedResponse < RunApi::Core::BaseModel
         optional :seed, Integer
         optional :error, String
+        optional :billing, RunApi::Core::TaskBillingFacts
       end
     end
   end

data/lib/runapi/midjourney.rb

@@ -6,6 +6,7 @@ require_relative "midjourney/contract_gen"
 require_relative "midjourney/resources/text_to_image"
 require_relative "midjourney/resources/edit_image"
 require_relative "midjourney/resources/image_to_video"
+require_relative "midjourney/resources/extend_video"
 require_relative "midjourney/resources/image_to_prompt"
 require_relative "midjourney/resources/shorten_prompt"
 require_relative "midjourney/resources/get_seed"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: runapi-midjourney
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.1
 platform: ruby
 authors:
 - RunAPI
@@ -15,16 +15,17 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.2.13
+        version: 0.3.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.2.13
+        version: 0.3.0
 description: Use `runapi-midjourney` for Midjourney image generation, editing, image-to-video,
-  image-to-prompt, prompt shortening, and seed lookup in Ruby applications and workers.
+  first-video extension, image-to-prompt, prompt shortening, and seed lookup in Ruby
+  applications and workers.
 email:
 - contact@runapi.ai
 executables: []
@@ -39,6 +40,7 @@ files:
 - lib/runapi/midjourney/client.rb
 - lib/runapi/midjourney/contract_gen.rb
 - lib/runapi/midjourney/resources/edit_image.rb
+- lib/runapi/midjourney/resources/extend_video.rb
 - lib/runapi/midjourney/resources/get_seed.rb
 - lib/runapi/midjourney/resources/image_to_prompt.rb
 - lib/runapi/midjourney/resources/image_to_video.rb
@@ -69,7 +71,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: Midjourney Ruby SDK for RunAPI
 test_files: []