rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-06-05 21:59:27 +02:00
Code Issues Packages Projects Releases Wiki Activity

Exported getChatCompletionModel with optional parameter

Browse Source
This commit is contained in:
bmen25124
2025-02-25 15:45:35 +03:00
parent 20a982491a
commit 7064ce81c7
2 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
public/scripts/openai.js
View File

@ -1497,8 +1497,14 @@ async function sendWindowAIRequest(messages, signal, stream) {
} }
} }
export function getChatCompletionModel() { /**
switch (oai_settings.chat_completion_source) { * Gets the API model for the selected chat completion source.
* @param {string} source If it's set, ignores active source
* @returns {string} API model
*/
export function getChatCompletionModel(source = null) {
const activeSource = source ?? oai_settings.chat_completion_source;
switch (activeSource) {
case chat_completion_sources.CLAUDE: case chat_completion_sources.CLAUDE:
return oai_settings.claude_model; return oai_settings.claude_model;
case chat_completion_sources.OPENAI: case chat_completion_sources.OPENAI:
@ -1532,7 +1538,7 @@ export function getChatCompletionModel() {
case chat_completion_sources.DEEPSEEK: case chat_completion_sources.DEEPSEEK:
return oai_settings.deepseek_model; return oai_settings.deepseek_model;
default: default:
throw new Error(`Unknown chat completion source: ${oai_settings.chat_completion_source}`); throw new Error(`Unknown chat completion source: ${activeSource}`);
} }
} }

3
public/scripts/st-context.js
View File

@ -57,7 +57,7 @@ import { groups, openGroupChat, selected_group } from './group-chats.js';
import { t, translate } from './i18n.js'; import { t, translate } from './i18n.js';
import { hideLoader, showLoader } from './loader.js'; import { hideLoader, showLoader } from './loader.js';
import { MacrosParser } from './macros.js'; import { MacrosParser } from './macros.js';
import { oai_settings } from './openai.js'; import { getChatCompletionModel, oai_settings } from './openai.js';
import { callGenericPopup, Popup, POPUP_RESULT, POPUP_TYPE } from './popup.js'; import { callGenericPopup, Popup, POPUP_RESULT, POPUP_TYPE } from './popup.js';
import { power_user, registerDebugFunction } from './power-user.js'; import { power_user, registerDebugFunction } from './power-user.js';
import { getPresetManager } from './preset-manager.js'; import { getPresetManager } from './preset-manager.js';
@ -200,6 +200,7 @@ export function getContext() {
getTextGenServer, getTextGenServer,
extractMessageFromData, extractMessageFromData,
getPresetManager, getPresetManager,
getChatCompletionModel,
}; };
} }