add embedding method

Author: leafo

examples/example4.lua

@@ -0,0 +1,8 @@
+local openai = require("openai")
+local client = openai.new(os.getenv("OPENAI_API_KEY"))
+
+local status, response = client:embedding("Lua is the best programming language ever")
+
+if status == 200 then
+  print(table.concat(response.data[1].embedding, ", "))
+end

openai/init.lua

@@ -220,6 +220,19 @@ do
       end
       return self:_request("POST", "/completions", payload)
     end,
+    embedding = function(self, input, opts)
+      assert(input, "input must be provided")
+      local payload = {
+        model = "text-embedding-ada-002",
+        input = input
+      }
+      if opts then
+        for k, v in pairs(opts) do
+          payload[k] = v
+        end
+      end
+      return self:_request("POST", "/embeddings", payload)
+    end,
     _request = function(self, method, path, payload, more_headers, stream_fn)
       assert(path, "missing path")
       assert(method, "missing method")

openai/init.moon

@@ -234,6 +234,24 @@ class OpenAI
 
     @_request "POST", "/completions", payload
 
+
+  -- Call /embeddings to generate an embedding for the given text
+  -- input: A string or array table of strings to generate embeddings for
+  -- opts: additional parameters as described in https://platform.openai.com/docs/api-reference/embeddings
+  embedding: (input, opts) =>
+    assert input, "input must be provided"
+
+    payload = {
+      model: "text-embedding-ada-002"
+      :input
+    }
+
+    if opts
+      for k,v in pairs opts
+        payload[k] = v
+
+    @_request "POST", "/embeddings", payload
+
   _request: (method, path, payload, more_headers, stream_fn) =>
     assert path, "missing path"
     assert method, "missing method"