openai-assistant 0.4.0 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 47c4f3c25796b691c23177c0305accd4d4eb10d13f3d5dd8d5828d6070a7f801
-  data.tar.gz: 22cdf48b820eed695c042ecb9f5892ff8cc05c78547a6d98c95de0a7591c1435
+  metadata.gz: 30d997127785423573776d064b5a0096f121df245af1c56511be53563b342505
+  data.tar.gz: 166a91a99f0caf0f5b08c9de791531c9f16c9493d2ffa1395db11237c982f620
 SHA512:
-  metadata.gz: be4e756a8888af5c0b1f0247a6a17bd1e02cbdc636f43f28b84c2a7c91dd1644d981357c8ead3d81b6cb5831f35b220a6e881d8fbefb5aaa36f0349130995ff9
-  data.tar.gz: 25566590081132fb1526f6970a472b2ca348881931e8052270aad14557fa53fb6e3020f56705866ed488dac04947d69bba7c32ea3b99f5eb3c3f355c0fc1f65a
+  metadata.gz: c5f306812242afaac48bd3da55e0280a36877238052ebdd5e75942d68bdf68cdbfc9a81d1c0ebeb2b9d8abd4b10644bdfaef37b1b22ba8580a5a00222a597693
+  data.tar.gz: 8aa5092ad31e0372595baf420020f8d953a500dce327cab67c9344221c7aceca230d4150ef11637304f8558eebcd7d32506ceb533a7de1f7e7647dd5c53dac96

data/lib/http/http.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+require "net/http"
+require "json"
+
+module Http
+  # An http client
+  class Client
+    def call_post(_url, req_body, headers)
+      http = Net::HTTP.new(_url.host, _url.port)
+      http.use_ssl = true
+      request = Net::HTTP::Post.new(_url.path, headers)
+      request.body = req_body unless req_body.nil?
+      http.request(request)
+    end
+
+    def call_delete(_url, headers)
+      http = Net::HTTP.new(_url.host, _url.port)
+      http.use_ssl = true
+      request = Net::HTTP::Delete.new(_url.path, headers)
+      http.request(request)
+    end
+
+    def call_get(_url, headers)
+      http = Net::HTTP.new(_url.host, _url.port)
+      http.use_ssl = true
+      request = Net::HTTP::Get.new(_url.path, headers)
+      http.request(request)
+    end
+  end
+end

data/lib/openai/assistant/version.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module Openai
-  class Assistant
-    VERSION = "0.4.0"
+  class Assistant < Openai::Base
+    VERSION = "0.6.0"
   end
 end

data/lib/openai/assistant.rb

@@ -2,94 +2,65 @@
 
 require_relative "assistant/version"
 require_relative "assistant_obj"
+require_relative "base"
+require_relative "../http/http"
 require "json"
 require "net/http"
-require "rest-client"
+require "uri"
+
 module Openai
   # An openai assistant
-  class Assistant
-    @openai_api_key = nil
-    @openai_url = nil
-
-    # @param api_key [String] The api key of openai
+  class Assistant < Base
+    # @param api_key [String] The api key of openai\
     def initialize(api_key = "")
-      @openai_api_key = api_key
-      # hard the host because if the official docs change the host, maybe it will change another
-      # we need to update this gem for any change
-      @openai_url = "https://api.openai.com/v1/assistants"
+      super(api_key)
     end
 
     # @param api_key [String] The api key of openai
     def self.setup(api_key = "")
-      @openai_api_key = api_key
-      @openai_url = "https://api.openai.com/v1/assistants"
+      initialize(api_key)
     end
 
     # @param model [String] Select model of the assistant. refer on: https://platform.openai.com/docs/api-reference/models/list.
     # @param instructions [String] The system instructions that the assistant uses.
     # @return [Openai::AssistantObj] A new response object of assistant.
     def create_assistant(model, instructions)
-      url = @openai_url
-      headers = {
-        "Authorization": "Bearer #{@openai_api_key}",
-        "OpenAI-Beta": "assistants=v1",
-        "Content-Type": "application/json"
-      }
+      url = URI.parse(@openai_url)
       req_body = {
         "instructions": instructions,
         "name": "assistant",
         "tools": [{ "type": "code_interpreter" }],
         "model": model
-      }
-      begin
-        resp = RestClient.post(url, req_body.to_json, headers)
-      rescue RestClient::ExceptionWithResponse => e
-        resp = e.response
-      end
-      unless resp.code == 200
-        parsed = JSON.parse(resp.body)
+      }.to_json
+      response = @http_client.call_post(url, req_body, default_headers)
+      unless response.code == "200"
+        parsed = JSON.parse(response.body)
         return parsed["error"]["code"]
       end
