Update recommend action with new model v2 shape

copilot-extensions · Sep 16, 2024 · 5fd650c · 5fd650c
1 parent 601cc27
commit 5fd650c
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/functions/recommend-model.ts b/src/functions/recommend-model.ts
@@ -32,12 +32,11 @@ export class recommendModel extends Tool {
       systemMessage.push(
         [
           `\t- Model Name: ${model.name}`,
-          `\t\tModel Version: ${model.model_version}`,
+          `\t\tModel Version: ${model.version}`,
           `\t\tPublisher: ${model.publisher}`,
-          `\t\tModel Family: ${model.model_family}`,
-          `\t\tModel Registry: ${model.model_registry}`,
+          `\t\tModel Registry: ${model.registryName}`,
           `\t\tLicense: ${model.license}`,
-          `\t\tTask: ${model.task}`,
+          `\t\tTask: ${model.inferenceTasks.join(", ")}`,
           `\t\tSummary: ${model.summary}`,
         ].join("\n")
       );