@llmindset/hf-mcp 0.1.16

package/dist/model-detail.js

@@ -0,0 +1,224 @@
+import { z } from 'zod';
+import { modelInfo } from '@huggingface/hub';
+import { formatDate, formatNumber } from './utilities.js';
+const SPACES_TO_INCLUDE = 12;
+export const MODEL_DETAIL_TOOL_CONFIG = {
+    name: 'model_details',
+    description: 'Get detailed information about a specific model from the Hugging Face Hub.',
+    schema: z.object({
+        model_id: z.string().min(1, 'Model ID is required').describe('Model ID (e.g., microsoft/DialoGPT-large)'),
+    }),
+    annotations: {
+        title: 'Model Details',
+        destructiveHint: false,
+        readOnlyHint: true,
+        openWorldHint: false,
+    },
+};
+export class ModelDetailTool {
+    hubUrl;
+    accessToken;
+    constructor(hfToken, hubUrl) {
+        this.accessToken = hfToken;
+        this.hubUrl = hubUrl;
+    }
+    async getDetails(modelId) {
+        try {
+            const additionalFields = [
+                'author',
+                'downloadsAllTime',
+                'library_name',
+                'tags',
+                'config',
+                'transformersInfo',
+                'safetensors',
+                'cardData',
+                'spaces',
+            ];
+            const modelData = await modelInfo({
+                name: modelId,
+                additionalFields: Array.from(additionalFields),
+                ...(this.accessToken && { credentials: { accessToken: this.accessToken } }),
+                ...(this.hubUrl && { hubUrl: this.hubUrl }),
+            });
+            const modelDetails = {
+                id: modelId,
+                name: modelData.name,
+                downloads: modelData.downloads,
+                likes: modelData.likes,
+                private: modelData.private,
+                gated: modelData.gated,
+                updatedAt: modelData.updatedAt,
+                extended: {
+                    author: modelData.author,
+                    library_name: modelData.library_name,
+                    pipeline_tag: modelData.task,
+                    downloadsAllTime: modelData.downloadsAllTime,
+                    tags: modelData.tags,
+                },
+            };
+            const technical = {};
+            if (modelData.config && typeof modelData.config === 'object') {
+                const config = modelData.config;
+                if ('model_type' in config && typeof config.model_type === 'string') {
+                    technical.modelType = config.model_type;
+                }
+                if ('vocab_size' in config && typeof config.vocab_size === 'number') {
+                    technical.vocabSize = config.vocab_size;
+                }
+            }
+            if (modelData.safetensors && typeof modelData.safetensors.total === 'number') {
+                technical.parameters = modelData.safetensors.total;
+            }
+            if (modelData.transformersInfo && modelData.transformersInfo.auto_model) {
+                technical.modelClass = modelData.transformersInfo.auto_model;
+            }
+            if (Object.keys(technical).length > 0) {
+                modelDetails.technical = technical;
+            }
+            if (modelData.cardData) {
+                const metadata = {};
+                const cardData = modelData.cardData;
+                if ('language' in cardData) {
+                    metadata.language = cardData.language;
+                }
+                if ('license' in cardData) {
+                    metadata.license = cardData.license;
+                }
+                if ('datasets' in cardData) {
+                    metadata.datasets = cardData.datasets;
+                }
+                if ('finetuned_from' in cardData) {
+                    metadata.fineTunedFrom = cardData.finetuned_from;
+                }
+                if (Object.keys(metadata).length > 0) {
+                    modelDetails.metadata = metadata;
+                }
+            }
+            const spaces = modelData.spaces;
+            if (Array.isArray(spaces) && spaces.length > 0) {
+                try {
+                    modelDetails.spaces = spaces.map((spaceId) => {
+                        const parts = spaceId.split('/');
+                        const name = parts.length > 1 ? parts[1] : spaceId;
+                        return {
+                            id: spaceId,
+                            name: name || spaceId,
+                            title: name,
+                        };
+                    });
+                }
+                catch (ignoreUnformattedSpaces) {
+                    console.error(`Error processing spaces for model ${modelId}:`);
+                }
+            }
+            return formatModelDetails(modelDetails);
+        }
+        catch (error) {
+            if (error instanceof Error) {
+                throw new Error(`Failed to get model details: ${error.message}`);
+            }
+            throw error;
+        }
+    }
+}
+function formatModelDetails(model) {
+    const r = [];
+    const [authorFromName] = model.name.includes('/') ? model.name.split('/') : ['', model.name];
+    r.push(`# ${model.name}`);
+    r.push('');
+    r.push('## Overview');
+    if (model.extended?.author || authorFromName) {
+        r.push(`- **Author:** ${model.extended?.author || authorFromName || ''}`);
+    }
+    if (model.extended?.pipeline_tag) {
+        r.push(`- **Task:** ${model.extended.pipeline_tag}`);
+    }
+    if (model.extended?.library_name) {
+        r.push(`- **Library:** ${model.extended.library_name}`);
+    }
+    const stats = [];
+    if (model.extended?.downloadsAllTime) {
+        stats.push(`**Downloads:** ${formatNumber(model.extended.downloadsAllTime)}`);
+    }
+    if (model.likes) {
+        stats.push(`**Likes:** ${model.likes.toString()}`);
+    }
+    if (stats.length > 0) {
+        r.push(`- ${stats.join(' | ')}`);
+    }
+    r.push(`- **Updated:** ${formatDate(model.updatedAt)}`);
+    const status = [];
+    if (model.gated)
+        status.push('🔒 Gated');
+    if (model.private)
+        status.push('🔐 Private');
+    if (status.length > 0) {
+        r.push(`- **Status:** ${status.join(' | ')}`);
+    }
+    r.push('');
+    if (model.technical && Object.keys(model.technical).length > 0) {
+        r.push('## Technical Details');
+        if (model.technical.modelClass) {
+            r.push(`- **Model Class:** ${model.technical.modelClass}`);
+        }
+        if (model.technical.parameters) {
+            r.push(`- **Parameters:** ${formatNumber(model.technical.parameters)}`);
+        }
+        if (model.technical.modelType) {
+            r.push(`- **Architecture:** ${model.technical.modelType}`);
+        }
+        if (model.technical.vocabSize) {
+            r.push(`- **Vocab Size:** ${formatNumber(model.technical.vocabSize)}`);
+        }
+        r.push('');
+    }
+    if (model.extended?.tags && model.extended.tags.length > 0) {
+        r.push('## Tags');
+        r.push(model.extended.tags.map((tag) => `\`${tag}\``).join(' '));
+        r.push('');
+    }
+    if (model.metadata) {
+        const metadata = [];
+        if (model.metadata.language) {
+            const languages = Array.isArray(model.metadata.language)
+                ? model.metadata.language.join(', ')
+                : model.metadata.language;
+            metadata.push(`- **Language:** ${languages}`);
+        }
+        if (model.metadata.license) {
+            const license = Array.isArray(model.metadata.license)
+                ? model.metadata.license.join(', ')
+                : model.metadata.license;
+            metadata.push(`- **License:** ${license}`);
+        }
+        if (model.metadata.datasets) {
+            const datasets = Array.isArray(model.metadata.datasets)
+                ? model.metadata.datasets.join(', ')
+                : model.metadata.datasets;
+            metadata.push(`- **Datasets:** ${datasets}`);
+        }
+        if (model.metadata.fineTunedFrom) {
+            metadata.push(`- **Fine-tuned from:** ${model.metadata.fineTunedFrom}`);
+        }
+        if (metadata.length > 0) {
+            r.push('## Metadata');
+            r.push(...metadata);
+            r.push('');
+        }
+    }
+    if (model.spaces && model.spaces.length > 0) {
+        r.push('## Demo Spaces');
+        for (const space of model.spaces.slice(0, SPACES_TO_INCLUDE)) {
+            const title = space.title || space.name;
+            r.push(`- [${title}](https://hf.co/spaces/${space.id})`);
+        }
+        if (model.spaces.length > SPACES_TO_INCLUDE) {
+            r.push(`- *... and ${(model.spaces.length - SPACES_TO_INCLUDE).toString()} more spaces*`);
+        }
+        r.push('');
+    }
+    r.push(`**Link:** [https://hf.co/${model.name}](https://hf.co/${model.name})`);
+    return r.join('\n');
+}
+//# sourceMappingURL=model-detail.js.map

