Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2025-12-20 10:20:15 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

fix: agents query

Browse source
This commit is contained in:
Danny Avila 2024-09-03 18:49:15 -04:00
parent 3f694f2e25
commit dd90907a18
No known key found for this signature in database
GPG key ID: 2DD9CC89B9B50364
2 changed files with 9 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

13
api/models/Agent.js
View file

@ -71,14 +71,13 @@ const deleteAgent = async (searchParameter) => {
const getListAgents = async (searchParameter) => {
const { author, ...otherParams } = searchParameter;
let query = {
$or: [{ author }, { projectIds: { $exists: true, $ne: [], $not: { $size: 0 } } }],
...otherParams,
};
let query = Object.assign({ author }, otherParams);
const globalProject = await getProjectByName(GLOBAL_PROJECT_NAME, 'agentIds');
if (globalProject && globalProject.agentIds.length > 0) {
query.$or.push({ _id: { $in: globalProject.agentIds } });
const globalProject = await getProjectByName(GLOBAL_PROJECT_NAME, ['agentIds']);
if (globalProject && (globalProject.agentIds?.length ?? 0) > 0) {
const globalQuery = { id: { $in: globalProject.agentIds }, ...otherParams };
delete globalQuery.author;
query = { $or: [globalQuery, query] };
}
const agents = await Agent.find(query, {

6
api/server/controllers/agents/v1.js
View file

@ -131,9 +131,9 @@ const deleteAgentHandler = async (req, res) => {
*/
const getListAgentsHandler = async (req, res) => {
try {
const { user } = req.query;
const filter = user ? { author: user } : {};
const data = await getListAgents(filter);
const data = await getListAgents({
author: req.user.id,
});
return res.json(data);
} catch (error) {
logger.error('[/Agents] Error listing Agents', error);