Added "custom_url" to ChatCompletionService

This commit is contained in:
bmen25124
2025-03-17 14:54:59 +03:00
parent fba2d809d0
commit 86de927ab9
2 changed files with 4 additions and 1 deletions

View File

@ -41,6 +41,7 @@ import { formatInstructModeChat, formatInstructModePrompt, names_behavior_types
* @property {string} chat_completion_source - Source provider for chat completion * @property {string} chat_completion_source - Source provider for chat completion
* @property {number} max_tokens - Maximum number of tokens to generate * @property {number} max_tokens - Maximum number of tokens to generate
* @property {number} [temperature] - Optional temperature parameter for response randomness * @property {number} [temperature] - Optional temperature parameter for response randomness
* @property {string} [custom_url] - Optional custom URL for chat completion
*/ */
/** @typedef {Record<string, any> & ChatCompletionPayloadBase} ChatCompletionPayload */ /** @typedef {Record<string, any> & ChatCompletionPayloadBase} ChatCompletionPayload */
@ -264,7 +265,7 @@ export class ChatCompletionService {
* @param {ChatCompletionPayload} custom * @param {ChatCompletionPayload} custom
* @returns {ChatCompletionPayload} * @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 = { const payload = {
...props, ...props,
messages, messages,
@ -272,6 +273,7 @@ export class ChatCompletionService {
chat_completion_source, chat_completion_source,
max_tokens, max_tokens,
temperature, temperature,
custom_url,
stream: false, stream: false,
}; };

View File

@ -323,6 +323,7 @@ export class ConnectionManagerRequestService {
max_tokens: maxTokens, max_tokens: maxTokens,
model: profile.model, model: profile.model,
chat_completion_source: selectedApiMap.source, chat_completion_source: selectedApiMap.source,
custom_url: profile['api-url'],
}, { }, {
presetName: includePreset ? profile.preset : undefined, presetName: includePreset ? profile.preset : undefined,
}, extractData); }, extractData);