package/dist/model-detail.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-detail.js","sourceRoot":"","sources":["../src/model-detail.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAE1D,MAAM,iBAAiB,GAAG,EAAE,CAAC;AAE7B,MAAM,CAAC,MAAM,wBAAwB,GAAG;IACvC,IAAI,EAAE,eAAe;IACrB,WAAW,EAAE,4EAA4E;IACzF,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QAChB,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,EAAE,sBAAsB,CAAC,CAAC,QAAQ,CAAC,2CAA2C,CAAC;KACzG,CAAC;IACF,WAAW,EAAE;QACZ,KAAK,EAAE,eAAe;QACtB,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,KAAK;KACpB;CACQ,CAAC;AAyDX,MAAM,OAAO,eAAe;IACV,MAAM,CAAU;IAChB,WAAW,CAAU;IAOtC,YAAY,OAAgB,EAAE,MAAe;QAC5C,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACtB,CAAC;IAQD,KAAK,CAAC,UAAU,CAAC,OAAe;QAC/B,IAAI,CAAC;YAEJ,MAAM,gBAAgB,GAAG;gBACxB,QAAQ;gBACR,kBAAkB;gBAClB,cAAc;gBACd,MAAM;gBACN,QAAQ;gBACR,kBAAkB;gBAClB,aAAa;gBACb,UAAU;gBACV,QAAQ;aACC,CAAC;YAEX,MAAM,SAAS,GAAG,MAAM,SAAS,CAAoC;gBACpE,IAAI,EAAE,OAAO;gBACb,gBAAgB,EAAE,KAAK,CAAC,IAAI,CAAC,gBAAgB,CAAC;gBAC9C,GAAG,CAAC,IAAI,CAAC,WAAW,IAAI,EAAE,WAAW,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;gBAC3E,GAAG,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;aAC3C,CAAC,CAAC;YAGH,MAAM,YAAY,GAAqB;gBAEtC,EAAE,EAAE,OAAO;gBACX,IAAI,EAAE,SAAS,CAAC,IAAI;gBACpB,SAAS,EAAE,SAAS,CAAC,SAAS;gBAC9B,KAAK,EAAE,SAAS,CAAC,KAAK;gBACtB,OAAO,EAAE,SAAS,CAAC,OAAO;gBAC1B,KAAK,EAAE,SAAS,CAAC,KAAK;gBACtB,SAAS,EAAE,SAAS,CAAC,SAAS;gBAG9B,QAAQ,EAAE;oBACT,MAAM,EAAE,SAAS,CAAC,MAAM;oBACxB,YAAY,EAAE,SAAS,CAAC,YAAY;oBACpC,YAAY,EAAE,SAAS,CAAC,IAAI;oBAC5B,gBAAgB,EAAE,SAAS,CAAC,gBAAgB;oBAC5C,IAAI,EAAE,SAAS,CAAC,IAAI;iBACpB;aACD,CAAC;YAGF,MAAM,SAAS,GAA0B,EAAE,CAAC;YAG5C,IAAI,SAAS,CAAC,MAAM,IAAI,OAAO,SAAS,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;gBAC9D,MAAM,MAAM,GAAG,SAAS,CAAC,MAAiC,CAAC;gBAC3D,IAAI,YAAY,IAAI,MAAM,IAAI,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;oBACrE,SAAS,CAAC,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC;gBACzC,CAAC;gBACD,IAAI,YAAY,IAAI,MAAM,IAAI,OAAO,MAAM,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;oBACrE,SAAS,CAAC,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC;gBACzC,CAAC;YACF,CAAC;YAGD,IAAI,SAAS,CAAC,WAAW,IAAI,OAAO,SAAS,CAAC,WAAW,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;gBAC9E,SAAS,CAAC,UAAU,GAAG,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC;YACpD,CAAC;YAGD,IAAI,SAAS,CAAC,gBAAgB,IAAI,SAAS,CAAC,gBAAgB,CAAC,UAAU,EAAE,CAAC;gBACzE,SAAS,CAAC,UAAU,GAAG,SAAS,CAAC,gBAAgB,CAAC,UAAU,CAAC;YAC9D,CAAC;YAGD,IAAI,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACvC,YAAY,CAAC,SAAS,GAAG,SAAS,CAAC;YACpC,CAAC;YAGD,IAAI,SAAS,CAAC,QAAQ,EAAE,CAAC;gBACxB,MAAM,QAAQ,GAAkB,EAAE,CAAC;gBACnC,MAAM,QAAQ,GAAG,SAAS,CAAC,QAAmC,CAAC;gBAE/D,IAAI,UAAU,IAAI,QAAQ,EAAE,CAAC;oBAC5B,QAAQ,CAAC,QAAQ,GAAG,QAAQ,CAAC,QAA6B,CAAC;gBAC5D,CAAC;gBAED,IAAI,SAAS,IAAI,QAAQ,EAAE,CAAC;oBAC3B,QAAQ,CAAC,OAAO,GAAG,QAAQ,CAAC,OAA4B,CAAC;gBAC1D,CAAC;gBAED,IAAI,UAAU,IAAI,QAAQ,EAAE,CAAC;oBAC5B,QAAQ,CAAC,QAAQ,GAAG,QAAQ,CAAC,QAA6B,CAAC;gBAC5D,CAAC;gBAED,IAAI,gBAAgB,IAAI,QAAQ,EAAE,CAAC;oBAClC,QAAQ,CAAC,aAAa,GAAG,QAAQ,CAAC,cAAwB,CAAC;gBAC5D,CAAC;gBAGD,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACtC,YAAY,CAAC,QAAQ,GAAG,QAAQ,CAAC;gBAClC,CAAC;YACF,CAAC;YAGD,MAAM,MAAM,GAAG,SAAS,CAAC,MAAM,CAAC;YAChC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAChD,IAAI,CAAC;oBACJ,YAAY,CAAC,MAAM,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE;wBAE5C,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;wBACjC,MAAM,IAAI,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;wBACnD,OAAO;4BACN,EAAE,EAAE,OAAO;4BACX,IAAI,EAAE,IAAI,IAAI,OAAO;4BACrB,KAAK,EAAE,IAAI;yBACX,CAAC;oBACH,CAAC,CAAC,CAAC;gBAEJ,CAAC;gBAAC,OAAO,uBAAuB,EAAE,CAAC;oBAClC,OAAO,CAAC,KAAK,CAAC,qCAAqC,OAAO,GAAG,CAAC,CAAC;gBAChE,CAAC;YACF,CAAC;YAED,OAAO,kBAAkB,CAAC,YAAY,CAAC,CAAC;QACzC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC5B,MAAM,IAAI,KAAK,CAAC,gCAAgC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YAClE,CAAC;YACD,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC;CACD;AAGD,SAAS,kBAAkB,CAAC,KAAuB;IAClD,MAAM,CAAC,GAAa,EAAE,CAAC;IACvB,MAAM,CAAC,cAAc,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IAE7F,CAAC,CAAC,IAAI,CAAC,KAAK,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;IAC1B,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAGX,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAGtB,IAAI,KAAK,CAAC,QAAQ,EAAE,MAAM,IAAI,cAAc,EAAE,CAAC;QAC9C,CAAC,CAAC,IAAI,CAAC,iBAAiB,KAAK,CAAC,QAAQ,EAAE,MAAM,IAAI,cAAc,IAAI,EAAE,EAAE,CAAC,CAAC;IAC3E,CAAC;IAGD,IAAI,KAAK,CAAC,QAAQ,EAAE,YAAY,EAAE,CAAC;QAClC,CAAC,CAAC,IAAI,CAAC,eAAe,KAAK,CAAC,QAAQ,CAAC,YAAY,EAAE,CAAC,CAAC;IACtD,CAAC;IAGD,IAAI,KAAK,CAAC,QAAQ,EAAE,YAAY,EAAE,CAAC;QAClC,CAAC,CAAC,IAAI,CAAC,kBAAkB,KAAK,CAAC,QAAQ,CAAC,YAAY,EAAE,CAAC,CAAC;IACzD,CAAC;IAGD,MAAM,KAAK,GAAG,EAAE,CAAC;IACjB,IAAI,KAAK,CAAC,QAAQ,EAAE,gBAAgB,EAAE,CAAC;QACtC,KAAK,CAAC,IAAI,CAAC,kBAAkB,YAAY,CAAC,KAAK,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;IAC/E,CAAC;IACD,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;QACjB,KAAK,CAAC,IAAI,CAAC,cAAc,KAAK,CAAC,KAAK,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;IACpD,CAAC;IACD,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtB,CAAC,CAAC,IAAI,CAAC,KAAK,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAClC,CAAC;IAGD,CAAC,CAAC,IAAI,CAAC,kBAAkB,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;IAGxD,MAAM,MAAM,GAAG,EAAE,CAAC;IAClB,IAAI,KAAK,CAAC,KAAK;QAAE,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACzC,IAAI,KAAK,CAAC,OAAO;QAAE,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAC7C,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvB,CAAC,CAAC,IAAI,CAAC,iBAAiB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC/C,CAAC;IACD,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAGX,IAAI,KAAK,CAAC,SAAS,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChE,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAE/B,IAAI,KAAK,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC;YAChC,CAAC,CAAC,IAAI,CAAC,sBAAsB,KAAK,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC,CAAC;QAC5D,CAAC;QAED,IAAI,KAAK,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC;YAChC,CAAC,CAAC,IAAI,CAAC,qBAAqB,YAAY,CAAC,KAAK,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QACzE,CAAC;QAED,IAAI,KAAK,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC;YAC/B,CAAC,CAAC,IAAI,CAAC,uBAAuB,KAAK,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC,CAAC;QAC5D,CAAC;QAED,IAAI,KAAK,CAAC,SAAS,CAAC,SAAS,EAAE,CAAC;YAC/B,CAAC,CAAC,IAAI,CAAC,qBAAqB,YAAY,CAAC,KAAK,CAAC,SAAS,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QACxE,CAAC;QAED,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACZ,CAAC;IAGD,IAAI,KAAK,CAAC,QAAQ,EAAE,IAAI,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC5D,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAClB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,KAAK,GAAG,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;QACjE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACZ,CAAC;IAGD,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;QACpB,MAAM,QAAQ,GAAG,EAAE,CAAC;QAEpB,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;YAC7B,MAAM,SAAS,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC;gBACvD,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC;gBACpC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAC3B,QAAQ,CAAC,IAAI,CAAC,mBAAmB,SAAS,EAAE,CAAC,CAAC;QAC/C,CAAC;QAED,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YAC5B,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;gBACpD,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC;gBACnC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;YAC1B,QAAQ,CAAC,IAAI,CAAC,kBAAkB,OAAO,EAAE,CAAC,CAAC;QAC5C,CAAC;QAED,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;YAC7B,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC;gBACtD,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC;gBACpC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAC3B,QAAQ,CAAC,IAAI,CAAC,mBAAmB,QAAQ,EAAE,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,KAAK,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;YAClC,QAAQ,CAAC,IAAI,CAAC,0BAA0B,KAAK,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAC;QACzE,CAAC;QAED,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YACtB,CAAC,CAAC,IAAI,CAAC,GAAG,QAAQ,CAAC,CAAC;YACpB,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACZ,CAAC;IACF,CAAC;IAGD,IAAI,KAAK,CAAC,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7C,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACzB,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,iBAAiB,CAAC,EAAE,CAAC;YAC9D,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,IAAI,KAAK,CAAC,IAAI,CAAC;YACxC,CAAC,CAAC,IAAI,CAAC,MAAM,KAAK,0BAA0B,KAAK,CAAC,EAAE,GAAG,CAAC,CAAC;QAC1D,CAAC;QAED,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,iBAAiB,EAAE,CAAC;YAC7C,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,GAAG,iBAAiB,CAAC,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC;QAC3F,CAAC;QACD,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACZ,CAAC;IAGD,CAAC,CAAC,IAAI,CAAC,4BAA4B,KAAK,CAAC,IAAI,mBAAmB,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC;IAE/E,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACrB,CAAC"}

