fix(js/plugins/vertexai): corrected plugin names

firebase · Oct 11, 2024 · 7eb223e · 7eb223e
1 parent a5b9408
commit 7eb223e
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 10 deletions.
diff --git 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
@@ -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
@@ -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: {