diff --git a/public/scripts/custom-request.js b/public/scripts/custom-request.js index 60e8b8f7f..d6fccf1a9 100644 --- a/public/scripts/custom-request.js +++ b/public/scripts/custom-request.js @@ -41,6 +41,7 @@ import { formatInstructModeChat, formatInstructModePrompt, names_behavior_types * @property {string} chat_completion_source - Source provider for chat completion * @property {number} max_tokens - Maximum number of tokens to generate * @property {number} [temperature] - Optional temperature parameter for response randomness + * @property {string} [custom_url] - Optional custom URL for chat completion */ /** @typedef {Record & ChatCompletionPayloadBase} ChatCompletionPayload */ @@ -264,7 +265,7 @@ export class ChatCompletionService { * @param {ChatCompletionPayload} custom * @returns {ChatCompletionPayload} */ - static createRequestData({ messages, model, chat_completion_source, max_tokens, temperature, ...props }) { + static createRequestData({ messages, model, chat_completion_source, max_tokens, temperature, custom_url, ...props }) { const payload = { ...props, messages, @@ -272,6 +273,7 @@ export class ChatCompletionService { chat_completion_source, max_tokens, temperature, + custom_url, stream: false, }; diff --git a/public/scripts/extensions/shared.js b/public/scripts/extensions/shared.js index 3e7eef8a4..d5549c40a 100644 --- a/public/scripts/extensions/shared.js +++ b/public/scripts/extensions/shared.js @@ -323,6 +323,7 @@ export class ConnectionManagerRequestService { max_tokens: maxTokens, model: profile.model, chat_completion_source: selectedApiMap.source, + custom_url: profile['api-url'], }, { presetName: includePreset ? profile.preset : undefined, }, extractData);