283 lines
12 KiB
JavaScript
283 lines
12 KiB
JavaScript
// SageUtils Settings Integration
|
|
// Integrates SageUtils settings with ComfyUI's native settings system
|
|
|
|
import { app } from "../../../../scripts/app.js";
|
|
import { api } from "../../../../scripts/api.js";
|
|
|
|
// All of the setting types are based on PrimeVue components.
|
|
// https://primevue.org/
|
|
|
|
// Props described in the PrimeVue documentation can be defined
|
|
// for ComfyUI settings by adding them in an attrs field.
|
|
|
|
// Helper functions for API communication
|
|
async function loadSageSettings() {
|
|
try {
|
|
const response = await api.fetchApi('/sage_utils/settings');
|
|
if (response.ok) {
|
|
const data = await response.json();
|
|
if (data.success) {
|
|
// Handle both legacy format (data.settings) and new format (data.data.settings)
|
|
return data.settings || data.data?.settings;
|
|
} else {
|
|
console.error('Server returned error:', data.error);
|
|
if (data.details) {
|
|
console.error('Error details:', data.details);
|
|
}
|
|
}
|
|
} else {
|
|
console.error('HTTP error:', response.status, response.statusText);
|
|
}
|
|
throw new Error('Failed to load settings from server');
|
|
} catch (error) {
|
|
console.error('Error loading SageUtils settings:', error);
|
|
// Return fallback settings structure
|
|
return {
|
|
enable_lmstudio_rest: { current_value: false },
|
|
enable_ollama_rest: { current_value: false },
|
|
enable_openai: { current_value: false },
|
|
openai_api_key: { current_value: '' },
|
|
openai_use_custom_url: { current_value: false },
|
|
openai_base_url: { current_value: '' },
|
|
default_llm_provider: { current_value: 'lmstudio_rest' },
|
|
ollama_use_custom_url: { current_value: false },
|
|
ollama_custom_url: { current_value: "" },
|
|
ollama_api_key: { current_value: '' },
|
|
lmstudio_use_custom_url: { current_value: false },
|
|
lmstudio_custom_url: { current_value: "" },
|
|
lmstudio_api_token: { current_value: '' }
|
|
};
|
|
}
|
|
}
|
|
|
|
async function saveSageSetting(key, value) {
|
|
try {
|
|
const response = await api.fetchApi('/sage_utils/settings', {
|
|
method: 'POST',
|
|
headers: {
|
|
'Content-Type': 'application/json',
|
|
},
|
|
body: JSON.stringify({ [key]: value })
|
|
});
|
|
|
|
const data = await response.json();
|
|
if (data.success) {
|
|
console.log(`Successfully saved setting ${key}: ${value}`);
|
|
if (data.errors && data.errors.length > 0) {
|
|
console.warn('Warnings during setting save:', data.errors);
|
|
}
|
|
} else {
|
|
console.error(`Failed to save setting ${key}:`, data.error);
|
|
}
|
|
return data.success;
|
|
} catch (error) {
|
|
console.error(`Error saving SageUtils setting ${key}:`, error);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
// Register SageUtils settings with ComfyUI's native settings system
|
|
app.registerExtension({
|
|
name: "arcum42.sage.utils.settings",
|
|
|
|
async setup() {
|
|
console.log("Setting up SageUtils settings integration...");
|
|
|
|
// Load current settings from server
|
|
const serverSettings = await loadSageSettings();
|
|
if (!serverSettings) {
|
|
console.warn('Could not load SageUtils settings from server - settings will use defaults');
|
|
return;
|
|
}
|
|
|
|
console.log('Loaded SageUtils settings from server:', serverSettings);
|
|
|
|
// Set current values for all settings from server
|
|
// Map backend keys to the new frontend setting IDs
|
|
const keyToIdMap = {
|
|
'default_llm_provider': 'SageUtils.LLM Providers.default_llm_provider',
|
|
'enable_lmstudio_rest': 'SageUtils.LLM Providers.enable_lmstudio_rest',
|
|
'enable_ollama_rest': 'SageUtils.LLM Providers.enable_ollama_rest',
|
|
'enable_openai': 'SageUtils.LLM Providers.enable_openai',
|
|
'openai_api_key': 'SageUtils.OpenAI.openai_api_key',
|
|
'openai_use_custom_url': 'SageUtils.OpenAI.openai_use_custom_url',
|
|
'openai_base_url': 'SageUtils.OpenAI.openai_base_url',
|
|
'ollama_custom_url': 'SageUtils.Local Custom Ollama URL.ollama_custom_url',
|
|
'ollama_use_custom_url': 'SageUtils.Local Custom Ollama URL.ollama_use_custom_url',
|
|
'ollama_api_key': 'SageUtils.Ollama.ollama_api_key',
|
|
'lmstudio_custom_url': 'SageUtils.Local Custom LM Studio URL.lmstudio_custom_url',
|
|
'lmstudio_use_custom_url': 'SageUtils.Local Custom LM Studio URL.lmstudio_use_custom_url',
|
|
'lmstudio_api_token': 'SageUtils.LM Studio.lmstudio_api_token'
|
|
};
|
|
|
|
for (const [key, settingInfo] of Object.entries(serverSettings)) {
|
|
const settingId = keyToIdMap[key];
|
|
if (settingId && settingInfo.current_value !== undefined) {
|
|
try {
|
|
await app.extensionManager.setting.set(settingId, settingInfo.current_value);
|
|
console.log(`Set initial value for ${settingId}:`, settingInfo.current_value);
|
|
} catch (error) {
|
|
console.warn(`Could not set initial value for ${settingId}:`, error);
|
|
}
|
|
}
|
|
}
|
|
|
|
console.log(`SageUtils settings integration completed`);
|
|
},
|
|
|
|
settings: [
|
|
{
|
|
id: "SageUtils.LLM Providers.default_llm_provider",
|
|
name: "Default LLM Provider",
|
|
type: "combo",
|
|
defaultValue: "lmstudio_rest",
|
|
options: ["lmstudio_rest", "ollama_rest", "openai", "native"],
|
|
tooltip: "Default provider used by the LLM sidebar and provider-switching LLM v3 nodes",
|
|
onChange: async (newVal, oldVal) => {
|
|
console.log(`Default LLM provider changed from ${oldVal} to ${newVal}`);
|
|
await saveSageSetting('default_llm_provider', newVal);
|
|
}
|
|
},
|
|
{
|
|
id: "SageUtils.LLM Providers.enable_lmstudio_rest",
|
|
name: "Enable LM Studio Integration",
|
|
type: "boolean",
|
|
defaultValue: false,
|
|
tooltip: "Enable LM Studio REST v1 integration",
|
|
onChange: async (newVal, oldVal) => {
|
|
console.log(`LM Studio REST integration changed from ${oldVal} to ${newVal}`);
|
|
await saveSageSetting('enable_lmstudio_rest', newVal);
|
|
}
|
|
},
|
|
{
|
|
id: "SageUtils.LLM Providers.enable_ollama_rest",
|
|
name: "Enable Ollama Integration",
|
|
type: "boolean",
|
|
defaultValue: false,
|
|
tooltip: "Enable Ollama native REST API integration (no SDK required)",
|
|
onChange: async (newVal, oldVal) => {
|
|
console.log(`Ollama REST integration changed from ${oldVal} to ${newVal}`);
|
|
await saveSageSetting('enable_ollama_rest', newVal);
|
|
}
|
|
},
|
|
{
|
|
id: "SageUtils.LLM Providers.enable_openai",
|
|
name: "Enable OpenAI Integration",
|
|
type: "boolean",
|
|
defaultValue: false,
|
|
tooltip: "Enable OpenAI (or OpenAI-compatible) REST API integration",
|
|
onChange: async (newVal, oldVal) => {
|
|
console.log(`OpenAI integration changed from ${oldVal} to ${newVal}`);
|
|
await saveSageSetting('enable_openai', newVal);
|
|
}
|
|
},
|
|
{
|
|
id: "SageUtils.OpenAI.openai_api_key",
|
|
name: "API Key",
|
|
type: "text",
|
|
defaultValue: "",
|
|
tooltip: "OpenAI API key (or leave blank to use OPENAI_API_KEY env var)",
|
|
onChange: async (newVal, oldVal) => {
|
|
console.log('OpenAI API key changed');
|
|
await saveSageSetting('openai_api_key', newVal);
|
|
}
|
|
},
|
|
{
|
|
id: "SageUtils.OpenAI.openai_use_custom_url",
|
|
name: "Enable Custom URL",
|
|
type: "boolean",
|
|
defaultValue: false,
|
|
tooltip: "Use a custom base URL instead of https://api.openai.com",
|
|
onChange: async (newVal, oldVal) => {
|
|
console.log(`OpenAI custom URL setting changed from ${oldVal} to ${newVal}`);
|
|
await saveSageSetting('openai_use_custom_url', newVal);
|
|
}
|
|
},
|
|
{
|
|
id: "SageUtils.OpenAI.openai_base_url",
|
|
name: "Base URL",
|
|
type: "text",
|
|
defaultValue: "",
|
|
tooltip: "Custom base URL for OpenAI-compatible endpoint (e.g., 'http://localhost:8080')",
|
|
onChange: async (newVal, oldVal) => {
|
|
console.log(`OpenAI base URL changed from '${oldVal}' to '${newVal}'`);
|
|
await saveSageSetting('openai_base_url', newVal);
|
|
}
|
|
},
|
|
{
|
|
id: "SageUtils.Local Custom Ollama URL.ollama_custom_url",
|
|
name: "Address",
|
|
type: "text",
|
|
defaultValue: "",
|
|
tooltip: "Custom URL for Ollama service (e.g., 'http://localhost:11434')",
|
|
onChange: async (newVal, oldVal) => {
|
|
console.log(`Ollama custom URL changed from '${oldVal}' to '${newVal}'`);
|
|
await saveSageSetting('ollama_custom_url', newVal);
|
|
}
|
|
},
|
|
{
|
|
id: "SageUtils.Local Custom Ollama URL.ollama_use_custom_url",
|
|
name: "Enable Custom URL",
|
|
type: "boolean",
|
|
defaultValue: false,
|
|
tooltip: "Use a custom URL for Ollama instead of the default",
|
|
onChange: async (newVal, oldVal) => {
|
|
console.log(`Ollama custom URL setting changed from ${oldVal} to ${newVal}`);
|
|
await saveSageSetting('ollama_use_custom_url', newVal);
|
|
}
|
|
},
|
|
{
|
|
id: "SageUtils.Ollama.ollama_api_key",
|
|
name: "API Key",
|
|
type: "text",
|
|
defaultValue: "",
|
|
attrs: {
|
|
type: 'password',
|
|
autocomplete: 'off'
|
|
},
|
|
tooltip: "Ollama API key (or leave blank to use OLLAMA_API_KEY env var)",
|
|
onChange: async (newVal) => {
|
|
console.log('Ollama API key changed');
|
|
await saveSageSetting('ollama_api_key', newVal);
|
|
}
|
|
},
|
|
{
|
|
id: "SageUtils.Local Custom LM Studio URL.lmstudio_custom_url",
|
|
name: "Address",
|
|
type: "text",
|
|
defaultValue: "",
|
|
tooltip: "Custom URL for LM Studio service (e.g., 'http://localhost:1234')",
|
|
onChange: async (newVal, oldVal) => {
|
|
console.log(`LM Studio custom URL changed from '${oldVal}' to '${newVal}'`);
|
|
await saveSageSetting('lmstudio_custom_url', newVal);
|
|
}
|
|
},
|
|
{
|
|
id: "SageUtils.Local Custom LM Studio URL.lmstudio_use_custom_url",
|
|
name: "Enable Custom URL",
|
|
type: "boolean",
|
|
defaultValue: false,
|
|
tooltip: "Use a custom URL for LM Studio instead of the default",
|
|
onChange: async (newVal, oldVal) => {
|
|
console.log(`LM Studio custom URL setting changed from ${oldVal} to ${newVal}`);
|
|
await saveSageSetting('lmstudio_use_custom_url', newVal);
|
|
}
|
|
},
|
|
{
|
|
id: "SageUtils.LM Studio.lmstudio_api_token",
|
|
name: "API Token",
|
|
type: "text",
|
|
defaultValue: "",
|
|
attrs: {
|
|
type: 'password',
|
|
autocomplete: 'off'
|
|
},
|
|
tooltip: "LM Studio API token (or leave blank to use LMSTUDIO_API_TOKEN env var)",
|
|
onChange: async (newVal) => {
|
|
console.log('LM Studio API token changed');
|
|
await saveSageSetting('lmstudio_api_token', newVal);
|
|
}
|
|
}
|
|
]
|
|
});
|