feat: add url_join (#856)

This commit is contained in:
yetone
2024-11-17 00:39:03 +08:00
committed by GitHub
parent 8bbcd64550
commit dfc51b3247
8 changed files with 43 additions and 18 deletions

View File

@@ -25,11 +25,10 @@ M.parse_curl_args = function(provider, code_opts)
if not P.env.is_local("azure") then headers["api-key"] = provider.parse_api_key() end
return {
url = Utils.trim(base.endpoint, { suffix = "/" })
.. "/openai/deployments/"
.. base.deployment
.. "/chat/completions?api-version="
.. base.api_version,
url = Utils.url_join(
base.endpoint,
"/openai/deployments/" .. base.deployment .. "/chat/completions?api-version=" .. base.api_version
),
proxy = base.proxy,
insecure = base.allow_insecure,
headers = headers,

View File

@@ -106,7 +106,7 @@ M.parse_curl_args = function(provider, prompt_opts)
local messages = M.parse_messages(prompt_opts)
return {
url = Utils.trim(base.endpoint, { suffix = "/" }) .. "/v1/messages",
url = Utils.url_join(base.endpoint, "/v1/messages"),
proxy = base.proxy,
insecure = base.allow_insecure,
headers = headers,

View File

@@ -85,7 +85,7 @@ M.parse_curl_args = function(provider, code_opts)
if not P.env.is_local("cohere") then headers["Authorization"] = "Bearer " .. provider.parse_api_key() end
return {
url = Utils.trim(base.endpoint, { suffix = "/" }) .. "/chat",
url = Utils.url_join(base.endpoint, "/chat"),
proxy = base.proxy,
insecure = base.allow_insecure,
headers = headers,

View File

@@ -86,11 +86,10 @@ M.parse_curl_args = function(provider, code_opts)
body_opts.max_tokens = nil
return {
url = Utils.trim(base.endpoint, { suffix = "/" })
.. "/"
.. base.model
.. ":streamGenerateContent?alt=sse&key="
.. provider.parse_api_key(),
url = Utils.url_join(
base.endpoint,
base.model .. ":streamGenerateContent?alt=sse&key=" .. provider.parse_api_key()
),
proxy = base.proxy,
insecure = base.allow_insecure,
headers = { ["Content-Type"] = "application/json" },

View File

@@ -136,7 +136,7 @@ M.parse_curl_args = function(provider, code_opts)
end
return {
url = Utils.trim(base.endpoint, { suffix = "/" }) .. "/chat/completions",
url = Utils.url_join(base.endpoint, "/chat/completions"),
proxy = base.proxy,
insecure = base.allow_insecure,
headers = headers,

View File

@@ -438,6 +438,32 @@ function M.trim_spaces(s) return s:match("^%s*(.-)%s*$") end
function M.fallback(v, default_value) return type(v) == "nil" and default_value or v end
---Join URL parts together, handling slashes correctly
---@param ... string URL parts to join
---@return string Joined URL
function M.url_join(...)
local parts = { ... }
local result = parts[1] or ""
for i = 2, #parts do
local part = parts[i]
if not part or part == "" then goto continue end
-- Remove trailing slash from result if present
if result:sub(-1) == "/" then result = result:sub(1, -2) end
-- Remove leading slash from part if present
if part:sub(1, 1) == "/" then part = part:sub(2) end
-- Join with slash
result = result .. "/" .. part
::continue::
end
return result
end
-- luacheck: push no max comment line length
---@param type_name "'nil'" | "'number'" | "'string'" | "'boolean'" | "'table'" | "'function'" | "'thread'" | "'userdata'" | "'list'" | '"map"'
---@return boolean