Merge pull request alexrudall#15 from alexrudall/add_search

Add search

Author: alexrudall

CHANGELOG.md

@@ -5,6 +5,12 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.2.0] - 2020-11-22
+
+### Added
+
+- Add method to use the search endpoint.
+
 ## [0.1.4] - 2020-10-18
 
 ### Changed

Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    ruby-openai (0.1.4)
+    ruby-openai (0.2.0)
       dotenv (~> 2.7.6)
       httparty (~> 0.18.1)
 

lib/ruby/openai/client.rb

@@ -7,7 +7,7 @@ def initialize(access_token: nil)
       @access_token = access_token || ENV["OPENAI_ACCESS_TOKEN"]
     end
 
-    def call(engine:, prompt:, max_tokens:, version: "v1")
+    def call(engine:, prompt:, max_tokens:, version: default_version)
       self.class.post(
         "/#{version}/engines/#{engine}/completions",
         headers: {
@@ -19,5 +19,24 @@ def call(engine:, prompt:, max_tokens:, version: "v1")
         }.to_json
       )
     end
+
+    def search(engine:, documents:, query:, version: default_version)
+      self.class.post(
+        "/#{version}/engines/#{engine}/search",
+        headers: {
+          "Content-Type" => "application/json",
+          "Authorization" => "Bearer #{@access_token}"
+        },
+        body: {
+          documents: documents, query: query
+        }.to_json
+      )
+    end
+
+    private
+
+    def default_version
+      "v1".freeze
+    end
   end
 end

lib/ruby/openai/version.rb

@@ -1,5 +1,5 @@
 module Ruby
   module OpenAI
-    VERSION = "0.1.4".freeze
+    VERSION = "0.2.0".freeze
   end
 end