Merge pull request #2351 from asimurka/fix_providers

LCORE-3414: Fix Vertex AI and Bedrock e2e model identifiers

Author: tisnik

.github/workflows/e2e_tests_providers.yaml

@@ -15,18 +15,18 @@ jobs:
         mode: ["server", "library"]
         environment: ["azure", "vertexai", "watsonx", "bedrock"]
         # Expected default LLM for Behave (matches tests/e2e/configs/run-<env>.yaml).
-        # | environment | model_id                            | provider_id   |
-        # |-------------|-------------------------------------|---------------|
-        # | azure       | gpt-4o-mini                         | azure         |
-        # | vertexai    | google/gemini-2.5-flash             | google-vertex |
-        # | watsonx     | meta-llama/llama-3-3-70b-instruct   | watsonx       |
-        # | bedrock     | deepseek.v3-v1:0                    | aws-bedrock   |
+        # | environment | model_id                                   | provider_id   |
+        # |-------------|--------------------------------------------| --------------|
+        # | azure       | gpt-4o-mini                                | azure         |
+        # | vertexai    | publishers/google/models/gemini-2.5-flash  | google-vertex |
+        # | watsonx     | meta-llama/llama-3-3-70b-instruct          | watsonx       |
+        # | bedrock     | deepseek.v3-v1:0                           | aws-bedrock   |
         include:
           - environment: azure
             e2e_default_model: gpt-4o-mini
             e2e_default_provider: azure
           - environment: vertexai
-            e2e_default_model: google/gemini-2.5-flash
+            e2e_default_model: publishers/google/models/gemini-2.5-flash
             e2e_default_provider: google-vertex
           - environment: watsonx
             e2e_default_model: meta-llama/llama-3-3-70b-instruct

examples/vertexai-run.yaml

@@ -19,7 +19,7 @@ providers:
     config:
       project: ${env.VERTEX_AI_PROJECT}
       location: ${env.VERTEX_AI_LOCATION}
-      allowed_models: ["google/gemini-2.5-flash"]
+      allowed_models: ["publishers/google/models/gemini-2.5-flash"]
   - provider_id: openai
     provider_type: remote::openai
     config:
@@ -97,10 +97,10 @@ storage:
       backend: sql_default
 registered_resources:
   models:
-  - model_id: google/gemini-2.5-flash
+  - model_id: publishers/google/models/gemini-2.5-flash
     provider_id: google-vertex
     model_type: llm
-    provider_model_id: google/gemini-2.5-flash
+    provider_model_id: publishers/google/models/gemini-2.5-flash
   - model_id: all-mpnet-base-v2
     model_type: embedding
     provider_id: sentence-transformers

tests/e2e/configs/run-bedrock.yaml

@@ -97,7 +97,7 @@ storage:
       backend: sql_default
 registered_resources:
   models:
-  - model_id: custom-bedrock-model
+  - model_id: deepseek.v3-v1:0
     model_type: llm
     provider_id: aws-bedrock
     provider_model_id: deepseek.v3-v1:0

tests/e2e/configs/run-vertexai.yaml

@@ -19,7 +19,7 @@ providers:
     config:
       project: ${env.VERTEX_AI_PROJECT}
       location: ${env.VERTEX_AI_LOCATION}
-      allowed_models: ["google/gemini-2.5-flash"]
+      allowed_models: ["publishers/google/models/gemini-2.5-flash"]
   - provider_id: openai
     provider_type: remote::openai
     config:
@@ -98,10 +98,10 @@ storage:
       backend: sql_default
 registered_resources:
   models:
-  - model_id: google/gemini-2.5-flash
+  - model_id: publishers/google/models/gemini-2.5-flash
     provider_id: google-vertex
     model_type: llm
-    provider_model_id: google/gemini-2.5-flash
+    provider_model_id: publishers/google/models/gemini-2.5-flash
   - model_id: all-mpnet-base-v2
     model_type: embedding
     provider_id: sentence-transformers