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

Move endpoints into their own folder

Browse Source
This commit is contained in:
valadaptive
2023-12-03 08:52:43 -05:00
parent 604e62298c
commit 4919a2e9a2
20 changed files with 48 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/assets.js → src/endpoints/assets.js
View File

@@ -4,7 +4,7 @@ const sanitize = require('sanitize-filename');
const fetch = require('node-fetch').default;
const { finished } = require('stream/promises');
const writeFileSyncAtomic = require('write-file-atomic').sync;
const { DIRECTORIES, UNSAFE_EXTENSIONS } = require('./constants');
const { DIRECTORIES, UNSAFE_EXTENSIONS } = require('../constants');
const VALID_CATEGORIES = ['bgm', 'ambient', 'blip', 'live2d'];

2
src/caption.js → src/endpoints/caption.js
View File

@@ -9,7 +9,7 @@ function registerEndpoints(app, jsonParser) {
try {
const { image } = req.body;
const module = await import('./transformers.mjs');
const module = await import('../transformers.mjs');
const rawImage = await module.default.getRawImage(image);
if (!rawImage) {

4
src/classify.js → src/endpoints/classify.js
View File

@@ -9,7 +9,7 @@ function registerEndpoints(app, jsonParser) {
app.post('/api/extra/classify/labels', jsonParser, async (req, res) => {
try {
const module = await import('./transformers.mjs');
const module = await import('../transformers.mjs');
const pipe = await module.default.getPipeline(TASK);
const result = Object.keys(pipe.model.config.label2id);
return res.json({ labels: result });
@@ -27,7 +27,7 @@ function registerEndpoints(app, jsonParser) {
if (Object.hasOwn(cacheObject, text)) {
return cacheObject[text];
} else {
const module = await import('./transformers.mjs');
const module = await import('../transformers.mjs');
const pipe = await module.default.getPipeline(TASK);
const result = await pipe(text, { topk: 5 });
result.sort((a, b) => b.score - a.score);

4
src/content-manager.js → src/endpoints/content-manager.js
View File

@@ -2,11 +2,11 @@ const fs = require('fs');
const path = require('path');
const fetch = require('node-fetch').default;
const sanitize = require('sanitize-filename');
const { getConfigValue } = require('./util');
const { getConfigValue } = require('../util');
const contentDirectory = path.join(process.cwd(), 'default/content');
const contentLogPath = path.join(contentDirectory, 'content.log');
const contentIndexPath = path.join(contentDirectory, 'index.json');
const { DIRECTORIES } = require('./constants');
const { DIRECTORIES } = require('../constants');
const presetFolders = [DIRECTORIES.koboldAI_Settings, DIRECTORIES.openAI_Settings, DIRECTORIES.novelAI_Settings, DIRECTORIES.textGen_Settings];
/**

2
src/extensions.js → src/endpoints/extensions.js
View File

@@ -2,7 +2,7 @@ const path = require('path');
const fs = require('fs');
const { default: simpleGit } = require('simple-git');
const sanitize = require('sanitize-filename');
const { DIRECTORIES } = require('./constants');
const { DIRECTORIES } = require('../constants');
/**
* This function extracts the extension information from the manifest file.

4
src/horde.js → src/endpoints/horde.js
View File

@@ -1,6 +1,6 @@
const fetch = require('node-fetch').default;
const AIHorde = require('./ai_horde');
const { getVersion, delay } = require('./util');
const AIHorde = require('../ai_horde');
const { getVersion, delay } = require('../util');
const { readSecret, SECRET_KEYS } = require('./secrets');
const ANONYMOUS_KEY = '0000000000';

2
src/novelai.js → src/endpoints/novelai.js
View File

@@ -2,7 +2,7 @@ const fetch = require('node-fetch').default;
const util = require('util');
const { Readable } = require('stream');
const { readSecret, SECRET_KEYS } = require('./secrets');
const { readAllChunks, extractFileFromZipBuffer } = require('./util');
const { readAllChunks, extractFileFromZipBuffer } = require('../util');
const API_NOVELAI = 'https://api.novelai.net';

0
src/openai.js → src/endpoints/openai.js
View File

2
src/presets.js → src/endpoints/presets.js
View File

@@ -2,7 +2,7 @@ const fs = require('fs');
const path = require('path');
const sanitize = require('sanitize-filename');
const writeFileAtomicSync = require('write-file-atomic').sync;
const { DIRECTORIES } = require('./constants');
const { DIRECTORIES } = require('../constants');
const { getDefaultPresetFile, getDefaultPresets } = require('./content-manager');
/**

2
src/secrets.js → src/endpoints/secrets.js
View File

@@ -1,6 +1,6 @@
const fs = require('fs');
const path = require('path');
const { getConfigValue } = require('./util');
const { getConfigValue } = require('../util');
const writeFileAtomicSync = require('write-file-atomic').sync;
const SECRETS_FILE = path.join(process.cwd(), './secrets.json');

0
src/serpapi.js → src/endpoints/serpapi.js
View File

4
src/sprites.js → src/endpoints/sprites.js
View File

@@ -4,8 +4,8 @@ const path = require('path');
const mime = require('mime-types');
const sanitize = require('sanitize-filename');
const writeFileAtomicSync = require('write-file-atomic').sync;
const { DIRECTORIES, UPLOADS_PATH } = require('./constants');
const { getImageBuffers } = require('./util');
const { DIRECTORIES, UPLOADS_PATH } = require('../constants');
const { getImageBuffers } = require('../util');
/**
* Gets the path to the sprites folder for the provided character name

6
src/stable-diffusion.js → src/endpoints/stable-diffusion.js
View File

@@ -1,8 +1,8 @@
const fetch = require('node-fetch').default;
const sanitize = require('sanitize-filename');
const { getBasicAuthHeader, delay } = require('./util');
const { getBasicAuthHeader, delay } = require('../util.js');
const fs = require('fs');
const { DIRECTORIES } = require('./constants.js');
const { DIRECTORIES } = require('../constants.js');
const writeFileAtomicSync = require('write-file-atomic').sync;
/**
@@ -341,7 +341,7 @@ function registerEndpoints(app, jsonParser) {
try {
const task = 'text-generation';
const module = await import('./transformers.mjs');
const module = await import('../transformers.mjs');
const pipe = await module.default.getPipeline(task);
const result = await pipe(prompt, { num_beams: 1, max_new_tokens: 256, do_sample: true });

4
src/thumbnails.js → src/endpoints/thumbnails.js
View File

@@ -3,8 +3,8 @@ const path = require('path');
const sanitize = require('sanitize-filename');
const jimp = require('jimp');
const writeFileAtomicSync = require('write-file-atomic').sync;
const { DIRECTORIES } = require('./constants');
const { getConfigValue } = require('./util');
const { DIRECTORIES } = require('../constants');
const { getConfigValue } = require('../util');
/**
* Gets a path to thumbnail folder based on the type.

2
src/tokenizers.js → src/endpoints/tokenizers.js
View File

@@ -3,7 +3,7 @@ const path = require('path');
const { SentencePieceProcessor } = require('@agnai/sentencepiece-js');
const tiktoken = require('@dqbd/tiktoken');
const { Tokenizer } = require('@agnai/web-tokenizers');
const { convertClaudePrompt } = require('./chat-completion');
const { convertClaudePrompt } = require('../chat-completion');
const { readSecret, SECRET_KEYS } = require('./secrets');
/**

2
src/translate.js → src/endpoints/translate.js
View File

@@ -1,7 +1,7 @@
const fetch = require('node-fetch').default;
const https = require('https');
const { readSecret, SECRET_KEYS } = require('./secrets');
const { getConfigValue } = require('./util');
const { getConfigValue } = require('../util');
const DEEPLX_URL_DEFAULT = 'http://127.0.0.1:1188/translate';
const ONERING_URL_DEFAULT = 'http://127.0.0.1:4990/translate';

6
src/vectors.js → src/endpoints/vectors.js
View File

@@ -11,11 +11,11 @@ const sanitize = require('sanitize-filename');
async function getVector(source, text) {
switch (source) {
case 'openai':
return require('./openai-vectors').getOpenAIVector(text);
return require('../openai-vectors').getOpenAIVector(text);
case 'transformers':
return require('./embedding').getTransformersVector(text);
return require('../embedding').getTransformersVector(text);
case 'palm':
return require('./palm-vectors').getPaLMVector(text);
return require('../palm-vectors').getPaLMVector(text);
}
throw new Error(`Unknown vector source ${source}`);

2
src/openai-vectors.js
View File

@@ -1,5 +1,5 @@
const fetch = require('node-fetch').default;
const { SECRET_KEYS, readSecret } = require('./secrets');
const { SECRET_KEYS, readSecret } = require('./endpoints/secrets');
/**
* Gets the vector for the given text from OpenAI ada model

2
src/palm-vectors.js
View File

@@ -1,5 +1,5 @@
const fetch = require('node-fetch').default;
const { SECRET_KEYS, readSecret } = require('./secrets');
const { SECRET_KEYS, readSecret } = require('./endpoints/secrets');
/**
* Gets the vector for the given text from PaLM gecko model