Skip to content

Commit 08128b9

Browse files
committed
rebuild, add image generation endpoint
1 parent 2e1c55e commit 08128b9

File tree

2 files changed

+26
-0
lines changed

2 files changed

+26
-0
lines changed

openai/init.lua

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -323,6 +323,29 @@ do
323323
delete_file = function(self, file_id)
324324
return self:_request("DELETE", "/files/" .. tostring(file_id))
325325
end,
326+
assistants = function(self)
327+
return self:_request("GET", "/assistants", nil, {
328+
["OpenAI-Beta"] = "assistants=v1"
329+
})
330+
end,
331+
threads = function(self)
332+
return self:_request("GET", "/threads", nil, {
333+
["OpenAI-Beta"] = "assistants=v1"
334+
})
335+
end,
336+
thread_messages = function(self, thread_id)
337+
return self:_request("GET", "/threads/" .. tostring(thread_id) .. "/messages", {
338+
["OpenAI-Beta"] = "assistants=v1"
339+
})
340+
end,
341+
delete_thread = function(self, thread_id)
342+
return self:_request("DELETE", "/threads/" .. tostring(thread_id), nil, {
343+
["OpenAI-Beta"] = "assistants=v1"
344+
})
345+
end,
346+
image_generation = function(self, params)
347+
return self:_request("POST", "/images/generations", params)
348+
end,
326349
_request = function(self, method, path, payload, more_headers, stream_fn)
327350
assert(path, "missing path")
328351
assert(method, "missing method")

openai/init.moon

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -359,6 +359,9 @@ class OpenAI
359359
"OpenAI-Beta": "assistants=v1"
360360
}
361361

362+
image_generation: (params) =>
363+
@_request "POST", "/images/generations", params
364+
362365
_request: (method, path, payload, more_headers, stream_fn) =>
363366
assert path, "missing path"
364367
assert method, "missing method"

0 commit comments

Comments
 (0)