-      parse_assistant_object(JSON.parse(resp.body))
+      parse_assistant_object(JSON.parse(response.body))
     end
 
     # @param assistant_id [String] The id of assistant after create
     # @return [Openai::AssistantObj] A new response object of assistant.
     def retrieve_assistant(assistant_id)
       url = "#{@openai_url}/#{assistant_id}"
-      headers = {
-        "Authorization": "Bearer #{@openai_api_key}",
-        "OpenAI-Beta": "assistants=v1",
-        "Content-Type": "application/json"
-      }
-      begin
-        resp = RestClient.get(url, headers)
-      rescue RestClient::ExceptionWithResponse => e
-        resp = e.response
-      end
-      unless resp.code == 200
-        parsed = JSON.parse(resp.body)
+      uri = URI(url)
+      response = @http_client.call_get(uri, default_headers)
+      unless response.code == "200"
+        parsed = JSON.parse(response.body)
         return parsed["error"]["code"]
       end
-      parse_assistant_object(JSON.parse(resp.body))
+      parse_assistant_object(JSON.parse(response.body))
     end
 
     # @param assistant_id [String] The id of assistant after create
     # @return [String] Message delete the assistant ok or not
     def delete_assistant(assistant_id)
       url = "#{@openai_url}/#{assistant_id}"
-      headers = {
-        "Authorization": "Bearer #{@openai_api_key}",
-        "OpenAI-Beta": "assistants=v1",
-        "Content-Type": "application/json"
-      }
-      begin
-        resp = RestClient.delete(url, headers)
-      rescue RestClient::ExceptionWithResponse => e
-        resp = e.response
-      end
-      parsed = JSON.parse(resp.body)
-      return parsed["error"]["code"] unless resp.code == 200
+      uri = URI(url)
+      response = @http_client.call_delete(uri, default_headers)
+      parsed = JSON.parse(response.body)
+      return parsed["error"]["code"] unless response.code == "200"
 
       parsed["deleted"]
     end
@@ -97,18 +68,10 @@ module Openai
     # @return [Array<Openai::AssistantObj>] List all assistant
     def list_assistant
       url = @openai_url
-      headers = {
-        "Authorization": "Bearer #{@openai_api_key}",
-        "OpenAI-Beta": "assistants=v1",
-        "Content-Type": "application/json"
-      }
-      begin
-        resp = RestClient.get(url, headers)
-      rescue RestClient::ExceptionWithResponse => e
-        resp = e.response
-      end
-      parsed = JSON.parse(resp.body)
-      return parsed["error"]["code"] unless resp.code == 200
+      uri = URI(url)
+      response = @http_client.call_get(uri, default_headers)
+      parsed = JSON.parse(response.body)
+      return parsed["error"]["code"] unless response.code == "200"
 
       assistants = []
       parsed["data"].each do |ast|
@@ -116,7 +79,8 @@ module Openai
       end
     end
 
-    # @return private
+    private
+
     def parse_assistant_object(data)
       Openai::AssistantObj.new(
         id: data["id"],
@@ -131,5 +95,13 @@ module Openai
         metadata: data["metadata"]
       )
     end
+
+    def default_headers
+      {
+        "Authorization": "Bearer #{@openai_api_key}",
+        "OpenAI-Beta": "assistants=v1",
+        "Content-Type": "application/json"
+      }
+    end
   end
 end

data/lib/openai/base.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+require_relative "../http/http"
+
+module Openai
+  # Base class of openai
+  class Base
+    @openai_api_key = nil
+    @openai_url = nil
+    @http_client = nil
+
+    def initialize(api_key = "")
+      @openai_api_key = api_key
+      # hard the host because if the official docs change the host, maybe it will change another
+      # we need to update this gem for any change
+      @openai_url = "https://api.openai.com/v1/assistants"
+      @http_client = Http::Client.new
+    end
+  end
+end

data/openai-assistant.gemspec

@@ -1,5 +1,6 @@
 # frozen_string_literal: true
 
+require_relative "lib/openai/base"
 require_relative "lib/openai/assistant/version"
 
 Gem::Specification.new do |spec|

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: openai-assistant
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.6.0
 platform: ruby
 authors:
 - duonghds
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-11-29 00:00:00.000000000 Z
+date: 2023-11-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
@@ -38,9 +38,11 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- lib/http/http.rb
 - lib/openai/assistant.rb
 - lib/openai/assistant/version.rb
 - lib/openai/assistant_obj.rb
+- lib/openai/base.rb
 - openai-assistant.gemspec
 - sig/openai/assistant.rbs
 homepage: https://rubygems.org/