30 lines
729 B
Lua
30 lines
729 B
Lua
|
local H = {}
|
||
|
|
local M = {}
|
||
|
|
|
||
|
|
H.get_os_name = function()
|
||
|
|
local os_name = vim.loop.os_uname().sysname
|
||
|
|
if os_name == "Linux" then
|
||
|
|
return "linux"
|
||
|
|
elseif os_name == "Darwin" then
|
||
|
|
return "macOS"
|
||
|
|
else
|
||
|
|
error("Unsupported operating system: " .. os_name)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
H.library_path = function()
|
||
|
|
local os_name = H.get_os_name()
|
||
|
|
local ext = os_name == "linux" and "so" or "dylib"
|
||
|
|
local dirname = string.sub(debug.getinfo(1).source, 2, #"/tiktoken_lib.lua" * -1)
|
||
|
|
return dirname .. ("../build/?.%s"):format(ext)
|
||
|
|
end
|
||
|
|
|
||
|
|
M.load = function()
|
||
|
|
local library_path = H.library_path()
|
||
|
|
if not string.find(package.cpath, library_path, 1, true) then
|
||
|
|
package.cpath = package.cpath .. ";" .. library_path
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
return M
|