Add main tool handler files

This commit is contained in:
Kaustabh Ganguly
2025-03-26 22:21:35 +05:30
parent 9ddbf4e56c
commit 9bf86febe4
6 changed files with 591 additions and 0 deletions

View File

@@ -0,0 +1,50 @@
import { ModelCache } from '../model-cache.js';
export interface ValidateModelToolRequest {
model: string;
}
export async function handleValidateModel(
request: { params: { arguments: ValidateModelToolRequest } },
modelCache: ModelCache
) {
const args = request.params.arguments;
try {
if (!modelCache.isCacheValid()) {
return {
content: [
{
type: 'text',
text: 'Model cache is empty or expired. Please call search_models first to populate the cache.',
},
],
isError: true,
};
}
const isValid = modelCache.hasModel(args.model);
return {
content: [
{
type: 'text',
text: JSON.stringify({ valid: isValid }),
},
],
};
} catch (error) {
if (error instanceof Error) {
return {
content: [
{
type: 'text',
text: `Error validating model: ${error.message}`,
},
],
isError: true,
};
}
throw error;
}
}