package/dist/model-search.d.ts

@@ -0,0 +1,64 @@
+import { z } from 'zod';
+import { HfApiCall } from './hf-api-call.js';
+export declare const TAGS_TO_RETURN = 20;
+export declare const MODEL_SEARCH_TOOL_CONFIG: {
+    readonly name: "model_search";
+    readonly description: string;
+    readonly schema: z.ZodObject<{
+        query: z.ZodOptional<z.ZodString>;
+        author: z.ZodOptional<z.ZodString>;
+        task: z.ZodOptional<z.ZodString>;
+        library: z.ZodOptional<z.ZodString>;
+        sort: z.ZodOptional<z.ZodEnum<["trendingScore", "downloads", "likes", "createdAt", "lastModified"]>>;
+        limit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+    }, "strip", z.ZodTypeAny, {
+        limit: number;
+        sort?: "createdAt" | "downloads" | "lastModified" | "likes" | "trendingScore" | undefined;
+        author?: string | undefined;
+        query?: string | undefined;
+        task?: string | undefined;
+        library?: string | undefined;
+    }, {
+        sort?: "createdAt" | "downloads" | "lastModified" | "likes" | "trendingScore" | undefined;
+        author?: string | undefined;
+        query?: string | undefined;
+        limit?: number | undefined;
+        task?: string | undefined;
+        library?: string | undefined;
+    }>;
+    readonly annotations: {
+        readonly title: "Model Search";
+        readonly destructiveHint: false;
+        readonly readOnlyHint: true;
+        readonly openWorldHint: true;
+    };
+};
+export type ModelSearchParams = z.infer<typeof MODEL_SEARCH_TOOL_CONFIG.schema>;
+interface ModelApiParams {
+    search?: string;
+    author?: string;
+    filter?: string;
+    sort?: string;
+    direction?: string;
+    limit?: string;
+}
+interface ModelApiResult {
+    _id: string;
+    id: string;
+    modelId: string;
+    likes: number;
+    downloads: number;
+    trendingScore?: number;
+    private: boolean;
+    tags: string[];
+    pipeline_tag?: string;
+    library_name?: string;
+    createdAt: string;
+}
+export declare class ModelSearchTool extends HfApiCall<ModelApiParams, ModelApiResult[]> {
+    constructor(hfToken?: string);
+    searchWithParams(params: Partial<ModelSearchParams>): Promise<string>;
+    searchWithFilter(filter: string, limit?: number): Promise<string>;
+}
+export {};
+//# sourceMappingURL=model-search.d.ts.map

