gpt-function 0.4.0 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cdeb301d12f6247b694accec4654d4d65bab0ca071cb4bf7ae73631bf540bc56
-  data.tar.gz: dd5cb067931ac6411a545641b96f40ef81d929bd1d388cfe93808d5c76b6e2ec
+  metadata.gz: 4382f4c9861d57510b154ab5179854eee8fad548dd849f3b283d870f736e599e
+  data.tar.gz: 498b2499d3d6ecdb03847a6b5c15f9d2f99c231d59655f3cf8a602898d525102
 SHA512:
-  metadata.gz: 665474c59064f7ae3401b0fea4e8ca562cfa2e91ff285523bbcdff9b367480f2d137b619b15d30c396df92a07c506bd73ed121bd8c54292d301785957486aa37
-  data.tar.gz: f1072362a33e579c4f52eda7b8abbadc3970807eaca91c5aa39010ca130c3e935a9ee5194b207a8c4de8b11c2079a68bbc3675d63ddeb74d811f9c2de7c33aac
+  metadata.gz: 6ba7a96725e99edb9d87f9467011357fbd8e8e40dc62f902420b480334af7b432bf071e1b63df7ce5e0645611bcb6355caff08d275e9905a21a638ec6033cc06
+  data.tar.gz: 680facff33d5f9737fb4dbc6cfc10a08e35f221245cb1e90c82a3e280c4ed32a0f1129d52561e471ca210f984a4c973b671aec01445d7776fdb5815e1d25339b

data/lib/gpt_function/batch.rb

@@ -28,8 +28,7 @@ class GptFunction
     attr_reader :request_counts_completed
     attr_reader :request_counts_failed
 
-    attr_reader :metadata_customer_id
-    attr_reader :metadata_batch_description
+    attr_reader :metadata
 
     def initialize(hash)
       @id = hash["id"]
@@ -55,8 +54,7 @@ class GptFunction
       @request_counts_completed = hash.dig("request_counts", "completed")
       @request_counts_failed = hash.dig("request_counts", "failed")
 
-      @metadata_customer_id = hash.dig("metadata", "customer_id")
-      @metadata_batch_description = hash.dig("metadata", "batch_description")
+      @metadata = hash.dig("metadata")
     end
 
     def to_hash
@@ -82,8 +80,7 @@ class GptFunction
         request_counts_total: request_counts_total,
         request_counts_completed: request_counts_completed,
         request_counts_failed: request_counts_failed,
-        metadata_customer_id: metadata_customer_id,
-        metadata_batch_description: metadata_batch_description,
+        metadata: metadata
       }
     end
 
@@ -189,7 +186,7 @@ class GptFunction
         end
       end
 
-      def create(requests)
+      def create(requests, metadata: nil)
         requests = requests.each_with_index.map do |request, index|
           {
             custom_id: "request-#{index + 1}",
@@ -206,11 +203,13 @@ class GptFunction
         uri = URI('https://api.openai.com/v1/batches')
         request = Net::HTTP::Post.new(uri, 'Content-Type' => 'application/json')
         request['Authorization'] = "Bearer #{GptFunction.api_key}"
-        request.body = {
+        body = {
           input_file_id: file.id,
           endpoint: '/v1/chat/completions',
           completion_window: '24h'
-        }.to_json
+        }
+        body[:metadata] = metadata unless metadata.nil?
+        request.body = body.to_json
 
         response = Net::HTTP.start(uri.hostname, uri.port, use_ssl: true) do |http|
           http.request(request)

data/lib/gpt_function/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 class GptFunction
-  VERSION = "0.4.0"
+  VERSION = "0.5.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: gpt-function
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.5.0
 platform: ruby
 authors:
 - etrex kuo