mirror of
https://github.com/rowboatlabs/rowboat.git
synced 2026-05-08 06:42:39 +02:00
housekeeping
This commit is contained in:
Ramnique Singh
parent
3755e76474
commit
cc8e9210d7
4 changed files with 0 additions and 1130 deletions
|
|
@ -1,295 +0,0 @@
|
|||
"use server";
|
||||
import { z } from "zod";
|
||||
import { WorkflowTool } from "../lib/types/workflow_types";
|
||||
import { projectAuthCheck } from "./project.actions";
|
||||
import { projectsCollection } from "../lib/mongodb";
|
||||
import { Project } from "../lib/types/project_types";
|
||||
import { McpServerTool, convertMcpServerToolToWorkflowTool } from "../lib/types/types";
|
||||
import { getMcpClient } from "../lib/mcp";
|
||||
|
||||
export async function fetchMcpTools(projectId: string): Promise<z.infer<typeof WorkflowTool>[]> {
|
||||
await projectAuthCheck(projectId);
|
||||
|
||||
const project = await projectsCollection.findOne({
|
||||
_id: projectId,
|
||||
});
|
||||
|
||||
const mcpServers = project?.mcpServers ?? [];
|
||||
const tools: z.infer<typeof WorkflowTool>[] = [];
|
||||
|
||||
for (const mcpServer of mcpServers) {
|
||||
if (!mcpServer.isActive) continue;
|
||||
|
||||
try {
|
||||
const client = await getMcpClient(mcpServer.serverUrl!, mcpServer.name);
|
||||
|
||||
// List tools
|
||||
const result = await client.listTools();
|
||||
|
||||
// Validate and parse each tool
|
||||
const validTools = await Promise.all(
|
||||
result.tools.map(async (tool) => {
|
||||
try {
|
||||
return McpServerTool.parse(tool);
|
||||
} catch (error) {
|
||||
console.error(`Invalid tool response from ${mcpServer.name}:`, {
|
||||
tool: tool.name,
|
||||
error: error instanceof Error ? error.message : 'Unknown error'
|
||||
});
|
||||
return null;
|
||||
}
|
||||
})
|
||||
);
|
||||
|
||||
// Filter out invalid tools and convert valid ones
|
||||
tools.push(...validTools
|
||||
.filter((tool): tool is z.infer<typeof McpServerTool> =>
|
||||
tool !== null &&
|
||||
mcpServer.tools.some(t => t.id === tool.name)
|
||||
)
|
||||
.map(mcpTool => convertMcpServerToolToWorkflowTool(mcpTool, mcpServer))
|
||||
);
|
||||
} catch (e) {
|
||||
console.error(`Error fetching MCP tools from ${mcpServer.name}:`, {
|
||||
error: e instanceof Error ? e.message : 'Unknown error',
|
||||
serverUrl: mcpServer.serverUrl
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
return tools;
|
||||
}
|
||||
|
||||
export async function fetchMcpToolsForServer(projectId: string, serverName: string): Promise<z.infer<typeof WorkflowTool>[]> {
|
||||
await projectAuthCheck(projectId);
|
||||
|
||||
console.log('[Klavis API] Fetching tools for specific server:', { projectId, serverName });
|
||||
|
||||
const project = await projectsCollection.findOne({
|
||||
_id: projectId,
|
||||
});
|
||||
|
||||
const mcpServer = project?.mcpServers?.find(server => server.name === serverName);
|
||||
if (!mcpServer) {
|
||||
console.error('[Klavis API] Server not found:', { serverName });
|
||||
return [];
|
||||
}
|
||||
|
||||
if (!mcpServer.isActive || !mcpServer.serverUrl) {
|
||||
console.log('[Klavis API] Server is not active or missing URL:', {
|
||||
serverName,
|
||||
isActive: mcpServer.isActive,
|
||||
hasUrl: !!mcpServer.serverUrl
|
||||
});
|
||||
return [];
|
||||
}
|
||||
|
||||
const tools: z.infer<typeof WorkflowTool>[] = [];
|
||||
|
||||
try {
|
||||
console.log('[Klavis API] Attempting MCP connection:', {
|
||||
serverName,
|
||||
url: mcpServer.serverUrl
|
||||
});
|
||||
|
||||
const client = await getMcpClient(mcpServer.serverUrl, mcpServer.name);
|
||||
|
||||
// List tools
|
||||
const result = await client.listTools();
|
||||
|
||||
// Log just essential info about tools
|
||||
console.log('[Klavis API] Received tools from server:', {
|
||||
serverName,
|
||||
toolCount: result.tools.length,
|
||||
tools: result.tools.map(tool => tool.name).join(', ')
|
||||
});
|
||||
|
||||
// Get all available tools from the server
|
||||
const availableToolNames = new Set(mcpServer.availableTools?.map(t => t.name) || []);
|
||||
|
||||
// Validate and parse each tool
|
||||
const validTools = await Promise.all(
|
||||
result.tools.map(async (tool) => {
|
||||
try {
|
||||
const parsedTool = McpServerTool.parse(tool);
|
||||
return parsedTool;
|
||||
} catch (error) {
|
||||
console.error(`Invalid tool response from ${mcpServer.name}:`, {
|
||||
tool: tool.name,
|
||||
error: error instanceof Error ? error.message : 'Unknown error'
|
||||
});
|
||||
return null;
|
||||
}
|
||||
})
|
||||
);
|
||||
|
||||
// Filter out invalid tools and convert valid ones
|
||||
const convertedTools = validTools
|
||||
.filter((tool): tool is z.infer<typeof McpServerTool> => tool !== null)
|
||||
.map(mcpTool => {
|
||||
const converted = convertMcpServerToolToWorkflowTool(mcpTool, mcpServer);
|
||||
return converted;
|
||||
});
|
||||
|
||||
tools.push(...convertedTools);
|
||||
|
||||
// Find tools that weren't enriched
|
||||
const enrichedToolNames = new Set(convertedTools.map(t => t.name));
|
||||
const unenrichedTools = Array.from(availableToolNames).filter(name => !enrichedToolNames.has(name));
|
||||
|
||||
if (unenrichedTools.length > 0) {
|
||||
console.log('[Klavis API] Tools that could not be enriched:', {
|
||||
serverName,
|
||||
unenrichedTools,
|
||||
totalAvailable: availableToolNames.size,
|
||||
totalEnriched: enrichedToolNames.size
|
||||
});
|
||||
}
|
||||
|
||||
console.log('[Klavis API] Successfully fetched tools for server:', {
|
||||
serverName,
|
||||
toolCount: tools.length,
|
||||
availableToolCount: availableToolNames.size,
|
||||
tools: tools.map(t => t.name).join(', ')
|
||||
});
|
||||
} catch (e) {
|
||||
console.error(`[Klavis API] Error fetching MCP tools from ${mcpServer.name}:`, {
|
||||
error: e instanceof Error ? e.message : 'Unknown error',
|
||||
serverUrl: mcpServer.serverUrl
|
||||
});
|
||||
}
|
||||
|
||||
return tools;
|
||||
}
|
||||
|
||||
export async function updateMcpServers(projectId: string, mcpServers: z.infer<typeof Project>['mcpServers']): Promise<void> {
|
||||
await projectAuthCheck(projectId);
|
||||
await projectsCollection.updateOne({
|
||||
_id: projectId,
|
||||
}, { $set: { mcpServers } });
|
||||
}
|
||||
|
||||
export async function toggleMcpTool(
|
||||
projectId: string,
|
||||
serverName: string,
|
||||
toolId: string,
|
||||
shouldAdd: boolean
|
||||
): Promise<void> {
|
||||
await projectAuthCheck(projectId);
|
||||
|
||||
// 1. Get the project and find the server
|
||||
const project = await projectsCollection.findOne({ _id: projectId });
|
||||
if (!project) throw new Error("Project not found");
|
||||
|
||||
const mcpServers = project.mcpServers || [];
|
||||
const serverIndex = mcpServers.findIndex(s => s.serverName === serverName);
|
||||
if (serverIndex === -1) throw new Error("Server not found");
|
||||
|
||||
const server = mcpServers[serverIndex];
|
||||
|
||||
if (shouldAdd) {
|
||||
// Add tool if it doesn't exist
|
||||
const toolExists = server.tools.some(t => t.id === toolId);
|
||||
if (!toolExists) {
|
||||
// Find the tool in availableTools to get its parameters
|
||||
const availableTool = server.availableTools?.find(t => t.name === toolId);
|
||||
|
||||
// Create a new tool with the parameters from availableTools
|
||||
const newTool = {
|
||||
id: toolId,
|
||||
name: toolId,
|
||||
description: availableTool?.description || '',
|
||||
parameters: availableTool?.parameters || {
|
||||
type: 'object' as const,
|
||||
properties: {},
|
||||
required: []
|
||||
}
|
||||
};
|
||||
server.tools.push(newTool);
|
||||
}
|
||||
} else {
|
||||
// Remove tool if it exists
|
||||
server.tools = server.tools.filter(t => t.id !== toolId);
|
||||
}
|
||||
|
||||
// Update the project
|
||||
await projectsCollection.updateOne(
|
||||
{ _id: projectId },
|
||||
{ $set: { mcpServers } }
|
||||
);
|
||||
}
|
||||
|
||||
export async function getSelectedMcpTools(projectId: string, serverName: string): Promise<string[]> {
|
||||
await projectAuthCheck(projectId);
|
||||
const project = await projectsCollection.findOne({ _id: projectId });
|
||||
if (!project) return [];
|
||||
|
||||
const server = project.mcpServers?.find(s => s.serverName === serverName);
|
||||
if (!server) return [];
|
||||
|
||||
return server.tools.map(t => t.id);
|
||||
}
|
||||
|
||||
export async function testMcpTool(
|
||||
projectId: string,
|
||||
serverName: string,
|
||||
toolId: string,
|
||||
parameters: Record<string, any>
|
||||
): Promise<any> {
|
||||
await projectAuthCheck(projectId);
|
||||
|
||||
const project = await projectsCollection.findOne({
|
||||
_id: projectId,
|
||||
});
|
||||
|
||||
// Find the server by name in mcpServers array
|
||||
const mcpServer = project?.mcpServers?.find(server => server.name === serverName);
|
||||
if (!mcpServer) {
|
||||
throw new Error(`Server ${serverName} not found`);
|
||||
}
|
||||
|
||||
if (!mcpServer.isActive) {
|
||||
throw new Error(`Server ${serverName} is not active`);
|
||||
}
|
||||
|
||||
if (!mcpServer.serverUrl) {
|
||||
throw new Error(`Server ${serverName} has no URL configured`);
|
||||
}
|
||||
|
||||
try {
|
||||
console.log('[MCP Test] Attempting to test tool:', {
|
||||
serverName,
|
||||
serverUrl: mcpServer.serverUrl,
|
||||
toolId
|
||||
});
|
||||
|
||||
const client = await getMcpClient(mcpServer.serverUrl, mcpServer.name);
|
||||
|
||||
console.log('[MCP Test] Connected to server, calling tool:', {
|
||||
toolId,
|
||||
parameters
|
||||
});
|
||||
|
||||
// Execute the tool with the correct parameter format
|
||||
const result = await client.callTool({
|
||||
name: toolId,
|
||||
arguments: parameters
|
||||
});
|
||||
|
||||
console.log('[MCP Test] Tool execution completed:', {
|
||||
toolId,
|
||||
success: true
|
||||
});
|
||||
|
||||
return result;
|
||||
|
||||
} catch (e) {
|
||||
console.error(`[MCP Test] Error testing tool from ${mcpServer.name}:`, {
|
||||
error: e instanceof Error ? e.message : 'Unknown error',
|
||||
serverUrl: mcpServer.serverUrl,
|
||||
toolId,
|
||||
parameters
|
||||
});
|
||||
throw e;
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue