rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-03-13 18:40:11 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #3558 from bmen25124/new_exports

Browse Source 
New exports - getChatCompletionModel
This commit is contained in:
Cohee 2025-02-25 15:05:30 +02:00 committed by GitHub
commit f0c7ea062b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 11 additions and 4 deletions
Show Stats 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:
return oai_settings.claude_model;
case chat_completion_sources.OPENAI:
@ -1532,7 +1538,7 @@ export function getChatCompletionModel() {
case chat_completion_sources.DEEPSEEK:
return oai_settings.deepseek_model;
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 { hideLoader, showLoader } from './loader.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 { power_user, registerDebugFunction } from './power-user.js';
import { getPresetManager } from './preset-manager.js';
@ -200,6 +200,7 @@ export function getContext() {
getTextGenServer,
extractMessageFromData,
getPresetManager,
getChatCompletionModel,
};
}