Resolve mapping IDs automatically to new HF endpoint

src/hf.ts

@@ -1,11 +1,13 @@
 interface ModelMappingItem {
+  _id?: string;
   task: string;
   hfModel: string;
   providerModel: string;
   status?: 'live' | 'staging';
 }
 
 interface TagFilterMappingItem {
+  _id?: string;
   task: string;
   providerModel: string;
   status?: 'live' | 'staging';
@@ -122,6 +124,32 @@ class HFInferenceProviderClient {
     );
   }
 
+  async getMappingIdByHfModel(hfModel: string): Promise<string | undefined> {
+    const mappings = await this.listMappingItems();
+    for (const taskMappings of Object.values(mappings)) {
+      const mapping = taskMappings[hfModel];
+      if (mapping && mapping._id) {
+        return mapping._id;
+      }
+    }
+    return undefined;
+  }
+
+  async createHfModelToMappingIdMap(): Promise<Map<string, string>> {
+    const mappings = await this.listMappingItems();
+    const map = new Map<string, string>();
+
+    for (const taskMappings of Object.values(mappings)) {
+      for (const [hfModel, mapping] of Object.entries(taskMappings)) {
+        if (mapping._id) {
+          map.set(hfModel, mapping._id);
+        }
+      }
+    }
+
+    return map;
+  }
+
   async getMappingsByProvider(provider: string): Promise<Record<string, Record<string, MappingItem>>> {
     const url = `${this.baseUrl}/api/partners/${provider}/models`;
     return this.request<Record<string, Record<string, MappingItem>>>(url, {

src/index.ts

@@ -81,6 +81,8 @@ if (unsupportedModels.length > 0) {
 
 // Use only supported models for mapping operations
 const existingHFModelIds = await hf.listMappingIds();
+// Create a lookup map for efficient mapping ID retrieval during status updates
+const hfModelToMappingIdMap = await hf.createHfModelToMappingIdMap();
 
 console.log("\n\nExisting HF model IDs:");
 console.log(existingHFModelIds);
@@ -102,10 +104,23 @@ if (existingMappings.length > 0) {
 	console.log(`\n\nUpdating statuses for ${existingMappings.length} existing mappings:`);
     for (const model of existingMappings) {
         console.log(`${model.hfModel} - ${model.status}`);
-        await hf.updateMappingItemStatus({
-            hfModel: model.hfModel,
-            status: model.status,
-        });
+        const mappingId = hfModelToMappingIdMap.get(model.hfModel);
+        if (mappingId) {
+            try {
+                await hf.updateMappingItemStatus({
+                    mappingId: mappingId,
+                    status: model.status,
+                });
+            } catch (error) {
+                if (error instanceof Error && error.message.includes('does not support task')) {
+                    console.log(`Skipping ${model.hfModel}: ${model.task} task not supported for status updates`);
+                } else {
+                    throw error;
+                }
+            }
+        } else {
+            console.error(`Could not find mapping ID for ${model.hfModel}`);
+        }
     }
 } else {
 	console.log("\n\nNo existing mappings to update.");