openai.rb 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f049bfa7ff5419acf81a7cabc6e34b6bd223b8f9fcb634008544d93d87170c2f
-  data.tar.gz: d614701b82eaf3ec84e9ddb432505bbbd786f71d75378534b23400446ddb8e7b
+  metadata.gz: 96f82f97fa18d7fc75b7efd8e01fe015b50c0d4e33c06894b37b2f807315138b
+  data.tar.gz: b563a01a6b737282477a7e47c29f1d514b0bcecd806f0b3d9b0ad3f9394fb605
 SHA512:
-  metadata.gz: c321d0184079826648569ab13b04675d9cbfb04027fed85b191d462c6585cc375d9144dede8ecf6d9995e48be935bf6879dba969e17492cb9c4ffe100f468fea
-  data.tar.gz: 3fd0d70f0007e3900e6df1f39ef18dcbd618b7e3384de229b50936924bb4bf80e79d19bd7a531be6caade2b915c03581cccabc32f01b5921c79bd88e224c27a0
+  metadata.gz: a51cc63b3023d1bd0ca33e3ab6325ea8bd9d8a64fbff301b2e044b113e656c0245fdcb891a2633fb0d9edd18a59073e02ff4ca14fd5d32a571f71e43948402c8
+  data.tar.gz: 8eea70ae9c26df48a0b0651354bf6aa4cf7996e348bc6a6e5714aaa239a6815e1831a063c9c3efe912f283aeacaf794908189709f4ebd34209192ab42942af55

data/CHANGELOG.md

@@ -0,0 +1,3 @@
+# v0.0.4
+
+* Added support for specifying `OpenAI.new(api_key, organization: 'org-123')`

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    openai.rb (0.0.3)
+    openai.rb (0.0.4)
       abstract_type (~> 0.0.7)
       anima (~> 0.3)
       concord (~> 0.1)

data/bin/console

@@ -17,5 +17,3 @@ openai = OpenAI.create(
 )
 
 Pry.start_without_pry_byebug(binding, quiet: true)
-
-start_repl

data/lib/openai/api/client.rb

@@ -3,18 +3,18 @@
 class OpenAI
   class API
     class Client
-      include Concord.new(:api_key, :http)
+      include Concord.new(:api_key, :organization_id, :http)
 
       public :api_key
 
       HOST = Addressable::URI.parse('https://api.openai.com/v1')
 
-      def initialize(api_key, http: HTTP)
-        super(api_key, http)
+      def initialize(api_key, organization_id: nil, http: HTTP)
+        super(api_key, organization_id, http)
       end
 
       def inspect
-        "#<#{self.class}>"
+        "#<#{self.class} organization_id=#{organization_id.inspect}>"
       end
 
       def get(route)
@@ -79,7 +79,9 @@ class OpenAI
       end
 
       def http_client
-        http.headers('Authorization' => "Bearer #{api_key}")
+        headers = { 'Authorization' => "Bearer #{api_key}" }
+        headers['OpenAI-Organization'] = organization_id if organization_id
+        http.headers(headers)
       end
     end
   end

data/lib/openai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 class OpenAI
-  VERSION = '0.0.3'
+  VERSION = '0.0.4'
 end

data/lib/openai.rb

@@ -28,8 +28,8 @@ class OpenAI
 
   ROOT = Pathname.new(__dir__).parent.expand_path.freeze
 
-  def self.create(api_key, cache: nil, logger: Logger.new('/dev/null'))
-    client = API::Client.new(api_key)
+  def self.create(api_key, cache: nil, organization: nil, logger: Logger.new('/dev/null'))
+    client = API::Client.new(api_key, organization_id: organization)
 
     if cache.is_a?(Pathname) && cache.directory?
       client = API::Cache.new(

data/spec/unit/openai_spec.rb

@@ -37,6 +37,25 @@ RSpec.describe OpenAI do
     )
   end
 
+  context 'when the organization ID is given to the client' do
+    let(:api_client) do
+      OpenAI::API::Client.new(
+        'sk-123',
+        organization_id: 'org-123',
+        http: http
+      )
+    end
+
+    it 'authenticates the request and includes the organization id' do
+      resource.create(model: 'text-davinci-002', prompt: 'Hello, world!')
+
+      expect(http).to have_received(:headers).with(
+        'OpenAI-Organization' => 'org-123',
+        'Authorization' => 'Bearer sk-123'
+      )
+    end
+  end
+
   context 'when the request is not 2xx' do
     let(:response_body) do
       {

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: openai.rb
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - John
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2023-04-02 00:00:00.000000000 Z
+date: 2023-04-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: abstract_type
@@ -127,6 +127,7 @@ files:
 - ".rspec"
 - ".rubocop.yml"
 - ".ruby-version"
+- CHANGELOG.md
 - Gemfile
 - Gemfile.lock
 - README.md