feat(plugins/anthropic): pull list of models from anthropic SDK

Author: CorieW

plugins/anthropic/src/index.ts

@@ -14,7 +14,7 @@
  * limitations under the License.
  */
 
-import { genkitPluginV2 } from 'genkit/plugin';
+import { genkitPluginV2, model, modelActionMetadata } from 'genkit/plugin';
 import Anthropic from '@anthropic-ai/sdk';
 
 import {
@@ -31,6 +31,7 @@ import {
 } from './claude.js';
 import { ModelAction } from 'genkit/model';
 import { ActionType } from 'genkit/registry';
+import { ActionMetadata } from 'genkit';
 
 export {
   claude4Sonnet,
@@ -48,6 +49,28 @@ export interface PluginOptions {
   cacheSystemPrompt?: boolean;
 }
 
+async function list(client: Anthropic): Promise<ActionMetadata[]> {
+  const clientModels = (await client.models.list()).data;
+
+  return clientModels
+    .map((modelInfo) => {
+      // Remove the final part if it's a date (8 digits)
+      const idParts = modelInfo.id.split('-');
+      const normalizedId = idParts[idParts.length - 1].match(/^\d{8}$/)
+        ? idParts.slice(0, -1).join('-')
+        : modelInfo.id;
+      const ref = SUPPORTED_CLAUDE_MODELS[normalizedId];
+      return ref
+        ? modelActionMetadata({
+            name: ref.name,
+            info: ref.info,
+            configSchema: ref.configSchema,
+          })
+        : null;
+    })
+    .filter(Boolean) as ActionMetadata[];
+}
+
 /**
  * This module provides an interface to the Anthropic AI models through the Genkit plugin system.
  * It allows users to interact with various Claude models by providing an API key and optional configuration.
@@ -92,6 +115,7 @@ export const anthropic = (options?: PluginOptions) => {
   const client = new Anthropic({ apiKey, defaultHeaders });
 
   const cachedActions: ModelAction[] = [];
+  let listActionsCache;
 
   return genkitPluginV2({
     name: 'anthropic',
@@ -115,12 +139,11 @@ export const anthropic = (options?: PluginOptions) => {
       }
       return undefined;
     },
-    list: () => {
-      return cachedActions.map((action) => ({
-        name: action.name,
-        type: 'model',
-      }));
-    },
+    list: async () => {
+      if (listActionsCache) return listActionsCache;
+      listActionsCache = await list(client);
+      return listActionsCache;
+    }
   });
 };