mirror of
https://github.com/Mintplex-Labs/anything-llm.git
synced 2024-09-12 07:10:08 +02:00
* renamed all indicies to vectors to avoid confusion of vocab * removing unneeded files * changed indicies on frontend sidebar to vectors
154 lines
4.2 KiB
JavaScript
154 lines
4.2 KiB
JavaScript
const { SystemSettings } = require("../../../models/systemSettings");
|
|
const { getVectorDbClass } = require("../../../utils/helpers");
|
|
const { dumpENV, updateENV } = require("../../../utils/helpers/updateENV");
|
|
const { reqBody } = require("../../../utils/http");
|
|
const { validApiKey } = require("../../../utils/middleware/validApiKey");
|
|
|
|
function apiSystemEndpoints(app) {
|
|
if (!app) return;
|
|
|
|
app.get("/v1/system/env-dump", async (_, response) => {
|
|
/*
|
|
#swagger.tags = ['System Settings']
|
|
#swagger.description = 'Dump all settings to file storage'
|
|
#swagger.responses[403] = {
|
|
schema: {
|
|
"$ref": "#/definitions/InvalidAPIKey"
|
|
}
|
|
}
|
|
*/
|
|
try {
|
|
if (process.env.NODE_ENV !== "production")
|
|
return response.sendStatus(200).end();
|
|
await dumpENV();
|
|
response.sendStatus(200).end();
|
|
} catch (e) {
|
|
console.log(e.message, e);
|
|
response.sendStatus(500).end();
|
|
}
|
|
});
|
|
|
|
app.get("/v1/system", [validApiKey], async (_, response) => {
|
|
/*
|
|
#swagger.tags = ['System Settings']
|
|
#swagger.description = 'Get all current system settings that are defined.'
|
|
#swagger.responses[200] = {
|
|
content: {
|
|
"application/json": {
|
|
schema: {
|
|
type: 'object',
|
|
example: {
|
|
"settings": {
|
|
"VectorDB": "pinecone",
|
|
"PineConeEnvironment": "us-west4-gcp-free",
|
|
"PineConeKey": true,
|
|
"PineConeIndex": "my-pinecone-index",
|
|
"LLMProvider": "azure",
|
|
"[KEY_NAME]": "KEY_VALUE",
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
#swagger.responses[403] = {
|
|
schema: {
|
|
"$ref": "#/definitions/InvalidAPIKey"
|
|
}
|
|
}
|
|
*/
|
|
try {
|
|
const settings = await SystemSettings.currentSettings();
|
|
response.status(200).json({ settings });
|
|
} catch (e) {
|
|
console.log(e.message, e);
|
|
response.sendStatus(500).end();
|
|
}
|
|
});
|
|
|
|
app.get("/v1/system/vector-count", [validApiKey], async (_, response) => {
|
|
/*
|
|
#swagger.tags = ['System Settings']
|
|
#swagger.description = 'Number of all vectors in connected vector database'
|
|
#swagger.responses[200] = {
|
|
content: {
|
|
"application/json": {
|
|
schema: {
|
|
type: 'object',
|
|
example: {
|
|
"vectorCount": 5450
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
#swagger.responses[403] = {
|
|
schema: {
|
|
"$ref": "#/definitions/InvalidAPIKey"
|
|
}
|
|
}
|
|
*/
|
|
try {
|
|
const VectorDb = getVectorDbClass();
|
|
const vectorCount = await VectorDb.totalVectors();
|
|
response.status(200).json({ vectorCount });
|
|
} catch (e) {
|
|
console.log(e.message, e);
|
|
response.sendStatus(500).end();
|
|
}
|
|
});
|
|
|
|
app.post(
|
|
"/v1/system/update-env",
|
|
[validApiKey],
|
|
async (request, response) => {
|
|
/*
|
|
#swagger.tags = ['System Settings']
|
|
#swagger.description = 'Update a system setting or preference.'
|
|
#swagger.requestBody = {
|
|
description: 'Key pair object that matches a valid setting and value. Get keys from GET /v1/system or refer to codebase.',
|
|
required: true,
|
|
type: 'object',
|
|
content: {
|
|
"application/json": {
|
|
example: {
|
|
VectorDB: "lancedb",
|
|
AnotherKey: "updatedValue"
|
|
}
|
|
}
|
|
}
|
|
}
|
|
#swagger.responses[200] = {
|
|
content: {
|
|
"application/json": {
|
|
schema: {
|
|
type: 'object',
|
|
example: {
|
|
newValues: {"[ENV_KEY]": 'Value'},
|
|
error: 'error goes here, otherwise null'
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
#swagger.responses[403] = {
|
|
schema: {
|
|
"$ref": "#/definitions/InvalidAPIKey"
|
|
}
|
|
}
|
|
*/
|
|
try {
|
|
const body = reqBody(request);
|
|
const { newValues, error } = updateENV(body);
|
|
if (process.env.NODE_ENV === "production") await dumpENV();
|
|
response.status(200).json({ newValues, error });
|
|
} catch (e) {
|
|
console.log(e.message, e);
|
|
response.sendStatus(500).end();
|
|
}
|
|
}
|
|
);
|
|
}
|
|
|
|
module.exports = { apiSystemEndpoints };
|