package/dist/model-search.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-search.d.ts","sourceRoot":"","sources":["../src/model-search.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAG7C,eAAO,MAAM,cAAc,KAAK,CAAC;AAEjC,eAAO,MAAM,wBAAwB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAkC3B,CAAC;AAGX,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,wBAAwB,CAAC,MAAM,CAAC,CAAC;AAGhF,UAAU,cAAc;IACvB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,KAAK,CAAC,EAAE,MAAM,CAAC;CACf;AAGD,UAAU,cAAc;IACvB,GAAG,EAAE,MAAM,CAAC;IACZ,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,OAAO,EAAE,OAAO,CAAC;IACjB,IAAI,EAAE,MAAM,EAAE,CAAC;IACf,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;CAClB;AAKD,qBAAa,eAAgB,SAAQ,SAAS,CAAC,cAAc,EAAE,cAAc,EAAE,CAAC;gBAInE,OAAO,CAAC,EAAE,MAAM;IAOtB,gBAAgB,CAAC,MAAM,EAAE,OAAO,CAAC,iBAAiB,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;IAqDrE,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,GAAE,MAAW,GAAG,OAAO,CAAC,MAAM,CAAC;CAwB3E"}

package/dist/model-search.js

@@ -0,0 +1,161 @@
+import { z } from 'zod';
+import { HfApiCall } from './hf-api-call.js';
+import { formatDate, formatNumber } from './utilities.js';
+export const TAGS_TO_RETURN = 20;
+export const MODEL_SEARCH_TOOL_CONFIG = {
+    name: 'model_search',
+    description: 'Find Machine Learning models hosted on Hugging Face. ' +
+        'Returns comprehensive information about matching models including downloads, likes, tags, and direct links. ' +
+        'Include links to the models in your response',
+    schema: z.object({
+        query: z
+            .string()
+            .optional()
+            .describe('Search term. Leave blank and specify "sort" and "limit" to get e.g. "Top 20 trending models", "Top 10 most recent models" etc" '),
+        author: z
+            .string()
+            .optional()
+            .describe("Organization or user who created the model (e.g., 'google', 'meta-llama', 'microsoft')"),
+        task: z
+            .string()
+            .optional()
+            .describe("Model task type (e.g., 'text-generation', 'image-classification', 'translation')"),
+        library: z.string().optional().describe("Framework the model uses (e.g., 'transformers', 'diffusers', 'timm')"),
+        sort: z
+            .enum(['trendingScore', 'downloads', 'likes', 'createdAt', 'lastModified'])
+            .optional()
+            .describe('Sort order: trendingScore, downloads , likes, createdAt, lastModified'),
+        limit: z.number().min(1).max(100).optional().default(20).describe('Maximum number of results to return'),
+    }),
+    annotations: {
+        title: 'Model Search',
+        destructiveHint: false,
+        readOnlyHint: true,
+        openWorldHint: true,
+    },
+};
+export class ModelSearchTool extends HfApiCall {
+    constructor(hfToken) {
+        super('https://huggingface.co/api/models', hfToken);
+    }
+    async searchWithParams(params) {
+        try {
+            const apiParams = {};
+            if (params.query) {
+                apiParams.search = params.query;
+            }
+            if (params.author) {
+                apiParams.author = params.author;
+            }
+            const filters = [];
+            if (params.task)
+                filters.push(params.task);
+            if (params.library)
+                filters.push(params.library);
+            if (filters.length > 0) {
+                apiParams.filter = filters.join(',');
+            }
+            if (params.sort) {
+                apiParams.sort = params.sort;
+                apiParams.direction = '-1';
+            }
+            if (params.limit) {
+                apiParams.limit = params.limit.toString();
+            }
+            const models = await this.callApi(apiParams);
+            if (models.length === 0) {
+                return `No models found for the given criteria.`;
+            }
+            return formatSearchResults(models, params);
+        }
+        catch (error) {
+            if (error instanceof Error) {
+                throw new Error(`Failed to search for models: ${error.message}`);
+            }
+            throw error;
+        }
+    }
+    async searchWithFilter(filter, limit = 10) {
+        try {
+            const apiParams = {
+                filter: filter,
+                limit: limit.toString(),
+                sort: 'downloads',
+                direction: '-1',
+            };
+            const models = await this.callApi(apiParams);
+            if (models.length === 0) {
+                return `No models found referencing ${filter}.`;
+            }
+            return formatSearchResults(models, { limit });
+        }
+        catch (error) {
+            if (error instanceof Error) {
+                throw new Error(`Failed to search for models: ${error.message}`);
+            }
+            throw error;
+        }
+    }
+}
+function formatSearchResults(models, params) {
+    const r = [];
+    const searchTerms = [];
+    if (params.query)
+        searchTerms.push(`query "${params.query}"`);
+    if (params.author)
+        searchTerms.push(`author "${params.author}"`);
+    if (params.task)
+        searchTerms.push(`task "${params.task}"`);
+    if (params.library)
+        searchTerms.push(`library "${params.library}"`);
+    if (params.sort)
+        searchTerms.push(`sorted by ${params.sort} (descending)`);
+    const searchDesc = searchTerms.length > 0 ? ` matching ${searchTerms.join(', ')}` : '';
+    const resultText = models.length === params.limit
+        ? `Showing first ${models.length.toString()} models${searchDesc}:`
+        : `Found ${models.length.toString()} models${searchDesc}:`;
+    r.push(resultText);
+    r.push('');
+    for (const model of models) {
+        r.push(`## ${model.id}`);
+        r.push('');
+        const info = [];
+        if (model.pipeline_tag)
+            info.push(`**Task:** ${model.pipeline_tag}`);
+        if (model.library_name)
+            info.push(`**Library:** ${model.library_name}`);
+        if (model.downloads)
+            info.push(`**Downloads:** ${formatNumber(model.downloads)}`);
+        if (model.likes)
+            info.push(`**Likes:** ${model.likes.toString()}`);
+        if (model.trendingScore)
+            info.push(`**Trending Score:** ${model.trendingScore.toString()}`);
+        if (info.length > 0) {
+            r.push(info.join(' | '));
+            r.push('');
+        }
+        if (model.tags && model.tags.length > 0) {
+            r.push(`**Tags:** ${model.tags.slice(0, TAGS_TO_RETURN).join(', ')}`);
+            if (model.tags.length > TAGS_TO_RETURN) {
+                r.push(`*and ${(model.tags.length - TAGS_TO_RETURN).toString()} more...*`);
+            }
+            r.push('');
+        }
+        const status = [];
+        if (model.private)
+            status.push('🔐 Private');
+        if (status.length > 0) {
+            r.push(status.join(' | '));
+            r.push('');
+        }
+        if (model.createdAt) {
+            r.push(`**Created:** ${formatDate(model.createdAt)}`);
+        }
+        r.push(`**Link:** [https://hf.co/${model.id}](https://hf.co/${model.id})`);
+        r.push('');
+        r.push('---');
+        r.push('');
+    }
+    return r.join('\n');
+}
+//# sourceMappingURL=model-search.js.map

