@llmindset/hf-mcp 0.1.19 → 0.1.20
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/model-detail.d.ts +1 -1
- package/dist/model-detail.d.ts.map +1 -1
- package/dist/model-detail.js +31 -2
- package/dist/model-detail.js.map +1 -1
- package/package.json +1 -1
- package/src/model-detail.ts +51 -2
package/dist/model-detail.d.ts
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
import { z } from 'zod';
|
|
2
2
|
export declare const MODEL_DETAIL_TOOL_CONFIG: {
|
|
3
3
|
readonly name: "model_details";
|
|
4
|
-
readonly description:
|
|
4
|
+
readonly description: string;
|
|
5
5
|
readonly schema: z.ZodObject<{
|
|
6
6
|
model_id: z.ZodString;
|
|
7
7
|
}, "strip", z.ZodTypeAny, {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"model-detail.d.ts","sourceRoot":"","sources":["../src/model-detail.ts"],"names":[],"mappings":"
|
|
1
|
+
{"version":3,"file":"model-detail.d.ts","sourceRoot":"","sources":["../src/model-detail.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAKxB,eAAO,MAAM,wBAAwB;;;;;;;;;;;;;;;;CAc3B,CAAC;AAEX,MAAM,MAAM,iBAAiB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,wBAAwB,CAAC,MAAM,CAAC,CAAC;AAgEhF,qBAAa,eAAe;IAC3B,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAS;gBAO1B,OAAO,CAAC,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,MAAM;IAWvC,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;CAiJlD"}
|
package/dist/model-detail.js
CHANGED
|
@@ -1,10 +1,11 @@
|
|
|
1
|
-
import { z } from 'zod';
|
|
2
1
|
import { modelInfo } from '@huggingface/hub';
|
|
2
|
+
import { z } from 'zod';
|
|
3
3
|
import { formatDate, formatNumber } from './utilities.js';
|
|
4
4
|
const SPACES_TO_INCLUDE = 12;
|
|
5
5
|
export const MODEL_DETAIL_TOOL_CONFIG = {
|
|
6
6
|
name: 'model_details',
|
|
7
|
-
description: 'Get detailed information about a
|
|
7
|
+
description: 'Get detailed information about a model from the Hugging Face Hub. Include relevant links ' +
|
|
8
|
+
'in result summaries.',
|
|
8
9
|
schema: z.object({
|
|
9
10
|
model_id: z.string().min(1, 'Model ID is required').describe('Model ID (e.g., microsoft/DialoGPT-large)'),
|
|
10
11
|
}),
|
|
@@ -34,6 +35,7 @@ export class ModelDetailTool {
|
|
|
34
35
|
'safetensors',
|
|
35
36
|
'cardData',
|
|
36
37
|
'spaces',
|
|
38
|
+
'inferenceProviderMapping',
|
|
37
39
|
];
|
|
38
40
|
const modelData = await modelInfo({
|
|
39
41
|
name: modelId,
|
|
@@ -41,6 +43,19 @@ export class ModelDetailTool {
|
|
|
41
43
|
...(this.accessToken && { credentials: { accessToken: this.accessToken } }),
|
|
42
44
|
...(this.hubUrl && { hubUrl: this.hubUrl }),
|
|
43
45
|
});
|
|
46
|
+
const inferenceProviders = Object.entries(modelData.inferenceProviderMapping)
|
|
47
|
+
.map(([provider, providerData]) => {
|
|
48
|
+
if (!providerData) {
|
|
49
|
+
return;
|
|
50
|
+
}
|
|
51
|
+
return {
|
|
52
|
+
provider,
|
|
53
|
+
status: providerData.status,
|
|
54
|
+
providerId: providerData.providerId,
|
|
55
|
+
task: providerData.task,
|
|
56
|
+
};
|
|
57
|
+
})
|
|
58
|
+
.filter((el) => !!el);
|
|
44
59
|
const modelDetails = {
|
|
45
60
|
id: modelId,
|
|
46
61
|
name: modelData.name,
|
|
@@ -56,6 +71,7 @@ export class ModelDetailTool {
|
|
|
56
71
|
downloadsAllTime: modelData.downloadsAllTime,
|
|
57
72
|
tags: modelData.tags,
|
|
58
73
|
},
|
|
74
|
+
...(inferenceProviders && inferenceProviders.length > 0 && { inferenceProviders }),
|
|
59
75
|
};
|
|
60
76
|
const technical = {};
|
|
61
77
|
if (modelData.config && typeof modelData.config === 'object') {
|
|
@@ -218,6 +234,19 @@ function formatModelDetails(model) {
|
|
|
218
234
|
}
|
|
219
235
|
r.push('');
|
|
220
236
|
}
|
|
237
|
+
if (model.inferenceProviders && model.inferenceProviders.length > 0) {
|
|
238
|
+
r.push('## Inference Providers');
|
|
239
|
+
for (const provider of model.inferenceProviders) {
|
|
240
|
+
let providerLine = `- **${provider.provider}**`;
|
|
241
|
+
if (provider.status) {
|
|
242
|
+
providerLine += ` (${provider.status})`;
|
|
243
|
+
}
|
|
244
|
+
r.push(providerLine);
|
|
245
|
+
}
|
|
246
|
+
r.push('');
|
|
247
|
+
r.push('Try this model in the [playground](https://hf.co/playground?modelId=' + model.name + ')');
|
|
248
|
+
r.push('');
|
|
249
|
+
}
|
|
221
250
|
r.push(`**Link:** [https://hf.co/${model.name}](https://hf.co/${model.name})`);
|
|
222
251
|
return r.join('\n');
|
|
223
252
|
}
|
package/dist/model-detail.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"model-detail.js","sourceRoot":"","sources":["../src/model-detail.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"model-detail.js","sourceRoot":"","sources":["../src/model-detail.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC7C,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,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,EACV,2FAA2F;QAC3F,sBAAsB;IACvB,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;AAkEX,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;gBACR,0BAA0B;aACjB,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,kBAAkB,GAAG,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,wBAAwB,CAAC;iBAC3E,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,YAAY,CAAC,EAAE,EAAE;gBACjC,IAAI,CAAC,YAAY,EAAE,CAAC;oBACnB,OAAO;gBACR,CAAC;gBACD,OAAO;oBACN,QAAQ;oBACR,MAAM,EAAE,YAAY,CAAC,MAAM;oBAC3B,UAAU,EAAE,YAAY,CAAC,UAAU;oBACnC,IAAI,EAAE,YAAY,CAAC,IAAI;iBACvB,CAAC;YACH,CAAC,CAAC;iBACD,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YAEvB,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;gBAGD,GAAG,CAAC,kBAAkB,IAAI,kBAAkB,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,kBAAkB,EAAE,CAAC;aAClF,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,IAAI,KAAK,CAAC,kBAAkB,IAAI,KAAK,CAAC,kBAAkB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACrE,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QAEjC,KAAK,MAAM,QAAQ,IAAI,KAAK,CAAC,kBAAkB,EAAE,CAAC;YACjD,IAAI,YAAY,GAAG,OAAO,QAAQ,CAAC,QAAQ,IAAI,CAAC;YAGhD,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;gBACrB,YAAY,IAAI,KAAK,QAAQ,CAAC,MAAM,GAAG,CAAC;YACzC,CAAC;YAED,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACtB,CAAC;QAED,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACX,CAAC,CAAC,IAAI,CAAC,sEAAsE,GAAG,KAAK,CAAC,IAAI,GAAG,GAAG,CAAC,CAAC;QAClG,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/package.json
CHANGED
package/src/model-detail.ts
CHANGED
|
@@ -1,12 +1,14 @@
|
|
|
1
|
-
import { z } from 'zod';
|
|
2
1
|
import { modelInfo } from '@huggingface/hub';
|
|
2
|
+
import { z } from 'zod';
|
|
3
3
|
import { formatDate, formatNumber } from './utilities.js';
|
|
4
4
|
|
|
5
5
|
const SPACES_TO_INCLUDE = 12;
|
|
6
6
|
// Model Detail Tool Configuration
|
|
7
7
|
export const MODEL_DETAIL_TOOL_CONFIG = {
|
|
8
8
|
name: 'model_details',
|
|
9
|
-
description:
|
|
9
|
+
description:
|
|
10
|
+
'Get detailed information about a model from the Hugging Face Hub. Include relevant links ' +
|
|
11
|
+
'in result summaries.',
|
|
10
12
|
schema: z.object({
|
|
11
13
|
model_id: z.string().min(1, 'Model ID is required').describe('Model ID (e.g., microsoft/DialoGPT-large)'),
|
|
12
14
|
}),
|
|
@@ -58,11 +60,20 @@ interface ModelMetadata {
|
|
|
58
60
|
fineTunedFrom?: string;
|
|
59
61
|
}
|
|
60
62
|
|
|
63
|
+
// Inference providers information
|
|
64
|
+
interface InferenceProvider {
|
|
65
|
+
provider: string;
|
|
66
|
+
status?: string;
|
|
67
|
+
providerId?: string;
|
|
68
|
+
task?: string;
|
|
69
|
+
}
|
|
70
|
+
|
|
61
71
|
// Complete model information structure
|
|
62
72
|
interface ModelInformation extends ModelBasicInfo {
|
|
63
73
|
extended?: ModelExtendedInfo;
|
|
64
74
|
technical?: ModelTechnicalDetails;
|
|
65
75
|
metadata?: ModelMetadata;
|
|
76
|
+
inferenceProviders?: InferenceProvider[];
|
|
66
77
|
spaces?: Array<{
|
|
67
78
|
id: string;
|
|
68
79
|
name: string;
|
|
@@ -106,6 +117,7 @@ export class ModelDetailTool {
|
|
|
106
117
|
'safetensors',
|
|
107
118
|
'cardData',
|
|
108
119
|
'spaces',
|
|
120
|
+
'inferenceProviderMapping',
|
|
109
121
|
] as const;
|
|
110
122
|
|
|
111
123
|
const modelData = await modelInfo<(typeof additionalFields)[number]>({
|
|
@@ -115,6 +127,20 @@ export class ModelDetailTool {
|
|
|
115
127
|
...(this.hubUrl && { hubUrl: this.hubUrl }),
|
|
116
128
|
});
|
|
117
129
|
|
|
130
|
+
// Extract inference providers from modelData if available
|
|
131
|
+
const inferenceProviders = Object.entries(modelData.inferenceProviderMapping)
|
|
132
|
+
.map(([provider, providerData]) => {
|
|
133
|
+
if (!providerData) {
|
|
134
|
+
return;
|
|
135
|
+
}
|
|
136
|
+
return {
|
|
137
|
+
provider,
|
|
138
|
+
status: providerData.status,
|
|
139
|
+
providerId: providerData.providerId,
|
|
140
|
+
task: providerData.task,
|
|
141
|
+
};
|
|
142
|
+
})
|
|
143
|
+
.filter((el) => !!el);
|
|
118
144
|
// Build the structured model information
|
|
119
145
|
const modelDetails: ModelInformation = {
|
|
120
146
|
// Basic info (required fields)
|
|
@@ -134,6 +160,9 @@ export class ModelDetailTool {
|
|
|
134
160
|
downloadsAllTime: modelData.downloadsAllTime,
|
|
135
161
|
tags: modelData.tags,
|
|
136
162
|
},
|
|
163
|
+
|
|
164
|
+
// Inference providers (if available)
|
|
165
|
+
...(inferenceProviders && inferenceProviders.length > 0 && { inferenceProviders }),
|
|
137
166
|
};
|
|
138
167
|
|
|
139
168
|
// Technical details (requires validation)
|
|
@@ -352,6 +381,26 @@ function formatModelDetails(model: ModelInformation): string {
|
|
|
352
381
|
r.push('');
|
|
353
382
|
}
|
|
354
383
|
|
|
384
|
+
// Inference Providers - if available
|
|
385
|
+
if (model.inferenceProviders && model.inferenceProviders.length > 0) {
|
|
386
|
+
r.push('## Inference Providers');
|
|
387
|
+
|
|
388
|
+
for (const provider of model.inferenceProviders) {
|
|
389
|
+
let providerLine = `- **${provider.provider}**`;
|
|
390
|
+
|
|
391
|
+
// Add status if available
|
|
392
|
+
if (provider.status) {
|
|
393
|
+
providerLine += ` (${provider.status})`;
|
|
394
|
+
}
|
|
395
|
+
|
|
396
|
+
r.push(providerLine);
|
|
397
|
+
}
|
|
398
|
+
|
|
399
|
+
r.push('');
|
|
400
|
+
r.push('Try this model in the [playground](https://hf.co/playground?modelId=' + model.name + ')');
|
|
401
|
+
r.push('');
|
|
402
|
+
}
|
|
403
|
+
|
|
355
404
|
// Link is reliable - based on model name which is required
|
|
356
405
|
r.push(`**Link:** [https://hf.co/${model.name}](https://hf.co/${model.name})`);
|
|
357
406
|
|