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

Check for API key in multimodal captions

Browse Source
This commit is contained in:
Cohee
2023-11-18 20:58:04 +02:00
parent 7045d242e8
commit 4999fbd97c
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
public/scripts/extensions/shared.js
View File

@ -1,5 +1,6 @@
import { getRequestHeaders } from "../../script.js"; import { getRequestHeaders } from "../../script.js";
import { extension_settings } from "../extensions.js"; import { extension_settings } from "../extensions.js";
import { SECRET_KEYS, secret_state } from "../secrets.js";
/** /**
* Generates a caption for an image using a multimodal model. * Generates a caption for an image using a multimodal model.
@ -8,6 +9,14 @@ import { extension_settings } from "../extensions.js";
* @returns {Promise<string>} Generated caption * @returns {Promise<string>} Generated caption
*/ */
export async function getMultimodalCaption(base64Img, prompt) { export async function getMultimodalCaption(base64Img, prompt) {
if (extension_settings.caption.multimodal_api === 'openai' && !secret_state[SECRET_KEYS.OPENAI]) {
throw new Error('OpenAI API key is not set.');
}
if (extension_settings.caption.multimodal_api === 'openrouter' && !secret_state[SECRET_KEYS.OPENROUTER]) {
throw new Error('OpenRouter API key is not set.');
}
const apiResult = await fetch('/api/openai/caption-image', { const apiResult = await fetch('/api/openai/caption-image', {
method: 'POST', method: 'POST',
headers: getRequestHeaders(), headers: getRequestHeaders(),