gpt-function 0.5.0 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4382f4c9861d57510b154ab5179854eee8fad548dd849f3b283d870f736e599e
-  data.tar.gz: 498b2499d3d6ecdb03847a6b5c15f9d2f99c231d59655f3cf8a602898d525102
+  metadata.gz: 61133cf0bb081bfcc03dc1e0b2a69bbc7f043c7dc6fd9c5e56d0d9734c975575
+  data.tar.gz: b153cad22540e8b2ad46293c8dffc1aa63b27d48f87a60cb46a9d6cc869abc5d
 SHA512:
-  metadata.gz: 6ba7a96725e99edb9d87f9467011357fbd8e8e40dc62f902420b480334af7b432bf071e1b63df7ce5e0645611bcb6355caff08d275e9905a21a638ec6033cc06
-  data.tar.gz: 680facff33d5f9737fb4dbc6cfc10a08e35f221245cb1e90c82a3e280c4ed32a0f1129d52561e471ca210f984a4c973b671aec01445d7776fdb5815e1d25339b
+  metadata.gz: 69fed6886b79e1b3e0034d3de36273b0bddc6b1ff59bb03067b737621ee85030ad334ceea48d72b9173025661f92f3f05d396d921720a4ee45a9f74dc7e9da50
+  data.tar.gz: f375fa50fc60c54663e8c488385541caa711ee97cae76495c885b734ce826fe79dc703d13f3594e54a7d44371be6e1f7458990a527b1095fe562a7195b011df4

data/README.md

@@ -4,11 +4,6 @@
 
 你可以確保每次呼叫 GPT 函數時，接近 100% 都會得到相同的結果。
 
-目前能夠使用的模型有：
-
-- gpt-4-1106-preview
-- gpt-3.5-turbo-1106
-
 
 ## Installation
 
@@ -99,20 +94,31 @@ end
 Batch Storage 整合 Active Record 的範例：
 
 ```ruby
-class Model < ApplicationRecord
+class BatchStatus < ApplicationRecord
   class << self
     def enqueue(hash)
-      create!(hash)
+      model = BatchStatus.new
+      model.batch_id = hash[:id]
+      model.status = hash[:status]
+      model.request_counts_completed = hash[:request_counts_completed]
+      model.request_counts_failed = hash[:request_counts_failed]
+      model.request_counts_total = hash[:request_counts_total]
+      model.metadata = hash[:metadata]
+      model.payload = hash
+      model.save
       true
     end
 
     def dequeue
-      first&.destroy
+      model = first
+      model.destroy
+      model.payload
     end
   end
 end
 
-GptFunction.configure(api_key: '...', model: 'gpt-4o-mini', batch_storage: Model)
+
+GptFunction.configure(api_key: '...', model: 'gpt-4o-mini', batch_storage: BatchStatus)
 ```
 
 ## License

data/lib/gpt_function/batch.rb

@@ -102,6 +102,11 @@ class GptFunction
       @output_file ||= File.from_id(output_file_id)
     end
 
+    def error_file
+      return nil if error_file_id.nil?
+      @error_file ||= File.from_id(error_file_id)
+    end
+
     def input_jsonl
       @input_jsonl ||= input_file&.jsonl || []
     end
@@ -167,6 +172,10 @@ class GptFunction
       ["failed", "completed", "expired", "cancelled"].include? status
     end
 
+    def auto_delete
+      metadata&.dig("auto_delete") || false
+    end
+
     class << self
       def list(limit: 20, after: nil)
         # 創建批次請求
@@ -287,7 +296,13 @@ class GptFunction
           yield batch
 
           # 如果 batch 還未處理完成，將批次請求重新加入 Storage
-          batch.enqueue unless batch.is_processed
+          if batch.is_processed && auto_delete
+            batch&.input_file&.delete rescue nil
+            batch&.output_file&.delete rescue nil
+            batch&.error_file&.delete rescue nil
+          else
+            batch.enqueue
+          end
         end
 
         true

data/lib/gpt_function/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: gpt-function
 version: !ruby/object:Gem::Version
-  version: 0.5.0
+  version: 0.6.0
 platform: ruby
 authors:
 - etrex kuo
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-08-03 00:00:00.000000000 Z
+date: 2024-08-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: dotenv