package/dist/model-search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-search.js","sourceRoot":"","sources":["../src/model-search.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAE1D,MAAM,CAAC,MAAM,cAAc,GAAG,EAAE,CAAC;AAEjC,MAAM,CAAC,MAAM,wBAAwB,GAAG;IACvC,IAAI,EAAE,cAAc;IACpB,WAAW,EACV,uDAAuD;QACvD,8GAA8G;QAC9G,8CAA8C;IAC/C,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC;QAChB,KAAK,EAAE,CAAC;aACN,MAAM,EAAE;aACR,QAAQ,EAAE;aACV,QAAQ,CACR,iIAAiI,CACjI;QACF,MAAM,EAAE,CAAC;aACP,MAAM,EAAE;aACR,QAAQ,EAAE;aACV,QAAQ,CAAC,wFAAwF,CAAC;QACpG,IAAI,EAAE,CAAC;aACL,MAAM,EAAE;aACR,QAAQ,EAAE;aACV,QAAQ,CAAC,kFAAkF,CAAC;QAC9F,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,sEAAsE,CAAC;QAC/G,IAAI,EAAE,CAAC;aACL,IAAI,CAAC,CAAC,eAAe,EAAE,WAAW,EAAE,OAAO,EAAE,WAAW,EAAE,cAAc,CAAC,CAAC;aAC1E,QAAQ,EAAE;aACV,QAAQ,CAAC,uEAAuE,CAAC;QACnF,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,qCAAqC,CAAC;KACxG,CAAC;IACF,WAAW,EAAE;QACZ,KAAK,EAAE,cAAc;QACrB,eAAe,EAAE,KAAK;QACtB,YAAY,EAAE,IAAI;QAClB,aAAa,EAAE,IAAI;KACnB;CACQ,CAAC;AAiCX,MAAM,OAAO,eAAgB,SAAQ,SAA2C;IAI/E,YAAY,OAAgB;QAC3B,KAAK,CAAC,mCAAmC,EAAE,OAAO,CAAC,CAAC;IACrD,CAAC;IAKD,KAAK,CAAC,gBAAgB,CAAC,MAAkC;QACxD,IAAI,CAAC;YAEJ,MAAM,SAAS,GAAmB,EAAE,CAAC;YAGrC,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBAClB,SAAS,CAAC,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC;YACjC,CAAC;YAGD,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBACnB,SAAS,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;YAClC,CAAC;YAGD,MAAM,OAAO,GAAG,EAAE,CAAC;YACnB,IAAI,MAAM,CAAC,IAAI;gBAAE,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC3C,IAAI,MAAM,CAAC,OAAO;gBAAE,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACjD,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxB,SAAS,CAAC,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACtC,CAAC;YAGD,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;gBACjB,SAAS,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;gBAC7B,SAAS,CAAC,SAAS,GAAG,IAAI,CAAC;YAC5B,CAAC;YAGD,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBAClB,SAAS,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;YAC3C,CAAC;YAGD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAmB,SAAS,CAAC,CAAC;YAE/D,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO,yCAAyC,CAAC;YAClD,CAAC;YAED,OAAO,mBAAmB,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC5C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC5B,MAAM,IAAI,KAAK,CAAC,gCAAgC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YAClE,CAAC;YACD,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC;IAKD,KAAK,CAAC,gBAAgB,CAAC,MAAc,EAAE,QAAgB,EAAE;QACxD,IAAI,CAAC;YACJ,MAAM,SAAS,GAAmB;gBACjC,MAAM,EAAE,MAAM;gBACd,KAAK,EAAE,KAAK,CAAC,QAAQ,EAAE;gBACvB,IAAI,EAAE,WAAW;gBACjB,SAAS,EAAE,IAAI;aACf,CAAC;YAGF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAmB,SAAS,CAAC,CAAC;YAE/D,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,OAAO,+BAA+B,MAAM,GAAG,CAAC;YACjD,CAAC;YAED,OAAO,mBAAmB,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC/C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC5B,MAAM,IAAI,KAAK,CAAC,gCAAgC,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;YAClE,CAAC;YACD,MAAM,KAAK,CAAC;QACb,CAAC;IACF,CAAC;CACD;AAGD,SAAS,mBAAmB,CAAC,MAAwB,EAAE,MAAkC;IACxF,MAAM,CAAC,GAAa,EAAE,CAAC;IAGvB,MAAM,WAAW,GAAG,EAAE,CAAC;IACvB,IAAI,MAAM,CAAC,KAAK;QAAE,WAAW,CAAC,IAAI,CAAC,UAAU,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC;IAC9D,IAAI,MAAM,CAAC,MAAM;QAAE,WAAW,CAAC,IAAI,CAAC,WAAW,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC;IACjE,IAAI,MAAM,CAAC,IAAI;QAAE,WAAW,CAAC,IAAI,CAAC,SAAS,MAAM,CAAC,IAAI,GAAG,CAAC,CAAC;IAC3D,IAAI,MAAM,CAAC,OAAO;QAAE,WAAW,CAAC,IAAI,CAAC,YAAY,MAAM,CAAC,OAAO,GAAG,CAAC,CAAC;IACpE,IAAI,MAAM,CAAC,IAAI;QAAE,WAAW,CAAC,IAAI,CAAC,aAAa,MAAM,CAAC,IAAI,eAAe,CAAC,CAAC;IAE3E,MAAM,UAAU,GAAG,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,aAAa,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAEvF,MAAM,UAAU,GACf,MAAM,CAAC,MAAM,KAAK,MAAM,CAAC,KAAK;QAC7B,CAAC,CAAC,iBAAiB,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,UAAU,UAAU,GAAG;QAClE,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,UAAU,UAAU,GAAG,CAAC;IAC7D,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACnB,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEX,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC5B,CAAC,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;QACzB,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAGX,MAAM,IAAI,GAAG,EAAE,CAAC;QAChB,IAAI,KAAK,CAAC,YAAY;YAAE,IAAI,CAAC,IAAI,CAAC,aAAa,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC;QACrE,IAAI,KAAK,CAAC,YAAY;YAAE,IAAI,CAAC,IAAI,CAAC,gBAAgB,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC;QACxE,IAAI,KAAK,CAAC,SAAS;YAAE,IAAI,CAAC,IAAI,CAAC,kBAAkB,YAAY,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QAClF,IAAI,KAAK,CAAC,KAAK;YAAE,IAAI,CAAC,IAAI,CAAC,cAAc,KAAK,CAAC,KAAK,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QACnE,IAAI,KAAK,CAAC,aAAa;YAAE,IAAI,CAAC,IAAI,CAAC,uBAAuB,KAAK,CAAC,aAAa,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QAE5F,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACrB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;YACzB,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACZ,CAAC;QAGD,IAAI,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzC,CAAC,CAAC,IAAI,CAAC,aAAa,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACtE,IAAI,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,cAAc,EAAE,CAAC;gBACxC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,GAAG,cAAc,CAAC,CAAC,QAAQ,EAAE,WAAW,CAAC,CAAC;YAC5E,CAAC;YACD,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACZ,CAAC;QAGD,MAAM,MAAM,GAAG,EAAE,CAAC;QAClB,IAAI,KAAK,CAAC,OAAO;YAAE,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC7C,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvB,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;YAC3B,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACZ,CAAC;QAGD,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YACrB,CAAC,CAAC,IAAI,CAAC,gBAAgB,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC;QACvD,CAAC;QAED,CAAC,CAAC,IAAI,CAAC,4BAA4B,KAAK,CAAC,EAAE,mBAAmB,KAAK,CAAC,EAAE,GAAG,CAAC,CAAC;QAC3E,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACX,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACd,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACZ,CAAC;IAED,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACrB,CAAC"}

