diff --git a/packages/ai/src/providers/google-vertex-messages.ts b/packages/ai/src/providers/google-vertex-messages.ts index c66c10ce6a1..33101cc0389 100644 --- a/packages/ai/src/providers/google-vertex-messages.ts +++ b/packages/ai/src/providers/google-vertex-messages.ts @@ -37,7 +37,7 @@ export type Settings = ProviderPackage.Settings & const route = Route.make({ id: "google-vertex-messages", provider: id, - providerMetadataKey: "anthropic", + providerMetadataKey: "vertex", protocol: Protocol.make({ id: AnthropicMessages.protocol.id, body: { diff --git a/packages/ai/src/providers/groq.ts b/packages/ai/src/providers/groq.ts index 080e78a5b7b..f30a3f5793d 100644 --- a/packages/ai/src/providers/groq.ts +++ b/packages/ai/src/providers/groq.ts @@ -71,7 +71,7 @@ export const protocol = Protocol.make({ export const route = Route.make({ id: "groq-chat", provider: id, - providerMetadataKey: "openai", + providerMetadataKey: "groq", protocol, endpoint: Endpoint.path("/chat/completions", { baseURL }), framing: OpenAIChat.framing, diff --git a/packages/ai/test/provider/google-vertex.test.ts b/packages/ai/test/provider/google-vertex.test.ts index 85d3335fba8..eb0e22aebc7 100644 --- a/packages/ai/test/provider/google-vertex.test.ts +++ b/packages/ai/test/provider/google-vertex.test.ts @@ -302,6 +302,7 @@ describe("Google Vertex providers", () => { expect(model.provider).toBe("google-vertex") expect(response.text).toBe("Hello.") + expect(response.usage?.providerMetadata).toHaveProperty("vertex") }), ) diff --git a/packages/ai/test/provider/groq.recorded.test.ts b/packages/ai/test/provider/groq.recorded.test.ts index 01d196751f1..93fd28931ad 100644 --- a/packages/ai/test/provider/groq.recorded.test.ts +++ b/packages/ai/test/provider/groq.recorded.test.ts @@ -171,6 +171,7 @@ describe("Groq recorded", () => { expect(response.text).not.toContain("") expect(response.reasoning.length).toBeGreaterThan(0) expect(response.events.some(LLMEvent.is.reasoningDelta)).toBe(true) + expect(response.usage?.providerMetadata).toHaveProperty("groq") expectUsage(response) }), 60_000, diff --git a/packages/ai/test/provider/native-providers.test.ts b/packages/ai/test/provider/native-providers.test.ts index ae90fc062f8..156dc99c3c9 100644 --- a/packages/ai/test/provider/native-providers.test.ts +++ b/packages/ai/test/provider/native-providers.test.ts @@ -48,7 +48,7 @@ describe("native OpenAI-compatible providers", () => { [GoogleVertex.configure(vertex).model("model"), "vertex"], [GoogleVertexChat.configure(vertex).model("model"), "vertex"], [GoogleVertexResponses.configure(vertex).model("model"), "vertex"], - [GoogleVertexMessages.configure(vertex).model("model"), "anthropic"], + [GoogleVertexMessages.configure(vertex).model("model"), "vertex"], [Anthropic.configure({ apiKey: "test" }).model("model"), "anthropic"], [ AnthropicCompatible.configure({ baseURL: "https://example.test/v1", provider: "minimax" }).model("model"), @@ -64,6 +64,7 @@ describe("native OpenAI-compatible providers", () => { [DeepSeek.configure({ apiKey: "test" }).model("model"), "deepseek"], [Fireworks.configure({ apiKey: "test" }).model("model"), "fireworks"], [DeepInfra.configure({ apiKey: "test" }).model("model"), "deepinfra"], + [Groq.configure({ apiKey: "test" }).model("model"), "groq"], [TogetherAI.configure({ apiKey: "test" }).model("model"), "togetherai"], [CloudflareAIGateway.configure({ accountId: "account" }).model("model"), "cloudflare-ai-gateway"], [CloudflareWorkersAI.configure({ accountId: "account" }).model("model"), "cloudflare-workers-ai"],