mirror of
https://github.com/danny-avila/LibreChat.git
synced 2026-02-19 08:58:09 +01:00
* chore: move database model methods to /packages/data-schemas * chore: add TypeScript ESLint rule to warn on unused variables * refactor: model imports to streamline access - Consolidated model imports across various files to improve code organization and reduce redundancy. - Updated imports for models such as Assistant, Message, Conversation, and others to a unified import path. - Adjusted middleware and service files to reflect the new import structure, ensuring functionality remains intact. - Enhanced test files to align with the new import paths, maintaining test coverage and integrity. * chore: migrate database models to packages/data-schemas and refactor all direct Mongoose Model usage outside of data-schemas * test: update agent model mocks in unit tests - Added `getAgent` mock to `client.test.js` to enhance test coverage for agent-related functionality. - Removed redundant `getAgent` and `getAgents` mocks from `openai.spec.js` and `responses.unit.spec.js` to streamline test setup and reduce duplication. - Ensured consistency in agent mock implementations across test files. * fix: update types in data-schemas * refactor: enhance type definitions in transaction and spending methods - Updated type definitions in `checkBalance.ts` to use specific request and response types. - Refined `spendTokens.ts` to utilize a new `SpendTxData` interface for better clarity and type safety. - Improved transaction handling in `transaction.ts` by introducing `TransactionResult` and `TxData` interfaces, ensuring consistent data structures across methods. - Adjusted unit tests in `transaction.spec.ts` to accommodate new type definitions and enhance robustness. * refactor: streamline model imports and enhance code organization - Consolidated model imports across various controllers and services to a unified import path, improving code clarity and reducing redundancy. - Updated multiple files to reflect the new import structure, ensuring all functionalities remain intact. - Enhanced overall code organization by removing duplicate import statements and optimizing the usage of model methods. * feat: implement loadAddedAgent and refactor agent loading logic - Introduced `loadAddedAgent` function to handle loading agents from added conversations, supporting multi-convo parallel execution. - Created a new `load.ts` file to encapsulate agent loading functionalities, including `loadEphemeralAgent` and `loadAgent`. - Updated the `index.ts` file to export the new `load` module instead of the deprecated `loadAgent`. - Enhanced type definitions and improved error handling in the agent loading process. - Adjusted unit tests to reflect changes in the agent loading structure and ensure comprehensive coverage. * refactor: enhance balance handling with new update interface - Introduced `IBalanceUpdate` interface to streamline balance update operations across the codebase. - Updated `upsertBalanceFields` method signatures in `balance.ts`, `transaction.ts`, and related tests to utilize the new interface for improved type safety. - Adjusted type imports in `balance.spec.ts` to include `IBalanceUpdate`, ensuring consistency in balance management functionalities. - Enhanced overall code clarity and maintainability by refining type definitions related to balance operations. * feat: add unit tests for loadAgent functionality and enhance agent loading logic - Introduced comprehensive unit tests for the `loadAgent` function, covering various scenarios including null and empty agent IDs, loading of ephemeral agents, and permission checks. - Enhanced the `initializeClient` function by moving `getConvoFiles` to the correct position in the database method exports, ensuring proper functionality. - Improved test coverage for agent loading, including handling of non-existent agents and user permissions. * chore: reorder memory method exports for consistency - Moved `deleteAllUserMemories` to the correct position in the exported memory methods, ensuring a consistent and logical order of method exports in `memory.ts`.
176 lines
4.8 KiB
JavaScript
176 lines
4.8 KiB
JavaScript
const express = require('express');
|
|
const { generateCheckAccess } = require('@librechat/api');
|
|
const { PermissionTypes, Permissions, PermissionBits } = require('librechat-data-provider');
|
|
const { requireJwtAuth, configMiddleware, canAccessAgentResource } = require('~/server/middleware');
|
|
const v1 = require('~/server/controllers/agents/v1');
|
|
const { getRoleByName } = require('~/models');
|
|
const actions = require('./actions');
|
|
const tools = require('./tools');
|
|
|
|
const router = express.Router();
|
|
const avatar = express.Router();
|
|
|
|
const checkAgentAccess = generateCheckAccess({
|
|
permissionType: PermissionTypes.AGENTS,
|
|
permissions: [Permissions.USE],
|
|
getRoleByName,
|
|
});
|
|
const checkAgentCreate = generateCheckAccess({
|
|
permissionType: PermissionTypes.AGENTS,
|
|
permissions: [Permissions.USE, Permissions.CREATE],
|
|
getRoleByName,
|
|
});
|
|
|
|
router.use(requireJwtAuth);
|
|
|
|
/**
|
|
* Agent actions route.
|
|
* @route GET|POST /agents/actions
|
|
*/
|
|
router.use('/actions', configMiddleware, actions);
|
|
|
|
/**
|
|
* Get a list of available tools for agents.
|
|
* @route GET /agents/tools
|
|
*/
|
|
router.use('/tools', configMiddleware, tools);
|
|
|
|
/**
|
|
* Get all agent categories with counts
|
|
* @route GET /agents/categories
|
|
*/
|
|
router.get('/categories', v1.getAgentCategories);
|
|
/**
|
|
* Creates an agent.
|
|
* @route POST /agents
|
|
* @param {AgentCreateParams} req.body - The agent creation parameters.
|
|
* @returns {Agent} 201 - Success response - application/json
|
|
*/
|
|
router.post('/', checkAgentCreate, v1.createAgent);
|
|
|
|
/**
|
|
* Retrieves basic agent information (VIEW permission required).
|
|
* Returns safe, non-sensitive agent data for viewing purposes.
|
|
* @route GET /agents/:id
|
|
* @param {string} req.params.id - Agent identifier.
|
|
* @returns {Agent} 200 - Basic agent info - application/json
|
|
*/
|
|
router.get(
|
|
'/:id',
|
|
checkAgentAccess,
|
|
canAccessAgentResource({
|
|
requiredPermission: PermissionBits.VIEW,
|
|
resourceIdParam: 'id',
|
|
}),
|
|
v1.getAgent,
|
|
);
|
|
|
|
/**
|
|
* Retrieves full agent details including sensitive configuration (EDIT permission required).
|
|
* Returns complete agent data for editing/configuration purposes.
|
|
* @route GET /agents/:id/expanded
|
|
* @param {string} req.params.id - Agent identifier.
|
|
* @returns {Agent} 200 - Full agent details - application/json
|
|
*/
|
|
router.get(
|
|
'/:id/expanded',
|
|
checkAgentAccess,
|
|
canAccessAgentResource({
|
|
requiredPermission: PermissionBits.EDIT,
|
|
resourceIdParam: 'id',
|
|
}),
|
|
(req, res) => v1.getAgent(req, res, true), // Expanded version
|
|
);
|
|
/**
|
|
* Updates an agent.
|
|
* @route PATCH /agents/:id
|
|
* @param {string} req.params.id - Agent identifier.
|
|
* @param {AgentUpdateParams} req.body - The agent update parameters.
|
|
* @returns {Agent} 200 - Success response - application/json
|
|
*/
|
|
router.patch(
|
|
'/:id',
|
|
checkAgentCreate,
|
|
canAccessAgentResource({
|
|
requiredPermission: PermissionBits.EDIT,
|
|
resourceIdParam: 'id',
|
|
}),
|
|
v1.updateAgent,
|
|
);
|
|
|
|
/**
|
|
* Duplicates an agent.
|
|
* @route POST /agents/:id/duplicate
|
|
* @param {string} req.params.id - Agent identifier.
|
|
* @returns {Agent} 201 - Success response - application/json
|
|
*/
|
|
router.post(
|
|
'/:id/duplicate',
|
|
checkAgentCreate,
|
|
canAccessAgentResource({
|
|
requiredPermission: PermissionBits.VIEW,
|
|
resourceIdParam: 'id',
|
|
}),
|
|
v1.duplicateAgent,
|
|
);
|
|
|
|
/**
|
|
* Deletes an agent.
|
|
* @route DELETE /agents/:id
|
|
* @param {string} req.params.id - Agent identifier.
|
|
* @returns {Agent} 200 - success response - application/json
|
|
*/
|
|
router.delete(
|
|
'/:id',
|
|
checkAgentCreate,
|
|
canAccessAgentResource({
|
|
requiredPermission: PermissionBits.DELETE,
|
|
resourceIdParam: 'id',
|
|
}),
|
|
v1.deleteAgent,
|
|
);
|
|
|
|
/**
|
|
* Reverts an agent to a previous version.
|
|
* @route POST /agents/:id/revert
|
|
* @param {string} req.params.id - Agent identifier.
|
|
* @param {number} req.body.version_index - Index of the version to revert to.
|
|
* @returns {Agent} 200 - success response - application/json
|
|
*/
|
|
router.post(
|
|
'/:id/revert',
|
|
checkAgentCreate,
|
|
canAccessAgentResource({
|
|
requiredPermission: PermissionBits.EDIT,
|
|
resourceIdParam: 'id',
|
|
}),
|
|
v1.revertAgentVersion,
|
|
);
|
|
|
|
/**
|
|
* Returns a list of agents.
|
|
* @route GET /agents
|
|
* @param {AgentListParams} req.query - The agent list parameters for pagination and sorting.
|
|
* @returns {AgentListResponse} 200 - success response - application/json
|
|
*/
|
|
router.get('/', checkAgentAccess, v1.getListAgents);
|
|
|
|
/**
|
|
* Uploads and updates an avatar for a specific agent.
|
|
* @route POST /agents/:agent_id/avatar
|
|
* @param {string} req.params.agent_id - The ID of the agent.
|
|
* @param {Express.Multer.File} req.file - The avatar image file.
|
|
* @param {string} [req.body.metadata] - Optional metadata for the agent's avatar.
|
|
* @returns {Object} 200 - success response - application/json
|
|
*/
|
|
avatar.post(
|
|
'/:agent_id/avatar/',
|
|
checkAgentAccess,
|
|
canAccessAgentResource({
|
|
requiredPermission: PermissionBits.EDIT,
|
|
resourceIdParam: 'agent_id',
|
|
}),
|
|
v1.uploadAgentAvatar,
|
|
);
|
|
|
|
module.exports = { v1: router, avatar };
|