package/dist/paper-search.d.ts

@@ -0,0 +1,58 @@
+import { z } from 'zod';
+import { HfApiCall } from './hf-api-call.js';
+export declare const DEFAULT_AUTHORS_TO_SHOW = 8;
+export declare const PAPER_SEARCH_TOOL_CONFIG: {
+    readonly name: "paper_search";
+    readonly description: string;
+    readonly schema: z.ZodObject<{
+        query: z.ZodString;
+        results_limit: z.ZodDefault<z.ZodOptional<z.ZodNumber>>;
+        concise_only: z.ZodDefault<z.ZodOptional<z.ZodBoolean>>;
+    }, "strip", z.ZodTypeAny, {
+        query: string;
+        results_limit: number;
+        concise_only: boolean;
+    }, {
+        query: string;
+        results_limit?: number | undefined;
+        concise_only?: boolean | undefined;
+    }>;
+    readonly annotations: {
+        readonly title: "Paper Search";
+        readonly destructiveHint: false;
+        readonly readOnlyHint: true;
+        readonly openWorldHint: true;
+    };
+};
+export interface Author {
+    name?: string;
+    user?: {
+        user: string;
+    };
+}
+interface Paper {
+    id: string;
+    authors?: Author[];
+    publishedAt?: string;
+    title?: string;
+    summary?: string;
+    upvotes?: number;
+    ai_keywords?: string[];
+    ai_summary?: string;
+}
+export interface PaperSearchResult {
+    paper: Paper;
+    numComments?: number;
+    isAuthorParticipating?: boolean;
+}
+interface PaperSearchParams {
+    q: string;
+}
+export declare class PaperSearchTool extends HfApiCall<PaperSearchParams, PaperSearchResult[]> {
+    constructor(hfToken?: string, apiUrl?: string);
+    search(query: string, limit?: number, conciseOnly?: boolean): Promise<string>;
+}
+export declare function published(publishedDate: string | undefined): string;
+export declare function authors(authors: Author[] | undefined, authorsToShow?: number): string;
+export {};
+//# sourceMappingURL=paper-search.d.ts.map

