refactor: use Constants value for GLOBAL_PROJECT_NAME (enum)

This commit is contained in:
Danny Avila 2024-09-03 18:00:39 -04:00
parent 7d303b3581
commit 72dfcb9dc9
No known key found for this signature in database
GPG key ID: 2DD9CC89B9B50364
5 changed files with 16 additions and 6 deletions

View file

@ -1,5 +1,5 @@
const { ObjectId } = require('mongodb');
const { SystemRoles, SystemCategories } = require('librechat-data-provider');
const { SystemRoles, SystemCategories, Constants } = require('librechat-data-provider');
const {
getProjectByName,
addGroupIdsToProject,
@ -9,6 +9,8 @@ const {
const { Prompt, PromptGroup } = require('./schema/promptSchema');
const { logger } = require('~/config');
const { GLOBAL_PROJECT_NAME } = Constants;
/**
* Create a pipeline for the aggregation to get prompt groups
* @param {Object} query
@ -123,7 +125,7 @@ const getAllPromptGroups = async (req, filter) => {
let combinedQuery = query;
if (searchShared) {
const project = await getProjectByName('instance', 'promptGroupIds');
const project = await getProjectByName(GLOBAL_PROJECT_NAME, 'promptGroupIds');
if (project && project.promptGroupIds.length > 0) {
const projectQuery = { _id: { $in: project.promptGroupIds }, ...query };
delete projectQuery.author;
@ -177,7 +179,7 @@ const getPromptGroups = async (req, filter) => {
if (searchShared) {
// const projects = req.user.projects || []; // TODO: handle multiple projects
const project = await getProjectByName('instance', 'promptGroupIds');
const project = await getProjectByName(GLOBAL_PROJECT_NAME, 'promptGroupIds');
if (project && project.promptGroupIds.length > 0) {
const projectQuery = { _id: { $in: project.promptGroupIds }, ...query };
delete projectQuery.author;