@sentry/junior 0.75.0 → 0.76.0

package/dist/{chunk-NPVUAXUE.js → chunk-NFTMTIP3.js}

@@ -1,37 +1,86 @@
 import {
   createPluginLogger,
   createPluginState
-} from "./chunk-BNJIEFQC.js";
+} from "./chunk-56TBVRJG.js";
+import {
+  getDb
+} from "./chunk-NYKJ3KON.js";
+import {
+  SANDBOX_WORKSPACE_ROOT
+} from "./chunk-G3E7SCME.js";
 import {
   isConversationChannel,
   isConversationScopedChannel,
   isDmChannel,
   normalizeSlackConversationId
-} from "./chunk-62FUNJYS.js";
+} from "./chunk-Q6XFTRV5.js";
 import {
-  getPluginDbForRegistration
-} from "./chunk-D7NFH5GD.js";
-import {
-  SANDBOX_WORKSPACE_ROOT
-} from "./chunk-G3E7SCME.js";
+  botConfig,
+  completeObject,
+  embedTexts
+} from "./chunk-T77LUIX3.js";
 import {
   isActorUserId,
   parseActorUserId
-} from "./chunk-EIYL7I4S.js";
+} from "./chunk-VALUBQ7R.js";
 import {
-  logInfo
-} from "./chunk-OK4KKR7B.js";
+  logInfo,
+  logWarn
+} from "./chunk-EJN6G5A2.js";
+
+// src/chat/plugins/agent-hooks.ts
+import { promptMessageSchema } from "@sentry/junior-plugin-api";
+
+// src/chat/plugins/model.ts
+function createPluginModel(pluginName, options = {}, runtime = {}) {
+  return {
+    async completeObject(input) {
+      const modelId = options.structuredModelId ?? (options.structuredModel === "default" ? botConfig.modelId : botConfig.fastModelId);
+      const result = await completeObject({
+        modelId,
+        schema: input.schema,
+        prompt: input.prompt,
+        ...input.system !== void 0 ? { system: input.system } : {},
+        ...input.maxTokens !== void 0 ? { maxTokens: input.maxTokens } : {},
+        signal: runtime.signal,
+        metadata: {
+          pluginName,
+          pluginModelRole: "structured"
+        }
+      });
+      return { object: result.object };
+    }
+  };
+}
+function createPluginEmbedder(pluginName, runtime = {}) {
+  return {
+    async embedTexts(input) {
+      return await embedTexts({
+        modelId: botConfig.embeddingModelId,
+        texts: input.texts,
+        signal: runtime.signal,
+        metadata: {
+          pluginName,
+          pluginModelRole: "embedding"
+        }
+      });
+    }
+  };
+}
 
 // src/chat/tools/slack/context.ts
 function getSlackToolContext(context) {
   if (context.source.platform !== "slack") {
     return void 0;
   }
+  if (context.destination.platform !== "slack") {
+    throw new TypeError("Slack source requires a Slack destination");
+  }
   return {
-    destination: context.destination?.platform === "slack" ? context.destination : void 0,
+    destination: context.destination,
     source: context.source,
     requester: context.requester?.platform === "slack" ? context.requester : void 0,
-    destinationChannelId: context.destination?.platform === "slack" ? context.destination.channelId : void 0,
+    destinationChannelId: context.destination.channelId,
     messageTs: context.source.messageTs,
     sourceChannelId: context.source.channelId,
     teamId: context.source.teamId,
@@ -142,6 +191,7 @@ function resolveChannelCapabilities(channelId) {
 }
 
 // src/chat/plugins/agent-hooks.ts
+import { z } from "zod";
 var PluginHookDeniedError = class extends Error {
   constructor(message) {
     super(message);
@@ -167,18 +217,80 @@ var PLUGIN_ROUTE_METHODS = /* @__PURE__ */ new Set([
   "OPTIONS",
   "ALL"
 ]);
+var PLUGIN_PROMPT_CONTRIBUTION_TOTAL_MAX_CHARS = 16e3;
+var systemPromptMessageArraySchema = z.array(promptMessageSchema);
+var userPromptMessageArraySchema = z.array(promptMessageSchema);
 function isRecord(value) {
   return Boolean(value && typeof value === "object" && !Array.isArray(value));
 }
 function basePluginContext(plugin) {
   const name = plugin.manifest.name;
-  const db = getPluginDbForRegistration(plugin);
   return {
     plugin: { name },
     log: createPluginLogger(name),
-    ...db ? { db } : {}
+    db: getDb()
   };
 }
+function systemPromptPluginContext(plugin) {
+  return {
+    ...basePluginContext(plugin)
+  };
+}
+function invocationPluginContext(plugin, context) {
+  const base = basePluginContext(plugin);
+  const common = {
+    ...base,
+    conversationId: context.conversationId,
+    embedder: createPluginEmbedder(plugin.manifest.name),
+    source: context.source,
+    text: context.userText ?? "",
+    state: createPluginState(plugin.manifest.name)
+  };
+  if (context.source.platform === "slack") {
+    if (context.destination.platform !== "slack") {
+      throw new TypeError(
+        "Slack plugin prompt context requires Slack destination"
+      );
+    }
+    return {
+      ...common,
+      destination: context.destination,
+      requester: context.requester?.platform === "slack" ? context.requester : void 0
+    };
+  }
+  if (context.destination.platform !== "local") {
+    throw new TypeError(
+      "Local plugin prompt context requires local destination"
+    );
+  }
+  return {
+    ...common,
+    destination: context.destination,
+    requester: context.requester?.platform === "local" ? context.requester : void 0
+  };
+}
+function safeErrorMessage(error) {
+  return error instanceof Error ? error.message : String(error);
+}
+function toPromptContributionContext(args) {
+  return {
+    id: `${args.hookName}:${args.index}`,
+    pluginName: args.pluginName,
+    text: args.message.text
+  };
+}
+function logInvalidPromptContributions(args) {
+  logWarn(
+    "plugin_prompt_contribution_result_invalid",
+    {},
+    {
+      "app.plugin.hook": args.hookName,
+      "app.plugin.name": args.pluginName,
+      "app.plugin.validation_reason": "invalid_shape"
+    },
+    "Plugin prompt contribution result invalid"
+  );
+}
 function validatePlugins(plugins) {
   const seen = /* @__PURE__ */ new Set();
   for (const plugin of plugins) {
@@ -191,6 +303,18 @@ function validatePlugins(plugins) {
     if (seen.has(name)) {
       throw new Error(`Duplicate plugin name "${name}"`);
     }
+    for (const [taskName, task] of Object.entries(plugin.tasks ?? {})) {
+      if (!PLUGIN_TOOL_NAME_RE.test(taskName)) {
+        throw new Error(
+          `Plugin task "${taskName}" from plugin "${name}" must be a camelCase identifier`
+        );
+      }
+      if (typeof task.run !== "function") {
+        throw new Error(
+          `Plugin task "${taskName}" from plugin "${name}" must define a run function`
+        );
+      }
+    }
     seen.add(name);
   }
 }
@@ -205,6 +329,130 @@ function setPlugins(nextPlugins) {
 function getPlugins() {
   return [...registeredPlugins];
 }
+async function getPluginSystemPromptContributions(source) {
+  const contributions = [];
+  let totalChars = 0;
+  for (const plugin of getPlugins()) {
+    const pluginName = plugin.manifest.name;
+    const hook = plugin.hooks?.systemPrompt;
+    if (!hook) {
+      continue;
+    }
+    try {
+      const pluginContributions = await hook({
+        ...systemPromptPluginContext(plugin),
+        platform: source.platform
+      });
+      const result = systemPromptMessageArraySchema.safeParse(pluginContributions);
+      if (!result.success) {
+        logInvalidPromptContributions({
+          hookName: "systemPrompt",
+          pluginName
+        });
+        continue;
+      }
+      const acceptedContributions = result.data.map(
+        (message, index) => toPromptContributionContext({
+          hookName: "systemPrompt",
+          index,
+          message,
+          pluginName
+        })
+      );
+      const pluginContributionChars = acceptedContributions.reduce(
+        (sum, contribution) => sum + contribution.text.length,
+        0
+      );
+      if (totalChars + pluginContributionChars > PLUGIN_PROMPT_CONTRIBUTION_TOTAL_MAX_CHARS) {
+        logWarn(
+          "plugin_system_prompt_contribution_budget_exceeded",
+          {},
+          {
+            "app.plugin.name": pluginName
+          },
+          "Plugin system prompt contributions exceeded budget"
+        );
+        continue;
+      }
+      totalChars += pluginContributionChars;
+      contributions.push(...acceptedContributions);
+    } catch (error) {
+      logWarn(
+        "plugin_system_prompt_hook_failed",
+        {},
+        {
+          "app.plugin.name": pluginName,
+          "exception.message": safeErrorMessage(error)
+        },
+        "Plugin system prompt hook failed"
+      );
+    }
+  }
+  return contributions;
+}
+async function getPluginUserPromptContributions(args) {
+  const contributions = [];
+  let totalChars = 0;
+  for (const plugin of getPlugins()) {
+    const pluginName = plugin.manifest.name;
+    const hook = plugin.hooks?.userPrompt;
+    if (!hook) {
+      continue;
+    }
+    try {
+      const rawResult = await hook({
+        ...invocationPluginContext(plugin, args.context)
+      });
+      if (rawResult === void 0) {
+        continue;
+      }
+      const result = userPromptMessageArraySchema.safeParse(rawResult);
+      if (!result.success) {
+        logInvalidPromptContributions({
+          hookName: "userPrompt",
+          pluginName
+        });
+        continue;
+      }
+      const acceptedContributions = result.data.map(
+        (message, index) => toPromptContributionContext({
+          hookName: "userPrompt",
+          index,
+          message,
+          pluginName
+        })
+      );
+      const pluginContributionChars = acceptedContributions.reduce(
+        (sum, contribution) => sum + contribution.text.length,
+        0
+      );
+      if (totalChars + pluginContributionChars > PLUGIN_PROMPT_CONTRIBUTION_TOTAL_MAX_CHARS) {
+        logWarn(
+          "plugin_user_prompt_contribution_budget_exceeded",
+          {},
+          {
+            "app.plugin.name": pluginName
+          },
+          "Plugin user prompt contributions exceeded budget"
+        );
+        continue;
+      }
+      totalChars += pluginContributionChars;
+      contributions.push(...acceptedContributions);
+    } catch (error) {
+      logWarn(
+        "plugin_user_prompt_hook_failed",
+        {},
+        {
+          "app.plugin.name": pluginName,
+          "exception.message": safeErrorMessage(error)
+        },
+        "Plugin user prompt hook failed"
+      );
+    }
+  }
+  return contributions;
+}
 function getPluginTools(context) {
   const tools = {};
   for (const plugin of getPlugins()) {
@@ -213,7 +461,6 @@ function getPluginTools(context) {
     if (!hook) {
       continue;
     }
-    const destination = context.destination;
     const slackToolContext = getSlackToolContext(context);
     const credentialSubject = slackToolContext ? createSlackDirectCredentialSubject({
       channelId: slackToolContext.sourceChannelId,
@@ -226,24 +473,43 @@ function getPluginTools(context) {
       ),
       ...credentialSubject ? { credentialSubject } : {}
     } : void 0;
-    const pluginContext = context.source.platform === "slack" ? {
-      ...basePluginContext(plugin),
-      requester: context.requester?.platform === "slack" ? context.requester : void 0,
-      conversationId: context.conversationId,
-      destination: destination?.platform === "slack" ? destination : void 0,
-      slack: slackContext,
-      source: context.source,
-      userText: context.userText,
-      state: createPluginState(pluginName)
-    } : {
-      ...basePluginContext(plugin),
-      requester: context.requester?.platform === "local" ? context.requester : void 0,
-      conversationId: context.conversationId,
-      destination: destination?.platform === "local" ? destination : void 0,
-      source: context.source,
-      userText: context.userText,
-      state: createPluginState(pluginName)
-    };
+    let pluginContext;
+    if (context.source.platform === "slack") {
+      if (context.destination.platform !== "slack") {
+        throw new TypeError(
+          "Slack plugin tool context requires Slack destination"
+        );
+      }
+      pluginContext = {
+        ...basePluginContext(plugin),
+        requester: context.requester?.platform === "slack" ? context.requester : void 0,
+        conversationId: context.conversationId,
+        destination: context.destination,
+        slack: slackContext,
+        source: context.source,
+        userText: context.userText,
+        embedder: createPluginEmbedder(pluginName),
+        model: createPluginModel(pluginName, plugin.model),
+        state: createPluginState(pluginName)
+      };
+    } else {
+      if (context.destination.platform !== "local") {
+        throw new TypeError(
+          "Local plugin tool context requires local destination"
+        );
+      }
+      pluginContext = {
+        ...basePluginContext(plugin),
+        requester: context.requester?.platform === "local" ? context.requester : void 0,
+        conversationId: context.conversationId,
+        destination: context.destination,
+        source: context.source,
+        userText: context.userText,
+        embedder: createPluginEmbedder(pluginName),
+        model: createPluginModel(pluginName, plugin.model),
+        state: createPluginState(pluginName)
+      };
+    }
     const pluginTools = hook(pluginContext);
     for (const [name, tool] of Object.entries(pluginTools)) {
       if (!PLUGIN_TOOL_NAME_RE.test(name)) {
@@ -678,6 +944,8 @@ function createPluginHookRunner(input = {}) {
 }
 
 export {
+  createPluginModel,
+  createPluginEmbedder,
   getSlackToolContext,
   bindSlackDirectCredentialSubject,
   verifySlackDirectCredentialSubject,
@@ -686,6 +954,8 @@ export {
   validatePlugins,
   setPlugins,
   getPlugins,
+  getPluginSystemPromptContributions,
+  getPluginUserPromptContributions,
   getPluginTools,
   getPluginRoutes,
   getPluginSlackConversationLink,