package/dist/paper-search.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"paper-search.d.ts","sourceRoot":"","sources":["../src/paper-search.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAgB7C,eAAO,MAAM,uBAAuB,IAAI,CAAC;AAGzC,eAAO,MAAM,wBAAwB;;;;;;;;;;;;;;;;;;;;;;CA2B3B,CAAC;AAEX,MAAM,WAAW,MAAM;IACtB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE;QACN,IAAI,EAAE,MAAM,CAAC;KACb,CAAC;CACF;AAED,UAAU,KAAK;IACd,EAAE,EAAE,MAAM,CAAC;IACX,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,WAAW,CAAC,EAAE,MAAM,EAAE,CAAC;IACvB,UAAU,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,iBAAiB;IACjC,KAAK,EAAE,KAAK,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,qBAAqB,CAAC,EAAE,OAAO,CAAC;CAChC;AAGD,UAAU,iBAAiB;IAC1B,CAAC,EAAE,MAAM,CAAC;CACV;AAKD,qBAAa,eAAgB,SAAQ,SAAS,CAAC,iBAAiB,EAAE,iBAAiB,EAAE,CAAC;gBAMzE,OAAO,CAAC,EAAE,MAAM,EAAE,MAAM,SAA6C;IAU3E,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,GAAE,MAA0B,EAAE,WAAW,GAAE,OAAe,GAAG,OAAO,CAAC,MAAM,CAAC;CAe7G;AAED,wBAAgB,SAAS,CAAC,aAAa,EAAE,MAAM,GAAG,SAAS,GAAG,MAAM,CAGnE;AAyDD,wBAAgB,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,SAAS,EAAE,aAAa,GAAE,MAAgC,GAAG,MAAM,CAa9G"}

