Import promptManager from openai.js

7dfba69fc1804f3650d3285f85288fd06292b0a3

Cohee <18619528+Cohee1207@users.noreply.github.com>

2 files changed, +6 -9Showing whitespace changes
public/scripts/openai.js+2 -2
@@ -379,8 +379,8 @@ export let selected_proxy = proxies[0];
379let openai_setting_names;379let openai_setting_names;
380let openai_settings;380let openai_settings;
381381
382382/** @type {import('./PromptManager.js').PromptManager} */
383let promptManager = null;383export let promptManager = null;
384384
385async function validateReverseProxy() {385async function validateReverseProxy() {
386 if (!oai_settings.reverse_proxy) {386 if (!oai_settings.reverse_proxy) {
public/scripts/slash-commands.js+4 -7
@@ -52,7 +52,7 @@ import { hideChatMessageRange } from './chats.js';
52import { getContext, saveMetadataDebounced } from './extensions.js';52import { getContext, saveMetadataDebounced } from './extensions.js';
53import { getRegexedString, regex_placement } from './extensions/regex/engine.js';53import { getRegexedString, regex_placement } from './extensions/regex/engine.js';
54import { findGroupMemberId, groups, is_group_generating, openGroupById, resetSelectedGroup, saveGroupChat, selected_group } from './group-chats.js';54import { findGroupMemberId, groups, is_group_generating, openGroupById, resetSelectedGroup, saveGroupChat, selected_group } from './group-chats.js';
55import { chat_completion_sources, oai_settings, setupChatCompletionPromptManager } from './openai.js';55import { chat_completion_sources, oai_settings, promptManager } from './openai.js';
56import { autoSelectPersona, retriggerFirstMessageOnEmptyChat, setPersonaLockState, togglePersonaLock, user_avatar } from './personas.js';56import { autoSelectPersona, retriggerFirstMessageOnEmptyChat, setPersonaLockState, togglePersonaLock, user_avatar } from './personas.js';
57import { addEphemeralStoppingString, chat_styles, flushEphemeralStoppingStrings, power_user } from './power-user.js';57import { addEphemeralStoppingString, chat_styles, flushEphemeralStoppingStrings, power_user } from './power-user.js';
58import { SERVER_INPUTS, textgen_types, textgenerationwebui_settings } from './textgen-settings.js';58import { SERVER_INPUTS, textgen_types, textgenerationwebui_settings } from './textgen-settings.js';
@@ -1712,7 +1712,7 @@ export function initDefaultSlashCommands() {
1712 enumProvider: () =>1712 enumProvider: () =>
1713 promptManager.serviceSettings.prompts1713 promptManager.serviceSettings.prompts
1714 .map(prompt => prompt.identifier)1714 .map(prompt => prompt.identifier)
1715 .map(identifier => new SlashCommandEnumValue(identifier, identifier)),1715 .map(identifier => new SlashCommandEnumValue(identifier)),
1716 }),1716 }),
1717 SlashCommandNamedArgument.fromProps({1717 SlashCommandNamedArgument.fromProps({
1718 name: 'name',1718 name: 'name',
@@ -1722,7 +1722,7 @@ export function initDefaultSlashCommands() {
1722 enumProvider: () =>1722 enumProvider: () =>
1723 promptManager.serviceSettings.prompts1723 promptManager.serviceSettings.prompts
1724 .map(prompt => prompt.name)1724 .map(prompt => prompt.name)
1725 .map(name => new SlashCommandEnumValue(name, name)),1725 .map(name => new SlashCommandEnumValue(name)),
1726 }),1726 }),
1727 SlashCommandNamedArgument.fromProps({1727 SlashCommandNamedArgument.fromProps({
1728 name: 'return',1728 name: 'return',
@@ -1752,7 +1752,6 @@ export function initDefaultSlashCommands() {
1752 typeList: [ARGUMENT_TYPE.STRING, ARGUMENT_TYPE.LIST],1752 typeList: [ARGUMENT_TYPE.STRING, ARGUMENT_TYPE.LIST],
1753 acceptsMultiple: true,1753 acceptsMultiple: true,
1754 enumProvider: () => {1754 enumProvider: () => {
1755 const promptManager = setupChatCompletionPromptManager(oai_settings);
1756 const prompts = promptManager.serviceSettings.prompts;1755 const prompts = promptManager.serviceSettings.prompts;
1757 return prompts.map(prompt => new SlashCommandEnumValue(prompt.identifier, prompt.name, enumTypes.enum));1756 return prompts.map(prompt => new SlashCommandEnumValue(prompt.identifier, prompt.name, enumTypes.enum));
1758 },1757 },
@@ -1763,7 +1762,6 @@ export function initDefaultSlashCommands() {
1763 typeList: [ARGUMENT_TYPE.STRING, ARGUMENT_TYPE.LIST],1762 typeList: [ARGUMENT_TYPE.STRING, ARGUMENT_TYPE.LIST],
1764 acceptsMultiple: true,1763 acceptsMultiple: true,
1765 enumProvider: () => {1764 enumProvider: () => {
1766 const promptManager = setupChatCompletionPromptManager(oai_settings);
1767 const prompts = promptManager.serviceSettings.prompts;1765 const prompts = promptManager.serviceSettings.prompts;
1768 return prompts.map(prompt => new SlashCommandEnumValue(prompt.name, prompt.identifier, enumTypes.enum));1766 return prompts.map(prompt => new SlashCommandEnumValue(prompt.name, prompt.identifier, enumTypes.enum));
1769 },1767 },
@@ -3834,10 +3832,10 @@ function modelCallback(args, model) {
3834 * @param {object} args Object containing arguments3832 * @param {object} args Object containing arguments
3835 * @param {string} args.identifier Select prompt entry using an identifier (uuid)3833 * @param {string} args.identifier Select prompt entry using an identifier (uuid)
3836 * @param {string} args.name Select prompt entry using name3834 * @param {string} args.name Select prompt entry using name
3835 * @param {string} args.return The type of return value to use (simple, list, dict)
3837 * @returns {Object} An object containing the states of the requested prompt entries3836 * @returns {Object} An object containing the states of the requested prompt entries
3838 */3837 */
3839function getPromptEntryCallback(args) {3838function getPromptEntryCallback(args) {
3840 const promptManager = setupChatCompletionPromptManager(oai_settings);
3841 const prompts = promptManager.serviceSettings.prompts;3839 const prompts = promptManager.serviceSettings.prompts;
3842 let returnType = args.return ?? 'simple';3840 let returnType = args.return ?? 'simple';
38433841
@@ -3908,7 +3906,6 @@ function getPromptEntryCallback(args) {
3908 */3906 */
3909function setPromptEntryCallback(args, targetState) {3907function setPromptEntryCallback(args, targetState) {
3910 // needs promptManager to manipulate prompt entries3908 // needs promptManager to manipulate prompt entries
3911 const promptManager = setupChatCompletionPromptManager(oai_settings);
3912 const prompts = promptManager.serviceSettings.prompts;3909 const prompts = promptManager.serviceSettings.prompts;
39133910
3914 function parseArgs(arg) {3911 function parseArgs(arg) {