@inkeep/agents-core 0.59.4 → 0.60.0

package/dist/utils/usage-cost-middleware.js

@@ -0,0 +1,60 @@
+import { SPAN_KEYS } from "../constants/otel-attributes.js";
+import { getLogger } from "./logger.js";
+import { trace } from "@opentelemetry/api";
+
+//#region src/utils/usage-cost-middleware.ts
+const logger = getLogger("usage-cost-middleware");
+function extractUsageTokens(usage) {
+	return {
+		inputTokens: typeof usage?.inputTokens === "object" ? usage.inputTokens.total ?? 0 : usage?.inputTokens ?? 0,
+		outputTokens: typeof usage?.outputTokens === "object" ? usage.outputTokens.total ?? 0 : usage?.outputTokens ?? 0,
+		reasoningTokens: typeof usage?.outputTokens === "object" ? usage.outputTokens.reasoning : void 0,
+		cachedReadTokens: typeof usage?.inputTokens === "object" ? usage.inputTokens.cacheRead : void 0,
+		cachedWriteTokens: typeof usage?.inputTokens === "object" ? usage.inputTokens.cacheWrite : void 0
+	};
+}
+function extractGatewayCost(providerMetadata) {
+	const gw = providerMetadata?.gateway;
+	if (!gw) return 0;
+	const cost = parseFloat(gw.cost);
+	if (!Number.isNaN(cost) && cost > 0) return cost;
+	const marketCost = parseFloat(gw.marketCost);
+	if (!Number.isNaN(marketCost) && marketCost > 0) return marketCost;
+	return 0;
+}
+function setGatewayCostOnSpan(providerMetadata) {
+	const activeSpan = trace.getActiveSpan();
+	if (!activeSpan) return;
+	const cost = extractGatewayCost(providerMetadata);
+	activeSpan.setAttribute(SPAN_KEYS.GEN_AI_COST_ESTIMATED_USD, cost);
+	if (providerMetadata?.gateway && cost === 0) logger.warn({ gateway: providerMetadata.gateway }, "Routed through gateway but no cost data in response");
+}
+const gatewayCostMiddleware = {
+	specificationVersion: "v3",
+	async wrapGenerate({ doGenerate }) {
+		const result = await doGenerate();
+		try {
+			setGatewayCostOnSpan(result.providerMetadata);
+		} catch (error) {
+			logger.warn({ error }, "Failed to extract gateway cost in wrapGenerate");
+		}
+		return result;
+	},
+	async wrapStream({ doStream }) {
+		const { stream, ...rest } = await doStream();
+		return {
+			stream: stream.pipeThrough(new TransformStream({ transform(chunk, controller) {
+				controller.enqueue(chunk);
+				if (chunk.type === "finish") try {
+					setGatewayCostOnSpan(chunk.providerMetadata);
+				} catch (error) {
+					logger.warn({ error }, "Failed to extract gateway cost in wrapStream");
+				}
+			} })),
+			...rest
+		};
+	}
+};
+
+//#endregion
+export { extractUsageTokens, gatewayCostMiddleware };

package/dist/utils/usage-tracker.d.ts

@@ -0,0 +1,2 @@
+import { GenerationType } from "../db/runtime/runtime-schema.js";
+export { type GenerationType };

package/dist/utils/usage-tracker.js

@@ -0,0 +1 @@
+export {  };

package/dist/validation/drizzle-schema-helpers.d.ts

@@ -1,10 +1,10 @@
 import { z } from "@hono/zod-openapi";
-import * as drizzle_zod15 from "drizzle-zod";
+import * as drizzle_zod0 from "drizzle-zod";
 import { AnySQLiteTable } from "drizzle-orm/sqlite-core";
 
 //#region src/validation/drizzle-schema-helpers.d.ts
-declare function createSelectSchemaWithModifiers<T extends AnySQLiteTable>(table: T, overrides?: Partial<Record<keyof T['_']['columns'], (schema: z.ZodTypeAny) => z.ZodTypeAny>>): drizzle_zod15.BuildSchema<"select", T["_"]["columns"], drizzle_zod15.BuildRefine<T["_"]["columns"], undefined>, undefined>;
-declare function createInsertSchemaWithModifiers<T extends AnySQLiteTable>(table: T, overrides?: Partial<Record<keyof T['_']['columns'], (schema: z.ZodTypeAny) => z.ZodTypeAny>>): drizzle_zod15.BuildSchema<"insert", T["_"]["columns"], drizzle_zod15.BuildRefine<Pick<T["_"]["columns"], keyof T["$inferInsert"]>, undefined>, undefined>;
+declare function createSelectSchemaWithModifiers<T extends AnySQLiteTable>(table: T, overrides?: Partial<Record<keyof T['_']['columns'], (schema: z.ZodTypeAny) => z.ZodTypeAny>>): drizzle_zod0.BuildSchema<"select", T["_"]["columns"], drizzle_zod0.BuildRefine<T["_"]["columns"], undefined>, undefined>;
+declare function createInsertSchemaWithModifiers<T extends AnySQLiteTable>(table: T, overrides?: Partial<Record<keyof T['_']['columns'], (schema: z.ZodTypeAny) => z.ZodTypeAny>>): drizzle_zod0.BuildSchema<"insert", T["_"]["columns"], drizzle_zod0.BuildRefine<Pick<T["_"]["columns"], keyof T["$inferInsert"]>, undefined>, undefined>;
 declare const createSelectSchema: typeof createSelectSchemaWithModifiers;
 declare const createInsertSchema: typeof createInsertSchemaWithModifiers;
 /**