mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-09-22 08:12:00 +02:00
a2fc7d312a
* refactor: move model definitions and database-related methods to packages/data-schemas * ci: update tests due to new DB structure fix: disable mocking `librechat-data-provider` feat: Add schema exports to data-schemas package - Introduced a new schema module that exports various schemas including action, agent, and user schemas. - Updated index.ts to include the new schema exports for better modularity and organization. ci: fix appleStrategy tests fix: Agent.spec.js ci: refactor handleTools tests to use MongoMemoryServer for in-memory database fix: getLogStores imports ci: update banViolation tests to use MongoMemoryServer and improve session mocking test: refactor samlStrategy tests to improve mock configurations and user handling ci: fix crypto mock in handleText tests for improved accuracy ci: refactor spendTokens tests to improve model imports and setup ci: refactor Message model tests to use MongoMemoryServer and improve database interactions * refactor: streamline IMessage interface and move feedback properties to types/message.ts * refactor: use exported initializeRoles from `data-schemas`, remove api workspace version (this serves as an example of future migrations that still need to happen) * refactor: update model imports to use destructuring from `~/db/models` for consistency and clarity * refactor: remove unused mongoose imports from model files for cleaner code * refactor: remove unused mongoose imports from Share, Prompt, and Transaction model files for cleaner code * refactor: remove unused import in Transaction model for cleaner code * ci: update deploy workflow to reference new Docker Dev Branch Images Build and add new workflow for building Docker images on dev branch * chore: cleanup imports
96 lines
2.8 KiB
JavaScript
96 lines
2.8 KiB
JavaScript
const { ToolCall } = require('~/db/models');
|
|
|
|
/**
|
|
* Create a new tool call
|
|
* @param {IToolCallData} toolCallData - The tool call data
|
|
* @returns {Promise<IToolCallData>} The created tool call document
|
|
*/
|
|
async function createToolCall(toolCallData) {
|
|
try {
|
|
return await ToolCall.create(toolCallData);
|
|
} catch (error) {
|
|
throw new Error(`Error creating tool call: ${error.message}`);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get a tool call by ID
|
|
* @param {string} id - The tool call document ID
|
|
* @returns {Promise<IToolCallData|null>} The tool call document or null if not found
|
|
*/
|
|
async function getToolCallById(id) {
|
|
try {
|
|
return await ToolCall.findById(id).lean();
|
|
} catch (error) {
|
|
throw new Error(`Error fetching tool call: ${error.message}`);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get tool calls by message ID and user
|
|
* @param {string} messageId - The message ID
|
|
* @param {string} userId - The user's ObjectId
|
|
* @returns {Promise<Array>} Array of tool call documents
|
|
*/
|
|
async function getToolCallsByMessage(messageId, userId) {
|
|
try {
|
|
return await ToolCall.find({ messageId, user: userId }).lean();
|
|
} catch (error) {
|
|
throw new Error(`Error fetching tool calls: ${error.message}`);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get tool calls by conversation ID and user
|
|
* @param {string} conversationId - The conversation ID
|
|
* @param {string} userId - The user's ObjectId
|
|
* @returns {Promise<IToolCallData[]>} Array of tool call documents
|
|
*/
|
|
async function getToolCallsByConvo(conversationId, userId) {
|
|
try {
|
|
return await ToolCall.find({ conversationId, user: userId }).lean();
|
|
} catch (error) {
|
|
throw new Error(`Error fetching tool calls: ${error.message}`);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Update a tool call
|
|
* @param {string} id - The tool call document ID
|
|
* @param {Partial<IToolCallData>} updateData - The data to update
|
|
* @returns {Promise<IToolCallData|null>} The updated tool call document or null if not found
|
|
*/
|
|
async function updateToolCall(id, updateData) {
|
|
try {
|
|
return await ToolCall.findByIdAndUpdate(id, updateData, { new: true }).lean();
|
|
} catch (error) {
|
|
throw new Error(`Error updating tool call: ${error.message}`);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Delete a tool call
|
|
* @param {string} userId - The related user's ObjectId
|
|
* @param {string} [conversationId] - The tool call conversation ID
|
|
* @returns {Promise<{ ok?: number; n?: number; deletedCount?: number }>} The result of the delete operation
|
|
*/
|
|
async function deleteToolCalls(userId, conversationId) {
|
|
try {
|
|
const query = { user: userId };
|
|
if (conversationId) {
|
|
query.conversationId = conversationId;
|
|
}
|
|
return await ToolCall.deleteMany(query);
|
|
} catch (error) {
|
|
throw new Error(`Error deleting tool call: ${error.message}`);
|
|
}
|
|
}
|
|
|
|
module.exports = {
|
|
createToolCall,
|
|
updateToolCall,
|
|
deleteToolCalls,
|
|
getToolCallById,
|
|
getToolCallsByConvo,
|
|
getToolCallsByMessage,
|
|
};
|