feat: add url_join (#856)
This commit is contained in:
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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" },
|
||||
|
||||
@@ -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,
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user