package/dist/paper-search.js

@@ -0,0 +1,114 @@
+import { z } from 'zod';
+import { HfApiCall } from './hf-api-call.js';
+import { formatUnknownDate } from './utilities.js';
+export const DEFAULT_AUTHORS_TO_SHOW = 8;
+const RESULTS_TO_RETURN = 10;
+export const PAPER_SEARCH_TOOL_CONFIG = {
+    name: 'paper_search',
+    description: 'Find Machine Learning research papers on the Hugging Face hub. ' +
+        "Include 'Link to paper' When presenting the results. " +
+        'Consider whether tabulating results matches user intent.',
+    schema: z.object({
+        query: z
+            .string()
+            .min(3, 'Supply at least one search term')
+            .max(200, 'Query too long')
+            .describe('Semantic Search query'),
+        results_limit: z.number().optional().default(12).describe('Number of results to return'),
+        concise_only: z
+            .boolean()
+            .optional()
+            .default(false)
+            .describe('Return a 2 sentence summary of the abstract. Use for broad search terms which may return a lot of results. Check with User if unsure.'),
+    }),
+    annotations: {
+        title: 'Paper Search',
+        destructiveHint: false,
+        readOnlyHint: true,
+        openWorldHint: true,
+    },
+};
+export class PaperSearchTool extends HfApiCall {
+    constructor(hfToken, apiUrl = 'https://huggingface.co/api/papers/search') {
+        super(apiUrl, hfToken);
+    }
+    async search(query, limit = RESULTS_TO_RETURN, conciseOnly = false) {
+        try {
+            if (!query)
+                return 'No query';
+            const papers = await this.callApi({ q: query });
+            if (papers.length === 0)
+                return `No papers found for query '${query}'`;
+            return formatSearchResults(query, papers.slice(0, limit), papers.length, conciseOnly);
+        }
+        catch (error) {
+            if (error instanceof Error) {
+                throw new Error(`Failed to search for papers: ${error.message}`);
+            }
+            throw error;
+        }
+    }
+}
+export function published(publishedDate) {
+    const formatted = formatUnknownDate(publishedDate ?? '');
+    return formatted ? `Published on ${formatted}` : 'Publication date not available';
+}
+function formatSearchResults(query, papers, totalCount, conciseOnly = false) {
+    const r = [];
+    const showingText = papers.length < totalCount
+        ? `${totalCount} papers matched the query '${query}'. Here are the first ${papers.length} results.`
+        : `All ${papers.length} papers that matched the query '${query}'`;
+    r.push(showingText);
+    for (const result of papers) {
+        r.push('');
+        r.push('---');
+        const title = result.paper.title ?? `Paper ID ${result.paper.id}`;
+        r.push('');
+        r.push(`## ${title}`);
+        r.push('');
+        const publishedDate = result.paper.publishedAt
+            ? `Published on ${published(result.paper.publishedAt)}`
+            : 'Publication date not available';
+        r.push(publishedDate);
+        r.push(authors(result.paper.authors));
+        r.push('');
+        const useAiSummary = conciseOnly || !result.paper.summary;
+        const summaryText = useAiSummary ? result.paper.ai_summary : result.paper.summary;
+        const summaryHeader = useAiSummary ? '### AI Generated Summary' : '### Abstract';
+        r.push(summaryHeader);
+        r.push('');
+        r.push(summaryText ?? 'No summary available');
+        r.push('');
+        r.push(result.paper.ai_keywords ? `**AI Keywords**: ${result.paper.ai_keywords.join(', ')}` : '');
+        const upvotes = result.paper.upvotes && result.paper.upvotes > 0 ? `Upvoted ${result.paper.upvotes} times` : '';
+        if (result.numComments && result.numComments > 0) {
+            if (result.isAuthorParticipating)
+                r.push(`${upvotes}. The authors are participating in a discussion with ${result.numComments} comments.`);
+            else
+                r.push(`${upvotes}. There is a community discussion with ${result.numComments} comments.`);
+        }
+        else {
+            if ('' != upvotes)
+                r.push(upvotes);
+        }
+        r.push(`**Link to paper:** [https://hf.co/papers/${result.paper.id}](https://hf.co/papers/${result.paper.id})`);
+    }
+    r.push('');
+    r.push('---');
+    return r.join('\n');
+}
+export function authors(authors, authorsToShow = DEFAULT_AUTHORS_TO_SHOW) {
+    if (!authors || 0 === authors.length)
+        return '**Authors:** Not available';
+    const f = [];
+    for (const author of authors.slice(0, authorsToShow)) {
+        const profileLink = author.user?.user ? ` ([${author.user.user}](https://hf.co/${author.user.user}))` : '';
+        const authorName = author.name ?? 'Unknown';
+        f.push(`${authorName}${profileLink}`);
+    }
+    if (authors.length > authorsToShow) {
+        f.push(`and ${authors.length - authorsToShow} more.`);
+    }
+    return `**Authors:** ${f.join(', ')}`;
+}
+//# sourceMappingURL=paper-search.js.map