rixty/SillyTavern
1
0
Fork 0
mirror of https://github.com/SillyTavern/SillyTavern.git synced 2025-02-08 08:08:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

Try detect LM Studio in status check

Browse Source 
Fixes #2849
This commit is contained in:
Cohee 2024-09-15 08:24:28 +03:00
parent 2478e17164
commit 2bc87a401e
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/endpoints/backends/text-completions.js
View File

@ -137,6 +137,7 @@ router.post('/status', jsonParser, async function (request, response) {
} }
const modelsReply = await fetch(url, args); const modelsReply = await fetch(url, args);
const isPossiblyLmStudio = modelsReply.headers.get('x-powered-by') === 'Express';
if (!modelsReply.ok) { if (!modelsReply.ok) {
console.log('Models endpoint is offline.'); console.log('Models endpoint is offline.');
@ -174,7 +175,7 @@ router.post('/status', jsonParser, async function (request, response) {
// Set result to the first model ID // Set result to the first model ID
result = modelIds[0] || 'Valid'; result = modelIds[0] || 'Valid';
if (apiType === TEXTGEN_TYPES.OOBA) { if (apiType === TEXTGEN_TYPES.OOBA && !isPossiblyLmStudio) {
try { try {
const modelInfoUrl = baseUrl + '/v1/internal/model/info'; const modelInfoUrl = baseUrl + '/v1/internal/model/info';
const modelInfoReply = await fetch(modelInfoUrl, args); const modelInfoReply = await fetch(modelInfoUrl, args);