This commit is contained in:
@@ -380,6 +380,7 @@ describe("Generate E2E Tests", () => {
|
||||
describe("Google Vertex Provider (gemini-3-flash-preview)", () => {
|
||||
const vertexProject = process.env.GOOGLE_CLOUD_PROJECT || process.env.GCLOUD_PROJECT;
|
||||
const vertexLocation = process.env.GOOGLE_CLOUD_LOCATION;
|
||||
const vertexApiKey = process.env.GOOGLE_CLOUD_API_KEY;
|
||||
const isVertexConfigured = Boolean(vertexProject && vertexLocation);
|
||||
const vertexOptions = { project: vertexProject, location: vertexLocation } as const;
|
||||
const llm = getModel("google-vertex", "gemini-3-flash-preview");
|
||||
@@ -388,6 +389,10 @@ describe("Generate E2E Tests", () => {
|
||||
await basicTextGeneration(llm, vertexOptions);
|
||||
});
|
||||
|
||||
it.skipIf(!vertexApiKey)("should complete basic text generation with Vertex API key", { retry: 3 }, async () => {
|
||||
await basicTextGeneration(llm, { apiKey: vertexApiKey! });
|
||||
});
|
||||
|
||||
it.skipIf(!isVertexConfigured)("should handle tool calling", { retry: 3 }, async () => {
|
||||
await handleToolCall(llm, vertexOptions);
|
||||
});
|
||||
|
||||
Reference in New Issue
Block a user