@economic/agents 2.7.1 → 2.7.2

package/README.md

@@ -83,14 +83,13 @@ const { anthropic, gemini } = createAiGatewayVertexProviders({
   cloudflareAiGatewayId: env.AI_GATEWAY_ID,
   cloudflareApiToken: env.CLOUDFLARE_API_TOKEN,
   googleCloudProjectId: env.GOOGLE_CLOUD_PROJECT_ID,
-  // Included in cf-aig-metadata on every request, including background calls.
-  metadata: { agentId: "support" },
 });
 ```
 
-Set a stable `agentId` when multiple agents may use the same gateway. AI Gateway
-stores this custom metadata with each request, allowing cost queries to filter
-by agent without copying cost data into another store.
+When these providers are used inside an SDK agent, the SDK automatically adds
+the entry-point agent class as `agentId` in `cf-aig-metadata`. This includes
+requests made by child chat/facet agents and allows a shared gateway's usage to
+be grouped by agent without application configuration or duplicated analytics.
 
 ## Quick start: an agent
 

package/dist/providers.d.mts

@@ -15,7 +15,7 @@ interface AnthropicVertexProviderOptions {
   googleCloudProjectId: string;
   location?: string;
   anthropicVersion?: string;
-  /** Metadata included on every request sent through AI Gateway. */
+  /** Custom metadata included on every request. `agentId` is injected by the SDK. */
   metadata?: AiGatewayMetadata;
 }
 declare function createAnthropicProvider(options: AnthropicVertexProviderOptions): (modelId: AnthropicVertexModelId, metadata?: AiGatewayMetadata) => AnthropicMessagesLanguageModel;
@@ -28,7 +28,7 @@ interface GeminiProviderOptions {
   cloudflareApiToken: string;
   googleCloudProjectId: string;
   location?: string;
-  /** Metadata included on every request sent through AI Gateway. */
+  /** Custom metadata included on every request. `agentId` is injected by the SDK. */
   metadata?: AiGatewayMetadata;
 }
 declare function createGeminiProvider(options: GeminiProviderOptions, metadata?: AiGatewayMetadata): GoogleGenerativeAIProvider;

package/dist/providers.mjs

@@ -1,13 +1,33 @@
+import { getCurrentAgent } from "agents";
 import { AnthropicMessagesLanguageModel } from "@ai-sdk/anthropic/internal";
 import { createGoogleGenerativeAI } from "@ai-sdk/google";
 //#region src/providers/metadata.ts
+function getCurrentAgentId() {
+	const agent = getCurrentAgent().agent;
+	if (!agent) return void 0;
+	return agent.selfPath[0]?.className ?? agent.constructor.name;
+}
 function createAiGatewayHeaders(cloudflareApiToken, defaultMetadata, requestMetadata) {
 	const headers = { "cf-aig-authorization": `Bearer ${cloudflareApiToken}` };
 	const metadata = { ...defaultMetadata };
 	for (const [key, value] of Object.entries(requestMetadata ?? {})) if (!(key in metadata)) metadata[key] = value;
+	const agentId = getCurrentAgentId();
+	if (agentId) metadata.agentId = agentId;
 	if (Object.values(metadata).some((value) => value !== void 0)) headers["cf-aig-metadata"] = JSON.stringify(metadata);
 	return headers;
 }
+/** Adds AI Gateway authentication and SDK-owned agent metadata at request time. */
+function createAiGatewayFetch(cloudflareApiToken, defaultMetadata, requestMetadata) {
+	return (input, init) => {
+		const headers = new Headers(input instanceof Request ? input.headers : void 0);
+		new Headers(init?.headers).forEach((value, key) => headers.set(key, value));
+		for (const [key, value] of Object.entries(createAiGatewayHeaders(cloudflareApiToken, defaultMetadata, requestMetadata))) headers.set(key, value);
+		return globalThis.fetch(input, {
+			...init,
+			headers
+		});
+	};
+}
 //#endregion
 //#region src/providers/anthropic.ts
 function createAnthropicProvider(options) {
@@ -17,7 +37,8 @@ function createAnthropicProvider(options) {
 		return new AnthropicMessagesLanguageModel(modelId, {
 			provider: "vertex.anthropic.messages",
 			baseURL: anthropicBaseURL,
-			headers: createAiGatewayHeaders(cloudflareApiToken, defaultMetadata, metadata),
+			headers: {},
+			fetch: createAiGatewayFetch(cloudflareApiToken, defaultMetadata, metadata),
 			buildRequestUrl: (baseURL, isStreaming) => `${baseURL}${modelId}:${isStreaming ? "streamRawPredict" : "rawPredict"}`,
 			transformRequestBody: (args) => {
 				const { model, ...rest } = args;
@@ -39,7 +60,7 @@ function createGeminiProvider(options, metadata) {
 	return createGoogleGenerativeAI({
 		apiKey: "unused",
 		baseURL: `https://gateway.ai.cloudflare.com/v1/${cloudflareAccountId}/${cloudflareAiGatewayId}/google-vertex-ai/v1/projects/${googleCloudProjectId}/locations/${location}/publishers/google/models`,
-		headers: createAiGatewayHeaders(cloudflareApiToken, defaultMetadata, metadata),
+		fetch: createAiGatewayFetch(cloudflareApiToken, defaultMetadata, metadata),
 		name: "gateway.google.vertex-ai"
 	});
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@economic/agents",
-  "version": "2.7.1",
+  "version": "2.7.2",
   "description": "A starter for creating a TypeScript package.",
   "license": "MIT",
   "files": [