@sentry/junior 0.28.0 → 0.30.0

package/dist/{chunk-375D5V4U.js → chunk-LEYD42MR.js}

@@ -1,6 +1,11 @@
 import {
+  extractGenAiUsageAttributes,
   getPluginRuntimeDependencies,
   getPluginRuntimePostinstall,
+  logException,
+  logWarn,
+  serializeGenAiAttribute,
+  setSpanAttributes,
   withSpan
 } from "./chunk-RZJDO55D.js";
 
@@ -8,6 +13,9 @@ import {
 import { createMemoryState } from "@chat-adapter/state-memory";
 import { createRedisState } from "@chat-adapter/state-redis";
 
+// src/chat/config.ts
+import { getModel } from "@mariozechner/pi-ai";
+
 // src/chat/optional-string.ts
 function toOptionalTrimmed(value) {
   if (!value) {
@@ -17,6 +25,233 @@ function toOptionalTrimmed(value) {
   return trimmed.length > 0 ? trimmed : void 0;
 }
 
+// src/chat/pi/client.ts
+import {
+  completeSimple,
+  getEnvApiKey,
+  getModels,
+  registerApiProvider
+} from "@mariozechner/pi-ai";
+import {
+  streamAnthropic,
+  streamSimpleAnthropic
+} from "@mariozechner/pi-ai/anthropic";
+registerApiProvider({
+  api: "anthropic-messages",
+  stream: streamAnthropic,
+  streamSimple: streamSimpleAnthropic
+});
+var GATEWAY_PROVIDER = "vercel-ai-gateway";
+var GEN_AI_PROVIDER_NAME = GATEWAY_PROVIDER;
+var GEN_AI_OPERATION_CHAT = "chat";
+var MISSING_GATEWAY_CREDENTIALS_ERROR = "Missing AI gateway credentials (AI_GATEWAY_API_KEY or VERCEL_OIDC_TOKEN)";
+function getGatewayApiKey() {
+  return toOptionalTrimmed(getEnvApiKey("vercel-ai-gateway")) ?? toOptionalTrimmed(process.env.VERCEL_OIDC_TOKEN);
+}
+function getPiGatewayApiKeyOverride() {
+  return toOptionalTrimmed(process.env.VERCEL_OIDC_TOKEN);
+}
+function extractText(message) {
+  return (message.content ?? []).filter((part) => part.type === "text" && typeof part.text === "string").map((part) => part.text ?? "").join("").trim();
+}
+function parseJsonCandidate(text) {
+  const trimmed = text.trim();
+  if (!trimmed) return void 0;
+  try {
+    return JSON.parse(trimmed);
+  } catch {
+    const fencedBlocks = [
+      ...trimmed.matchAll(/```(?:json)?\s*([\s\S]*?)\s*```/gi)
+    ];
+    for (const block of fencedBlocks) {
+      try {
+        return JSON.parse(block[1]);
+      } catch {
+      }
+    }
+    const openBraceIndex = trimmed.indexOf("{");
+    if (openBraceIndex >= 0) {
+      let depth = 0;
+      let inString = false;
+      let escaped = false;
+      for (let index = openBraceIndex; index < trimmed.length; index += 1) {
+        const char = trimmed[index];
+        if (inString) {
+          if (escaped) {
+            escaped = false;
+            continue;
+          }
+          if (char === "\\") {
+            escaped = true;
+            continue;
+          }
+          if (char === '"') {
+            inString = false;
+          }
+          continue;
+        }
+        if (char === '"') {
+          inString = true;
+          continue;
+        }
+        if (char === "{") {
+          depth += 1;
+          continue;
+        }
+        if (char === "}") {
+          depth -= 1;
+          if (depth === 0) {
+            const slice = trimmed.slice(openBraceIndex, index + 1);
+            try {
+              return JSON.parse(slice);
+            } catch {
+              break;
+            }
+          }
+        }
+      }
+    }
+    return void 0;
+  }
+}
+function resolveGatewayModel(modelId) {
+  const matched = getModels(GATEWAY_PROVIDER).find(
+    (model) => model.id === modelId
+  );
+  if (!matched) {
+    throw new Error(`Unknown AI Gateway model id: ${modelId}`);
+  }
+  return matched;
+}
+async function completeText(params) {
+  const model = resolveGatewayModel(params.modelId);
+  const apiKey = getPiGatewayApiKeyOverride();
+  const requestMessagesAttribute = serializeGenAiAttribute(params.messages);
+  const systemInstructionsAttribute = params.system ? serializeGenAiAttribute([{ type: "text", content: params.system }]) : void 0;
+  const startAttributes = {
+    "gen_ai.provider.name": GEN_AI_PROVIDER_NAME,
+    "gen_ai.operation.name": GEN_AI_OPERATION_CHAT,
+    "gen_ai.request.model": params.modelId,
+    ...systemInstructionsAttribute ? { "gen_ai.system_instructions": systemInstructionsAttribute } : {},
+    ...requestMessagesAttribute ? { "gen_ai.input.messages": requestMessagesAttribute } : {},
+    "app.ai.auth_mode": apiKey ? "oidc" : "api_key",
+    ...params.thinkingLevel ? { "app.ai.reasoning_effort": params.thinkingLevel } : {}
+  };
+  setSpanAttributes(startAttributes);
+  const message = await completeSimple(
+    model,
+    {
+      systemPrompt: params.system,
+      messages: params.messages
+    },
+    {
+      ...apiKey ? { apiKey } : {},
+      temperature: params.temperature,
+      maxTokens: params.maxTokens,
+      reasoning: params.thinkingLevel,
+      signal: params.signal,
+      metadata: params.metadata
+    }
+  );
+  const outputText = extractText(message);
+  const outputMessagesAttribute = serializeGenAiAttribute([
+    {
+      role: "assistant",
+      content: outputText ? [{ type: "text", text: outputText }] : []
+    }
+  ]);
+  const usageAttributes = extractGenAiUsageAttributes(message);
+  const endAttributes = {
+    "gen_ai.provider.name": GEN_AI_PROVIDER_NAME,
+    "gen_ai.operation.name": GEN_AI_OPERATION_CHAT,
+    "gen_ai.request.model": params.modelId,
+    ...outputMessagesAttribute ? { "gen_ai.output.messages": outputMessagesAttribute } : {},
+    ...usageAttributes,
+    ...message.stopReason ? { "gen_ai.response.finish_reasons": [message.stopReason] } : {},
+    ...params.thinkingLevel ? { "app.ai.reasoning_effort": params.thinkingLevel } : {}
+  };
+  setSpanAttributes(endAttributes);
+  if (message.stopReason === "error") {
+    const providerMessage = message.errorMessage?.trim() || "Unknown provider error";
+    logWarn(
+      "ai_completion_provider_error",
+      {},
+      {
+        "gen_ai.provider.name": GEN_AI_PROVIDER_NAME,
+        "gen_ai.operation.name": GEN_AI_OPERATION_CHAT,
+        "gen_ai.request.model": params.modelId,
+        "error.message": providerMessage
+      },
+      "AI completion returned provider error"
+    );
+    throw new Error(`AI provider error: ${providerMessage}`);
+  }
+  return {
+    message,
+    text: outputText
+  };
+}
+async function completeObject(params) {
+  const startedAt = Date.now();
+  let text = "";
+  try {
+    ({ text } = await completeText({
+      modelId: params.modelId,
+      system: params.system,
+      thinkingLevel: params.thinkingLevel,
+      temperature: params.temperature,
+      maxTokens: params.maxTokens,
+      signal: params.signal,
+      metadata: params.metadata,
+      messages: [
+        {
+          role: "user",
+          content: params.prompt,
+          timestamp: Date.now()
+        }
+      ]
+    }));
+  } catch (error) {
+    logException(
+      error,
+      "ai_completion_failed",
+      {},
+      {
+        "gen_ai.provider.name": GEN_AI_PROVIDER_NAME,
+        "gen_ai.operation.name": GEN_AI_OPERATION_CHAT,
+        "gen_ai.request.model": params.modelId,
+        "app.ai.duration_ms": Date.now() - startedAt
+      },
+      "AI object completion failed"
+    );
+    throw error;
+  }
+  const candidate = parseJsonCandidate(text);
+  const parsed = params.schema.safeParse(candidate);
+  if (!parsed.success) {
+    const preview = text.length > 400 ? `${text.slice(0, 400)}...` : text;
+    logWarn(
+      "ai_completion_schema_parse_failed",
+      {},
+      {
+        "gen_ai.provider.name": GEN_AI_PROVIDER_NAME,
+        "gen_ai.operation.name": GEN_AI_OPERATION_CHAT,
+        "gen_ai.request.model": params.modelId,
+        "app.ai.duration_ms": Date.now() - startedAt,
+        "app.ai.response_preview": preview
+      },
+      "AI object completion schema parse failed"
+    );
+    throw new Error(
+      `Model did not return valid JSON for schema: ${parsed.error.message}. Raw response: ${preview}`
+    );
+  }
+  return {
+    object: parsed.data,
+    text
+  };
+}
+
 // src/chat/config.ts
 var MIN_AGENT_TURN_TIMEOUT_MS = 10 * 1e3;
 var DEFAULT_AGENT_TURN_TIMEOUT_MS = 12 * 60 * 1e3;
@@ -79,15 +314,26 @@ function parseLoadingMessages(rawValue) {
     return value.trim();
   });
 }
+var DEFAULT_MODEL_ID = getModel("vercel-ai-gateway", "openai/gpt-5.4").id;
+var DEFAULT_FAST_MODEL_ID = getModel(
+  "vercel-ai-gateway",
+  "openai/gpt-5.4-mini"
+).id;
+function validateGatewayModelId(raw) {
+  const trimmed = toOptionalTrimmed(raw);
+  if (trimmed === void 0) return void 0;
+  resolveGatewayModel(trimmed);
+  return trimmed;
+}
 function readBotConfig(env) {
   const functionMaxDurationSeconds = resolveFunctionMaxDurationSeconds(env);
   const maxTurnTimeoutMs = resolveMaxTurnTimeoutMs(functionMaxDurationSeconds);
   return {
     userName: env.JUNIOR_BOT_NAME ?? "junior",
-    modelId: env.AI_MODEL ?? "openai/gpt-5.4",
-    fastModelId: env.AI_FAST_MODEL ?? env.AI_MODEL ?? "openai/gpt-5.4-mini",
+    modelId: validateGatewayModelId(env.AI_MODEL) ?? DEFAULT_MODEL_ID,
+    fastModelId: validateGatewayModelId(env.AI_FAST_MODEL ?? env.AI_MODEL) ?? DEFAULT_FAST_MODEL_ID,
     loadingMessages: parseLoadingMessages(env.JUNIOR_LOADING_MESSAGES),
-    visionModelId: toOptionalTrimmed(env.AI_VISION_MODEL),
+    visionModelId: validateGatewayModelId(env.AI_VISION_MODEL),
     turnTimeoutMs: parseAgentTurnTimeoutMs(
       env.AGENT_TURN_TIMEOUT_MS,
       maxTurnTimeoutMs
@@ -842,7 +1088,13 @@ function isSnapshotMissingError(error) {
 }
 
 export {
-  toOptionalTrimmed,
+  GEN_AI_PROVIDER_NAME,
+  MISSING_GATEWAY_CREDENTIALS_ERROR,
+  getGatewayApiKey,
+  getPiGatewayApiKeyOverride,
+  resolveGatewayModel,
+  completeText,
+  completeObject,
   botConfig,
   getSlackBotToken,
   getSlackSigningSecret,

package/dist/cli/snapshot-warmup.js

@@ -1,7 +1,7 @@
 import {
   disconnectStateAdapter,
   resolveRuntimeDependencySnapshot
-} from "../chunk-375D5V4U.js";
+} from "../chunk-LEYD42MR.js";
 import {
   getPluginProviders,
   getPluginRuntimeDependencies,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sentry/junior",
-  "version": "0.28.0",
+  "version": "0.30.0",
   "private": false,
   "publishConfig": {
     "access": "public"
@@ -25,8 +25,8 @@
     "@chat-adapter/state-memory": "4.26.0",
     "@chat-adapter/state-redis": "4.26.0",
     "@logtape/logtape": "^2.0.5",
-    "@mariozechner/pi-agent-core": "0.59.0",
-    "@mariozechner/pi-ai": "0.59.0",
+    "@mariozechner/pi-agent-core": "0.68.1",
+    "@mariozechner/pi-ai": "0.68.1",
     "@modelcontextprotocol/sdk": "1.29.0",
     "@sinclair/typebox": "^0.34.49",
     "@slack/web-api": "^7.15.1",