/**
* Utility functions for handling Gradio endpoint detection and configuration
*/
import { GRADIO_FILES_TOOL_CONFIG } from '@llmindset/hf-mcp';
import { GRADIO_PREFIX, GRADIO_PRIVATE_PREFIX } from '../../shared/constants.js';
/**
* Determines if a tool name represents a Gradio endpoint
* Gradio tools follow the pattern: gr<number>_<name> or grp<number>_<name>
*
* @param toolName - The name of the tool to check
* @returns true if the tool is a Gradio endpoint, false otherwise
*
* @example
* isGradioTool('gr1_evalstate_flux1_schnell') // true
* isGradioTool('grp2_private_tool') // true
* isGradioTool('hf_doc_search') // false
* isGradioTool('regular_tool') // false
*/
export function isGradioTool(toolName: string): boolean {
// Gradio tools follow pattern: gr<number>_<name> or grp<number>_<name>
return /^grp?\d+_/.test(toolName) || toolName === GRADIO_FILES_TOOL_CONFIG.name;
}
/**
* Creates a Gradio tool name based on tool name, index, and privacy status
* This is the core logic used throughout the application for generating tool names
*
* @param toolName - The tool name (e.g., "flux1_schnell", "EasyGhibli")
* @param index - Zero-based index position (will be converted to 1-based)
* @param isPrivate - Whether this is a private space (determines gr vs grp prefix)
* @param toolIndex - Optional tool index within the endpoint for uniqueness when truncating
* @returns The generated tool name following Gradio naming convention
*
* @example
* createGradioToolName('flux1_schnell', 0, false) // 'gr1_flux1_schnell'
* createGradioToolName('EasyGhibli', 1, false) // 'gr2_easyghibli'
* createGradioToolName('private.model', 2, true) // 'grp3_private_model'
*/
export function createGradioToolName(
toolName: string,
index: number,
isPrivate: boolean | undefined,
toolIndex?: number
): string {
// Choose prefix based on privacy status
const prefix = isPrivate ? GRADIO_PRIVATE_PREFIX : GRADIO_PREFIX;
const indexStr = (index + 1).toString();
// Calculate available space for the sanitized name (49 - prefix - index - underscore)
const maxNameLength = 49 - prefix.length - indexStr.length - 1;
// Sanitize the tool name: replace special characters with underscores, normalize multiple underscores, and lowercase
let sanitizedName = toolName
? toolName
.replace(/[-\s.]+/g, '_') // Replace special chars with underscores
.toLowerCase()
: 'unknown';
// Handle based on length
if (sanitizedName.length > maxNameLength) {
// Over limit: insert tool index at beginning if provided, then truncate
if (toolIndex !== undefined) {
// Insert tool index after the underscore: gr1_0_toolname
const toolIndexPrefix = `${toolIndex}_`;
const availableForName = maxNameLength - toolIndexPrefix.length;
// Keep first 20 chars, add underscore, then keep as many chars from the end as possible
const keepFromEnd = availableForName - 20 - 1; // -1 for the underscore
sanitizedName = toolIndexPrefix + sanitizedName.substring(0, 20) + '_' + sanitizedName.slice(-keepFromEnd);
} else {
// No tool index, just do middle truncation as before
const keepFromEnd = maxNameLength - 20 - 1; // -1 for the underscore
sanitizedName = sanitizedName.substring(0, 20) + '_' + sanitizedName.slice(-keepFromEnd);
}
}
// Under limit: keep as-is, no normalization
// Create tool name: {prefix}{1-based-index}_{sanitized_name}
return `${prefix}${indexStr}_${sanitizedName}`;
}
