@apo-ai/sdk 0.8.0 → 0.9.1

package/dist/{chunk-HLRXQAZ5.js → chunk-HRYXLRXJ.js}

@@ -19,7 +19,7 @@ function extractTokenUsage(attrs) {
   return out;
 }
 function toNum(val) {
-  if (typeof val === "number") return val;
+  if (typeof val === "number") return Number.isFinite(val) ? val : void 0;
   if (typeof val === "string") {
     const n = Number(val);
     return Number.isFinite(n) ? n : void 0;
@@ -39,4 +39,4 @@ export {
   extractTokenUsage,
   extractTokenCounts
 };
-//# sourceMappingURL=chunk-HLRXQAZ5.js.map
+//# sourceMappingURL=chunk-HRYXLRXJ.js.map

package/dist/{chunk-HLRXQAZ5.js.map → chunk-HRYXLRXJ.js.map}

@@ -1 +1 @@
-{"version":3,"sources":["../src/agent-task/integrations/token-usage.ts"],"sourcesContent":["/**\n * Shared token-usage extraction for the tracing integrations.\n *\n * Both the `createApoTracer` path (`ai-sdk.ts`) and the OTel-native path\n * (`otel-translate.ts` → `otel-processor.ts`) must extract token counts from\n * AI SDK span attributes using the same logic. This helper is the single\n * source of truth — it prevents the two paths from drifting apart (the bug\n * where one path reported tokens and the other silently dropped them).\n *\n * The AI SDK emits usage under `ai.usage.promptTokens` /\n * `ai.usage.completionTokens`. Standard OTel GenAI conventions use\n * `gen_ai.usage.input_tokens` / `gen_ai.usage.output_tokens`. Both are\n * recognized; standard `gen_ai.usage.*` takes precedence when both are present.\n *\n * @internal\n */\n\ntype SpanAttributes = Record<string, unknown>;\n\n/**\n * Extract `prompt_tokens` / `completion_tokens` / `reasoning_tokens` from a\n * span's attributes. Returns an object shaped to spread directly into\n * `trace.endSpan()` params. Omits keys when the corresponding attribute is\n * absent or non-numeric.\n *\n * Recognizes all known attribute-name conventions:\n *   - Vercel AI SDK: `ai.usage.promptTokens` / `ai.usage.completionTokens` /\n *     `ai.usage.reasoningTokens`\n *   - OTel GenAI: `gen_ai.usage.input_tokens` / `gen_ai.usage.output_tokens` /\n *     `gen_ai.usage.reasoning.output_tokens`\n *   - OTel GenAI (alt): `gen_ai.usage.prompt_tokens` / `gen_ai.usage.completion_tokens`\n *   - OpenInference: `llm.token_count.completion_details.reasoning`\n *\n * `gen_ai.usage.*` takes precedence over `ai.usage.*` when both are present.\n * Reasoning is omitted when unreported — a provider that never sends the\n * dimension must surface as unknown downstream, never as zero.\n */\nexport function extractTokenUsage(attrs: SpanAttributes): {\n  prompt_tokens?: number;\n  completion_tokens?: number;\n  reasoning_tokens?: number;\n} {\n  const prompt_tokens = toNum(\n    attrs[\"gen_ai.usage.input_tokens\"] ??\n      attrs[\"gen_ai.usage.prompt_tokens\"] ??\n      attrs[\"ai.usage.promptTokens\"],\n  );\n  const completion_tokens = toNum(\n    attrs[\"gen_ai.usage.output_tokens\"] ??\n      attrs[\"gen_ai.usage.completion_tokens\"] ??\n      attrs[\"ai.usage.completionTokens\"],\n  );\n  const reasoning_tokens = toNum(\n    attrs[\"gen_ai.usage.reasoning.output_tokens\"] ??\n      attrs[\"ai.usage.reasoningTokens\"] ??\n      attrs[\"llm.token_count.completion_details.reasoning\"],\n  );\n  const out: {\n    prompt_tokens?: number;\n    completion_tokens?: number;\n    reasoning_tokens?: number;\n  } = {};\n  if (prompt_tokens !== undefined) out.prompt_tokens = prompt_tokens;\n  if (completion_tokens !== undefined) out.completion_tokens = completion_tokens;\n  if (reasoning_tokens !== undefined) out.reasoning_tokens = reasoning_tokens;\n  return out;\n}\n\n/**\n * Coerce a span attribute value to a number. Accepts numbers directly and\n * parses finite numeric strings; ignores everything else.\n */\nfunction toNum(val: unknown): number | undefined {\n  if (typeof val === \"number\") return val;\n  if (typeof val === \"string\") {\n    const n = Number(val);\n    return Number.isFinite(n) ? n : undefined;\n  }\n  return undefined;\n}\n\n/**\n * Re-exported so the integrations can read prompt/completion tokens as\n * separate values (the OTel translate path returns them individually, not\n * as an endSpan-ready object).\n */\nexport function extractTokenCounts(attrs: SpanAttributes): {\n  promptTokens?: number;\n  completionTokens?: number;\n  reasoningTokens?: number;\n} {\n  const { prompt_tokens, completion_tokens, reasoning_tokens } =\n    extractTokenUsage(attrs);\n  // Plain assignments, not conditional spreads: this runs once per span on\n  // the translate hot path, and each spread allocates an intermediate object\n  // (three of them measurably regressed the 300-span benchmark).\n  const out: {\n    promptTokens?: number;\n    completionTokens?: number;\n    reasoningTokens?: number;\n  } = {};\n  if (prompt_tokens !== undefined) out.promptTokens = prompt_tokens;\n  if (completion_tokens !== undefined) out.completionTokens = completion_tokens;\n  if (reasoning_tokens !== undefined) out.reasoningTokens = reasoning_tokens;\n  return out;\n}\n"],"mappings":";;;;AAqCO,SAAS,kBAAkB,OAIhC;AACA,QAAM,gBAAgB;AAAA,IACpB,MAAM,2BAA2B,KAC/B,MAAM,4BAA4B,KAClC,MAAM,uBAAuB;AAAA,EACjC;AACA,QAAM,oBAAoB;AAAA,IACxB,MAAM,4BAA4B,KAChC,MAAM,gCAAgC,KACtC,MAAM,2BAA2B;AAAA,EACrC;AACA,QAAM,mBAAmB;AAAA,IACvB,MAAM,sCAAsC,KAC1C,MAAM,0BAA0B,KAChC,MAAM,8CAA8C;AAAA,EACxD;AACA,QAAM,MAIF,CAAC;AACL,MAAI,kBAAkB,OAAW,KAAI,gBAAgB;AACrD,MAAI,sBAAsB,OAAW,KAAI,oBAAoB;AAC7D,MAAI,qBAAqB,OAAW,KAAI,mBAAmB;AAC3D,SAAO;AACT;AAMA,SAAS,MAAM,KAAkC;AAC/C,MAAI,OAAO,QAAQ,SAAU,QAAO;AACpC,MAAI,OAAO,QAAQ,UAAU;AAC3B,UAAM,IAAI,OAAO,GAAG;AACpB,WAAO,OAAO,SAAS,CAAC,IAAI,IAAI;AAAA,EAClC;AACA,SAAO;AACT;AAOO,SAAS,mBAAmB,OAIjC;AACA,QAAM,EAAE,eAAe,mBAAmB,iBAAiB,IACzD,kBAAkB,KAAK;AAIzB,QAAM,MAIF,CAAC;AACL,MAAI,kBAAkB,OAAW,KAAI,eAAe;AACpD,MAAI,sBAAsB,OAAW,KAAI,mBAAmB;AAC5D,MAAI,qBAAqB,OAAW,KAAI,kBAAkB;AAC1D,SAAO;AACT;","names":[]}
+{"version":3,"sources":["../src/agent-task/integrations/token-usage.ts"],"sourcesContent":["/**\n * Shared token-usage extraction for the tracing integrations.\n *\n * Both the `createApoTracer` path (`ai-sdk.ts`) and the OTel-native path\n * (`otel-translate.ts` → `otel-processor.ts`) must extract token counts from\n * AI SDK span attributes using the same logic. This helper is the single\n * source of truth — it prevents the two paths from drifting apart (the bug\n * where one path reported tokens and the other silently dropped them).\n *\n * The AI SDK emits usage under `ai.usage.promptTokens` /\n * `ai.usage.completionTokens`. Standard OTel GenAI conventions use\n * `gen_ai.usage.input_tokens` / `gen_ai.usage.output_tokens`. Both are\n * recognized; standard `gen_ai.usage.*` takes precedence when both are present.\n *\n * @internal\n */\n\ntype SpanAttributes = Record<string, unknown>;\n\n/**\n * Extract `prompt_tokens` / `completion_tokens` / `reasoning_tokens` from a\n * span's attributes. Returns an object shaped to spread directly into\n * `trace.endSpan()` params. Omits keys when the corresponding attribute is\n * absent or non-numeric.\n *\n * Recognizes all known attribute-name conventions:\n *   - Vercel AI SDK: `ai.usage.promptTokens` / `ai.usage.completionTokens` /\n *     `ai.usage.reasoningTokens`\n *   - OTel GenAI: `gen_ai.usage.input_tokens` / `gen_ai.usage.output_tokens` /\n *     `gen_ai.usage.reasoning.output_tokens`\n *   - OTel GenAI (alt): `gen_ai.usage.prompt_tokens` / `gen_ai.usage.completion_tokens`\n *   - OpenInference: `llm.token_count.completion_details.reasoning`\n *\n * `gen_ai.usage.*` takes precedence over `ai.usage.*` when both are present.\n * Reasoning is omitted when unreported — a provider that never sends the\n * dimension must surface as unknown downstream, never as zero.\n */\nexport function extractTokenUsage(attrs: SpanAttributes): {\n  prompt_tokens?: number;\n  completion_tokens?: number;\n  reasoning_tokens?: number;\n} {\n  const prompt_tokens = toNum(\n    attrs[\"gen_ai.usage.input_tokens\"] ??\n      attrs[\"gen_ai.usage.prompt_tokens\"] ??\n      attrs[\"ai.usage.promptTokens\"],\n  );\n  const completion_tokens = toNum(\n    attrs[\"gen_ai.usage.output_tokens\"] ??\n      attrs[\"gen_ai.usage.completion_tokens\"] ??\n      attrs[\"ai.usage.completionTokens\"],\n  );\n  const reasoning_tokens = toNum(\n    attrs[\"gen_ai.usage.reasoning.output_tokens\"] ??\n      attrs[\"ai.usage.reasoningTokens\"] ??\n      attrs[\"llm.token_count.completion_details.reasoning\"],\n  );\n  const out: {\n    prompt_tokens?: number;\n    completion_tokens?: number;\n    reasoning_tokens?: number;\n  } = {};\n  if (prompt_tokens !== undefined) out.prompt_tokens = prompt_tokens;\n  if (completion_tokens !== undefined) out.completion_tokens = completion_tokens;\n  if (reasoning_tokens !== undefined) out.reasoning_tokens = reasoning_tokens;\n  return out;\n}\n\n/**\n * Coerce a span attribute value to a number. Accepts numbers directly and\n * parses finite numeric strings; ignores everything else.\n */\nfunction toNum(val: unknown): number | undefined {\n  if (typeof val === \"number\") return Number.isFinite(val) ? val : undefined;\n  if (typeof val === \"string\") {\n    const n = Number(val);\n    return Number.isFinite(n) ? n : undefined;\n  }\n  return undefined;\n}\n\n/**\n * Re-exported so the integrations can read prompt/completion tokens as\n * separate values (the OTel translate path returns them individually, not\n * as an endSpan-ready object).\n */\nexport function extractTokenCounts(attrs: SpanAttributes): {\n  promptTokens?: number;\n  completionTokens?: number;\n  reasoningTokens?: number;\n} {\n  const { prompt_tokens, completion_tokens, reasoning_tokens } =\n    extractTokenUsage(attrs);\n  // Plain assignments, not conditional spreads: this runs once per span on\n  // the translate hot path, and each spread allocates an intermediate object\n  // (three of them measurably regressed the 300-span benchmark).\n  const out: {\n    promptTokens?: number;\n    completionTokens?: number;\n    reasoningTokens?: number;\n  } = {};\n  if (prompt_tokens !== undefined) out.promptTokens = prompt_tokens;\n  if (completion_tokens !== undefined) out.completionTokens = completion_tokens;\n  if (reasoning_tokens !== undefined) out.reasoningTokens = reasoning_tokens;\n  return out;\n}\n"],"mappings":";;;;AAqCO,SAAS,kBAAkB,OAIhC;AACA,QAAM,gBAAgB;AAAA,IACpB,MAAM,2BAA2B,KAC/B,MAAM,4BAA4B,KAClC,MAAM,uBAAuB;AAAA,EACjC;AACA,QAAM,oBAAoB;AAAA,IACxB,MAAM,4BAA4B,KAChC,MAAM,gCAAgC,KACtC,MAAM,2BAA2B;AAAA,EACrC;AACA,QAAM,mBAAmB;AAAA,IACvB,MAAM,sCAAsC,KAC1C,MAAM,0BAA0B,KAChC,MAAM,8CAA8C;AAAA,EACxD;AACA,QAAM,MAIF,CAAC;AACL,MAAI,kBAAkB,OAAW,KAAI,gBAAgB;AACrD,MAAI,sBAAsB,OAAW,KAAI,oBAAoB;AAC7D,MAAI,qBAAqB,OAAW,KAAI,mBAAmB;AAC3D,SAAO;AACT;AAMA,SAAS,MAAM,KAAkC;AAC/C,MAAI,OAAO,QAAQ,SAAU,QAAO,OAAO,SAAS,GAAG,IAAI,MAAM;AACjE,MAAI,OAAO,QAAQ,UAAU;AAC3B,UAAM,IAAI,OAAO,GAAG;AACpB,WAAO,OAAO,SAAS,CAAC,IAAI,IAAI;AAAA,EAClC;AACA,SAAO;AACT;AAOO,SAAS,mBAAmB,OAIjC;AACA,QAAM,EAAE,eAAe,mBAAmB,iBAAiB,IACzD,kBAAkB,KAAK;AAIzB,QAAM,MAIF,CAAC;AACL,MAAI,kBAAkB,OAAW,KAAI,eAAe;AACpD,MAAI,sBAAsB,OAAW,KAAI,mBAAmB;AAC5D,MAAI,qBAAqB,OAAW,KAAI,kBAAkB;AAC1D,SAAO;AACT;","names":[]}

package/dist/{chunk-CKB3SUDO.js → chunk-IK4GJX7H.js}

@@ -2,7 +2,7 @@
 // Do not edit by hand — edit the src/ originals and rebuild.
 import {
   extractTokenUsage
-} from "./chunk-HLRXQAZ5.js";
+} from "./chunk-HRYXLRXJ.js";
 
 // src/agent-task/integrations/ai-sdk.ts
 function safeParse(raw) {
@@ -230,4 +230,4 @@ function extractTextFromAttrs(attrs) {
 export {
   createApoTracer
 };
-//# sourceMappingURL=chunk-CKB3SUDO.js.map
+//# sourceMappingURL=chunk-IK4GJX7H.js.map

package/dist/{chunk-IPQ6IIEG.js → chunk-KLZLE3VQ.js}

@@ -2,7 +2,7 @@
 // Do not edit by hand — edit the src/ originals and rebuild.
 import {
   extractTokenCounts
-} from "./chunk-HLRXQAZ5.js";
+} from "./chunk-HRYXLRXJ.js";
 
 // src/otel/index.ts
 import {
@@ -493,4 +493,4 @@ export {
   injectTraceparent,
   extractTraceparent
 };
-//# sourceMappingURL=chunk-IPQ6IIEG.js.map
+//# sourceMappingURL=chunk-KLZLE3VQ.js.map

package/dist/otel/index.js

@@ -13,8 +13,8 @@ import {
   traceRetriever,
   traceTool,
   withApoTrace
-} from "../chunk-IPQ6IIEG.js";
-import "../chunk-HLRXQAZ5.js";
+} from "../chunk-KLZLE3VQ.js";
+import "../chunk-HRYXLRXJ.js";
 export {
   buildApoAuthHeaders,
   configureApoTelemetry,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@apo-ai/sdk",
-  "version": "0.8.0",
+  "version": "0.9.1",
   "description": "TypeScript SDK for executable agent capability specifications and harness engineering with Apo",
   "keywords": [
     "ai-agents",
@@ -69,6 +69,7 @@
   },
   "devDependencies": {
     "@codspeed/vitest-plugin": "^5.7.1",
+    "@modelcontextprotocol/sdk": "^1.31.0",
     "@types/node": "^20.0.0",
     "@vitest/coverage-v8": "^4.1.9",
     "esbuild": "^0.21.5",
@@ -78,6 +79,7 @@
     "vitest": "^4.1.9"
   },
   "dependencies": {
+    "@ai-sdk/mcp": "^2.0.62",
     "@ai-sdk/openai-compatible": "^3.0.48",
     "@opentelemetry/api": "^1.9.1",
     "@opentelemetry/context-async-hooks": "^2.9.0",