diff --git a/js/plugins/vertexai/src/model-garden/anthropic.ts b/js/plugins/vertexai/src/model-garden/anthropic.ts index 28bca6b59..82ce2ea70 100644 --- a/js/plugins/vertexai/src/model-garden/anthropic.ts +++ b/js/plugins/vertexai/src/model-garden/anthropic.ts @@ -51,7 +51,7 @@ export const AnthropicConfigSchema = GenerationCommonConfigSchema.extend({ }); export const claude35Sonnet = modelRef({ - name: 'vertexai/claude-3-5-sonnet', + name: 'vertexaiModelGarden/claude-3-5-sonnet', info: { label: 'Vertex AI Model Garden - Claude 3.5 Sonnet', versions: ['claude-3-5-sonnet@20240620'], @@ -67,7 +67,7 @@ export const claude35Sonnet = modelRef({ }); export const claude3Sonnet = modelRef({ - name: 'vertexai/claude-3-sonnet', + name: 'vertexaiModelGarden/claude-3-sonnet', info: { label: 'Vertex AI Model Garden - Claude 3 Sonnet', versions: ['claude-3-sonnet@20240229'], @@ -83,7 +83,7 @@ export const claude3Sonnet = modelRef({ }); export const claude3Haiku = modelRef({ - name: 'vertexai/claude-3-haiku', + name: 'vertexaiModelGarden/claude-3-haiku', info: { label: 'Vertex AI Model Garden - Claude 3 Haiku', versions: ['claude-3-haiku@20240307'], @@ -99,7 +99,7 @@ export const claude3Haiku = modelRef({ }); export const claude3Opus = modelRef({ - name: 'vertexai/claude-3-opus', + name: 'vertexaiModelGarden/claude-3-opus', info: { label: 'Vertex AI Model Garden - Claude 3 Opus', versions: ['claude-3-opus@20240229'], diff --git a/js/plugins/vertexai/src/model-garden/index.ts b/js/plugins/vertexai/src/model-garden/index.ts index 7922e6d83..c68e882c6 100644 --- a/js/plugins/vertexai/src/model-garden/index.ts +++ b/js/plugins/vertexai/src/model-garden/index.ts @@ -48,13 +48,11 @@ export interface PluginOptions extends BasePluginOptions { openAiBaseUrlTemplate?: string; } -const PLUGIN_NAME = 'vertexAiModelGarden'; - /** * Plugin for Vertex AI Model Garden */ export const vertexAIModelGarden: Plugin<[PluginOptions] | []> = genkitPlugin( - PLUGIN_NAME, + 'vertexaiModelGarden', async (options?: PluginOptions) => { // Authenticate with Google Cloud const authOptions = options?.googleAuth; diff --git a/js/plugins/vertexai/src/model-garden/model_garden.ts b/js/plugins/vertexai/src/model-garden/model_garden.ts index f9eb4bf95..0b850781d 100644 --- a/js/plugins/vertexai/src/model-garden/model_garden.ts +++ b/js/plugins/vertexai/src/model-garden/model_garden.ts @@ -28,7 +28,7 @@ export const ModelGardenModelConfigSchema = OpenAIConfigSchema.extend({ }); export const llama31 = modelRef({ - name: 'vertexai/llama-3.1', + name: 'vertexaiModelGarden/llama-3.1', info: { label: 'Llama 3.1', supports: { @@ -48,7 +48,7 @@ export const llama31 = modelRef({ }); export const llama32 = modelRef({ - name: 'vertexai/llama-3.2', + name: 'vertexaiModelGarden/llama-3.2', info: { label: 'Llama 3.2', supports: { @@ -68,7 +68,7 @@ export const llama32 = modelRef({ * @deprecated use `llama31` instead */ export const llama3 = modelRef({ - name: 'vertexai/llama3-405b', + name: 'vertexaiModelGarden/llama3-405b', info: { label: 'Llama 3.1 405b', supports: {