openai_ruby 0.3.1 → 0.3.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a4c166c6a904ae2e24c863aa5fc906f92ae6be0a01b239524c5f6ceb9f1db125
-  data.tar.gz: ce98d485c2a40eda5dd2139721bd4507e7eb9b7e947757cfe790e6061c4534a6
+  metadata.gz: 973ae690504a531c6b3b95415d622346a7fedca7e1fe7d2b509fe689fa733f7a
+  data.tar.gz: ec6b40954dccffbed531aad98b79458061eb1cba7e88b370d30a639c891352c2
 SHA512:
-  metadata.gz: e3e938b3e14930e9c12f5b4aa68e1f56e528e298a48089a6a0e96a2ae0dddff6ce6a54b66cfeec0f09512dc7d62a65117f889606bfe9570d133c629e022853c0
-  data.tar.gz: acad979520949c2d763f3834f97411b44a0532795451af51f63e72737957cae79bc219b421ed41187f3352232600914cf52e9e79fbc7f1dc1bc9234a741ce276
+  metadata.gz: 379578031e02141ee8ce4b7949cd08c84a98806cfdeb277a12e1a46f4a0394d59a402007b566ab4247bafda0daa646869866b6fc1bd29acebf338d312bbce76d
+  data.tar.gz: e8a1fc7ee1822493e11b1e1b4eb318491af75252b2873040379f102ce997961d0afa23e1a9abff40459d89a7af6a81c29418f22487287c379bcf23aae8b83d3a

data/README.md

@@ -30,27 +30,52 @@ create a client like this:
 client = OpenAI::Client.new("your OpenAI key here")
 ```
 
-### Completion
+### Completion 
+https://platform.openai.com/docs/api-reference/completions
 
 ```ruby
-res = client.create_completion(
-  model: "text-davinci-003", # The model which will generate the completion
-  prompt: "Hello, who are you?",
-  max_tokens: 100 # The maximum number of tokens to generate
-  temperature: 0.5, # Control randomness
-  top_p: 1,
-  frequency_penalty: 0,
-  presence_penalty: 0,
-  stop: "\n"
-)
-p res.status  # 200
+# params = {
+#   model: "text-davinci-003",
+#   prompt: "Hello, who are you?",
+#   max_tokens: 100,
+#   temperature: 0.5,
+#   frequency_penalty: 0,
+#   presence_penalty: 0,
+# }
+res = client.create_completion(params)
 response = JSON.parse(res.body)
 
 p response.dig("choices", 0, "text")  # "I am an AI created by OpenAI."
 p response.dig('usage', 'total_tokens')  # 18
 ```
 
+### Chat Completion 
+https://platform.openai.com/docs/api-reference/chat/create
+
+If you set `stream` param to `true`, a block will be called with the chunk data:
+
+```ruby
+# params = {
+#   model: "gpt-3.5-turbo",
+#   prompt: "Hello",
+#   max_tokens: 100,
+#   temperature: 1,
+#   stream: true,
+# }
+res = client.create_chat_completion(params) do |chunk, overall_received_bytes, env|
+  data = chunk[/data: (.*)\n\n$/, 1]
+  p data # {"id":"chatcmpl-6xcOWMQcilJUwJiosi7Rht6Fvuu3D","object":"chat.completion.chunk","created":1679666960,"model":"gpt-3.5-turbo-0301","choices":[{"delta":{"content":"Hello"},"index":0,"finish_reason":null}]}
+  if data == "[DONE]"
+    # the stream is end
+  else
+    response = JSON.parse(data)
+    p response.dig("choices", 0, "text")  # "Hello"
+  end
+end
+```
+
 ### Edit
+https://platform.openai.com/docs/api-reference/edits
 
 ```ruby
 res = client.create_edit(

data/lib/openai_ruby/client.rb

@@ -18,6 +18,23 @@ module OpenAI
       )
     end
 
+    def create_chat_completion(params = {}, &block)
+      if params[:stream]
+        connection.post("/v1/chat/completions") do |req|
+          req.body = params.to_json
+          req.options.on_data = Proc.new do |chunk, overall_received_bytes, env|
+            block.call(chunk, overall_received_bytes, env)
+          end
+        end
+      else
+        connection.post(
+          "/v1/chat/completions",
+          params.to_json,
+          headers
+        )
+      end
+    end
+
     def create_edit(params = {})
       Faraday.post(
         "#{BASE_URL}/v1/edits",
@@ -28,6 +45,10 @@ module OpenAI
 
     private
 
+    def connection
+      Faraday.new(url: BASE_URL, headers: headers)
+    end
+
     def headers
       {
         "Content-Type" => "application/json",

data/lib/openai_ruby/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module OpenAI
-  VERSION = "0.3.1"
+  VERSION = "0.3.2"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: openai_ruby
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.3.2
 platform: ruby
 authors:
 - Renny Ren
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-02-15 00:00:00.000000000 Z
+date: 2023-03-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -42,7 +42,6 @@ files:
 - lib/openai_ruby.rb
 - lib/openai_ruby/client.rb
 - lib/openai_ruby/version.rb
-- openai_ruby.gemspec
 - sig/openai_ruby/ruby.rbs
 homepage: https://github.com/renny-ren/openai_ruby
 licenses:

data/openai_ruby.gemspec

@@ -1,36 +0,0 @@
-# frozen_string_literal: true
-
-require_relative "lib/openai_ruby/version"
-
-Gem::Specification.new do |spec|
-  spec.name = "openai_ruby"
-  spec.version = OpenAI::VERSION
-  spec.authors = ["Renny Ren"]
-  spec.email = ["rennyrjh@gmail.com"]
-
-  spec.summary = "A Ruby wrapper for OpenAI API"
-  spec.homepage = "https://github.com/renny-ren/openai_ruby"
-  spec.license = "MIT"
-  spec.required_ruby_version = ">= 2.6.0"
-
-  spec.metadata["homepage_uri"] = spec.homepage
-  spec.metadata["source_code_uri"] = "https://github.com/renny-ren/openai_ruby"
-  spec.metadata["changelog_uri"] = "https://github.com/renny-ren/openai_ruby/blob/main/CHANGELOG.md"
-
-  # Specify which files should be added to the gem when it is released.
-  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
-  spec.files = Dir.chdir(__dir__) do
-    `git ls-files -z`.split("\x0").reject do |f|
-      (f == __FILE__) || f.match(%r{\A(?:(?:bin|test|spec|features)/|\.(?:git|travis|circleci)|appveyor)})
-    end
-  end
-
-  spec.bindir = "exe"
-  spec.executables = spec.files.grep(%r{\Aexe/}) { |f| File.basename(f) }
-  spec.require_paths = ["lib"]
-
-  spec.add_dependency "faraday", "~> 2.7"
-
-  # For more information and examples about making a new gem, check out our
-  # guide at: https://bundler.io/guides/creating_gem.html
-end