add update and delete mutations

This commit is contained in:
Dustin Healy 2025-06-27 12:30:38 -07:00
parent 568ec2f7d5
commit 5a7bf0b35f
8 changed files with 114 additions and 11 deletions

View file

@ -1,5 +1,5 @@
const express = require('express');
const { addTool } = require('@librechat/api');
const { addTool, updateTool, deleteTool } = require('@librechat/api');
const { callTool, verifyToolAuth, getToolCalls } = require('~/server/controllers/tools');
const { getAvailableTools } = require('~/server/controllers/PluginController');
const { toolCallLimiter } = require('~/server/middleware/limiters');
@ -45,4 +45,21 @@ router.post('/:toolId/call', toolCallLimiter, callTool);
*/
router.post('/add', addTool);
/**
* Update an existing tool/MCP in the system
* @route PUT /agents/tools/:mcp_id
* @param {string} mcp_id - The ID of the MCP to update
* @param {object} req.body - Request body containing updated tool/MCP data
* @returns {object} Updated tool/MCP object
*/
router.put('/:mcp_id', updateTool);
/**
* Delete a tool/MCP from the system
* @route DELETE /agents/tools/:mcp_id
* @param {string} mcp_id - The ID of the MCP to delete
* @returns {object} Deletion confirmation
*/
router.delete('/:mcp_id', deleteTool);
module.exports = router;

View file

@ -62,3 +62,45 @@ export const useCreateMCPMutation = (
},
);
};
export const useUpdateMCPMutation = (
options?: t.UpdateMCPMutationOptions,
): UseMutationResult<Record<string, unknown>, Error, { mcp_id: string; data: t.MCP }> => {
const queryClient = useQueryClient();
return useMutation(
({ mcp_id, data }: { mcp_id: string; data: t.MCP }) => {
return dataService.updateMCP({ mcp_id, data });
},
{
onMutate: (variables) => options?.onMutate?.(variables),
onError: (error, variables, context) => options?.onError?.(error, variables, context),
onSuccess: (data, variables, context) => {
// Invalidate tools list to trigger refetch
queryClient.invalidateQueries([QueryKeys.tools]);
return options?.onSuccess?.(data, variables, context);
},
},
);
};
export const useDeleteMCPMutation = (
options?: t.DeleteMCPMutationOptions,
): UseMutationResult<Record<string, unknown>, Error, { mcp_id: string }> => {
const queryClient = useQueryClient();
return useMutation(
({ mcp_id }: { mcp_id: string }) => {
return dataService.deleteMCP({ mcp_id });
},
{
onMutate: (variables) => options?.onMutate?.(variables),
onError: (error, variables, context) => options?.onError?.(error, variables, context),
onSuccess: (data, variables, context) => {
// Invalidate tools list to trigger refetch
queryClient.invalidateQueries([QueryKeys.tools]);
return options?.onSuccess?.(data, variables, context);
},
},
);
};

View file

@ -30,6 +30,8 @@ export function useMCPSelect({ conversationId }: UseMCPSelectOptions) {
const [ephemeralAgent, setEphemeralAgent] = useRecoilState(ephemeralAgentByConvoId(key));
const { data: mcpToolDetails, isFetched } = useAvailableToolsQuery(EModelEndpoint.agents, {
select: (data: TPlugin[]) => {
console.log('🔍 Raw tools data received:', JSON.stringify(data, null, 2));
const mcpToolsMap = new Map<string, TPlugin>();
data.forEach((tool) => {
const isMCP = tool.pluginKey.includes(Constants.mcp_delimiter);
@ -46,7 +48,10 @@ export function useMCPSelect({ conversationId }: UseMCPSelectOptions) {
}
}
});
return Array.from(mcpToolsMap.values());
const result = Array.from(mcpToolsMap.values());
console.log('🔧 Processed MCP tools:', JSON.stringify(result, null, 2));
return result;
},
});

View file

@ -2,7 +2,7 @@
export * from './mcp/manager';
export * from './mcp/oauth';
export * from './mcp/auth';
export * from './mcp/add';
export * from './mcp/mcpOps';
/* Utilities */
export * from './mcp/utils';
export * from './utils';

View file

@ -1,8 +0,0 @@
import { MCP } from 'librechat-data-provider';
import { Response } from 'express';
// just log the request
export const addTool = (req: { body: MCP }, res: Response) => {
console.log(JSON.stringify(req.body, null, 2));
res.send('ok');
};

View file

@ -0,0 +1,17 @@
import { MCP } from 'librechat-data-provider';
import { Response } from 'express';
export const addTool = (req: { body: MCP }, res: Response) => {
console.log('CREATE MCP:', JSON.stringify(req.body, null, 2));
res.send('ok');
};
export const updateTool = (req: { body: MCP; params: { mcp_id: string } }, res: Response) => {
console.log('UPDATE MCP:', req.params.mcp_id, JSON.stringify(req.body, null, 2));
res.send('ok');
};
export const deleteTool = (req: { params: { mcp_id: string } }, res: Response) => {
console.log('DELETE MCP:', req.params.mcp_id);
res.send('ok');
};

View file

@ -841,3 +841,26 @@ export const createMCP = (mcp: ag.MCP): Promise<Record<string, unknown>> => {
mcp,
);
};
export const updateMCP = ({
mcp_id,
data,
}: {
mcp_id: string;
data: ag.MCP;
}): Promise<Record<string, unknown>> => {
return request.put(
endpoints.agents({
path: `tools/${mcp_id}`,
}),
data,
);
};
export const deleteMCP = ({ mcp_id }: { mcp_id: string }): Promise<Record<string, unknown>> => {
return request.delete(
endpoints.agents({
path: `tools/${mcp_id}`,
}),
);
};

View file

@ -321,6 +321,13 @@ export type UpdatePluginAuthOptions = MutationOptions<types.TUser, types.TUpdate
export type CreateMCPMutationOptions = MutationOptions<Record<string, unknown>, MCP>;
export type UpdateMCPMutationOptions = MutationOptions<
Record<string, unknown>,
{ mcp_id: string; data: MCP }
>;
export type DeleteMCPMutationOptions = MutationOptions<Record<string, unknown>, { mcp_id: string }>;
export type ToolParamsMap = {
[Tools.execute_code]: {
lang: string;