@frengki0707/google-cloud-clone 1.33.0

package/lib/model-armor.mjs

@@ -0,0 +1,181 @@
+import { ModelArmorClient } from "@google-cloud/modelarmor";
+import { GenkitError } from "genkit";
+import { runInNewSpan } from "genkit/tracing";
+function extractText(parts) {
+  return parts.map((p) => p.text || "").join("");
+}
+function applySdp(messages, targetIndex, result, options) {
+  const sdpFilterResult = result.filterResults?.["sdp"]?.sdpFilterResult;
+  if (!sdpFilterResult) {
+    return { sdpApplied: false, messages };
+  }
+  if (typeof options.applyDeidentificationResults === "function") {
+    const newMessages = options.applyDeidentificationResults({
+      messages,
+      sdpResult: sdpFilterResult
+    });
+    if (!newMessages) {
+      return { sdpApplied: false, messages };
+    }
+    const sdpApplied = !!sdpFilterResult.deidentifyResult?.data?.text;
+    return { sdpApplied, messages: newMessages };
+  }
+  if (options.applyDeidentificationResults === true) {
+    const deidentifyResult = sdpFilterResult.deidentifyResult;
+    if (deidentifyResult && deidentifyResult.data?.text) {
+      const targetMessage = messages[targetIndex];
+      const nonTextParts = targetMessage.content.filter((p) => !p.text);
+      const newContent = [
+        ...nonTextParts,
+        { text: deidentifyResult.data.text }
+      ];
+      const newMessages = [...messages];
+      newMessages[targetIndex] = { ...targetMessage, content: newContent };
+      return {
+        sdpApplied: true,
+        messages: newMessages
+      };
+    }
+  }
+  return { sdpApplied: false, messages };
+}
+function shouldBlock(result, options, sdpApplied) {
+  if (result.filterMatchState !== "MATCH_FOUND") {
+    return false;
+  }
+  if (options.strictSdpEnforcement && sdpApplied) {
+    return true;
+  }
+  if (result.filterResults) {
+    for (const [key, filterResult] of Object.entries(result.filterResults)) {
+      if (options.filters && !options.filters.includes(key)) continue;
+      if (key === "sdp" && sdpApplied) continue;
+      const nestedResult = Object.values(filterResult)[0];
+      if (nestedResult?.matchState === "MATCH_FOUND") {
+        return true;
+      }
+    }
+  }
+  return false;
+}
+async function sanitizeUserPrompt(req, client, options) {
+  let targetMessageIndex = -1;
+  for (let i = req.messages.length - 1; i >= 0; i--) {
+    if (req.messages[i].role === "user") {
+      targetMessageIndex = i;
+      break;
+    }
+  }
+  if (targetMessageIndex !== -1) {
+    const userMessage = req.messages[targetMessageIndex];
+    const promptText = extractText(userMessage.content);
+    if (promptText) {
+      await runInNewSpan(
+        { metadata: { name: "sanitizeUserPrompt" } },
+        async (meta) => {
+          meta.input = {
+            name: options.templateName,
+            userPromptData: {
+              text: promptText
+            }
+          };
+          const [response] = await client.sanitizeUserPrompt({
+            name: options.templateName,
+            userPromptData: {
+              text: promptText
+            }
+          });
+          meta.output = response;
+          if (response.sanitizationResult) {
+            const result = response.sanitizationResult;
+            const { sdpApplied, messages: modifiedMessages } = applySdp(
+              req.messages,
+              targetMessageIndex,
+              result,
+              options
+            );
+            if (sdpApplied || typeof options.applyDeidentificationResults === "function") {
+              req.messages = modifiedMessages;
+            }
+            if (shouldBlock(result, options, sdpApplied)) {
+              throw new GenkitError({
+                status: "PERMISSION_DENIED",
+                message: "Model Armor blocked user prompt.",
+                detail: result
+              });
+            }
+          }
+        }
+      );
+    }
+  }
+}
+async function sanitizeModelResponse(response, client, options) {
+  const usingMessageProp = !!response.message;
+  const candidates = response.message ? [{ index: 0, message: response.message, finishReason: "stop" }] : response.candidates || [];
+  for (const candidate of candidates) {
+    const modelText = extractText(candidate.message.content);
+    if (modelText) {
+      await runInNewSpan(
+        { metadata: { name: "sanitizeModelResponse" } },
+        async (meta) => {
+          meta.input = {
+            name: options.templateName,
+            modelResponseData: {
+              text: modelText
+            }
+          };
+          const [apiResponse] = await client.sanitizeModelResponse({
+            name: options.templateName,
+            modelResponseData: {
+              text: modelText
+            }
+          });
+          meta.output = apiResponse;
+          if (apiResponse.sanitizationResult) {
+            const result = apiResponse.sanitizationResult;
+            const { sdpApplied, messages: modifiedMessages } = applySdp(
+              [candidate.message],
+              0,
+              result,
+              options
+            );
+            if (sdpApplied || typeof options.applyDeidentificationResults === "function") {
+              candidate.message = modifiedMessages[0];
+            }
+            if (shouldBlock(result, options, sdpApplied)) {
+              throw new GenkitError({
+                status: "PERMISSION_DENIED",
+                message: "Model Armor blocked model response.",
+                detail: result
+              });
+            }
+          }
+        }
+      );
+    }
+  }
+  if (usingMessageProp && candidates.length > 0) {
+    response.message = candidates[0].message;
+  }
+}
+function modelArmor(options) {
+  const client = options.client || new ModelArmorClient(options.clientOptions);
+  const protectionTarget = options.protectionTarget ?? "all";
+  const protectUserPrompt = protectionTarget === "all" || protectionTarget === "userPrompt";
+  const protectModelResponse = protectionTarget === "all" || protectionTarget === "modelResponse";
+  return async (req, next) => {
+    if (protectUserPrompt) {
+      await sanitizeUserPrompt(req, client, options);
+    }
+    const response = await next(req);
+    if (protectModelResponse) {
+      await sanitizeModelResponse(response, client, options);
+    }
+    return response;
+  };
+}
+export {
+  modelArmor
+};
+//# sourceMappingURL=model-armor.mjs.map

package/lib/model-armor.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/model-armor.ts"],"sourcesContent":["/**\n * Copyright 2025 Google LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { ModelArmorClient, protos } from '@google-cloud/modelarmor';\nimport { GenkitError } from 'genkit';\nimport {\n  GenerateRequest,\n  GenerateResponseData,\n  MessageData,\n  ModelMiddleware,\n  Part,\n} from 'genkit/model';\nimport { runInNewSpan } from 'genkit/tracing';\n\nexport interface ModelArmorOptions {\n  templateName: string;\n  client?: ModelArmorClient;\n  /**\n   * Options for the Model Armor client (e.g. apiEndpoint).\n   */\n  clientOptions?: ConstructorParameters<typeof ModelArmorClient>[0];\n  /**\n   * What to sanitize. Defaults to 'all'.\n   */\n  protectionTarget?: 'all' | 'userPrompt' | 'modelResponse';\n  /**\n   * Whether to block on SDP match even if the content was successfully de-identified.\n   * Defaults to false (lenient).\n   */\n  strictSdpEnforcement?: boolean;\n  /**\n   * List of filters to enforce. If not specified, all filters are enforced.\n   * Possible values: 'rai', 'pi_and_jailbreak', 'malicious_uris', 'csam', 'sdp'.\n   */\n  filters?: (\n    | 'rai'\n    | 'pi_and_jailbreak'\n    | 'malicious_uris'\n    | 'csam'\n    | 'sdp'\n    | (string & {})\n  )[];\n  /**\n   * Whether to apply the de-identification results to the content.\n   * - If true, the default logic (replace text, preserve structure) is used.\n   * - If false, no changes are applied.\n   * - If a function, it is called with the messages and SDP result, and should return the new messages.\n   *\n   * Defaults to false.\n   */\n  applyDeidentificationResults?:\n    | boolean\n    | ((data: {\n        messages: MessageData[];\n        sdpResult: protos.google.cloud.modelarmor.v1.ISdpFilterResult;\n      }) => MessageData[] | undefined);\n}\n\nfunction extractText(parts: Part[]): string {\n  return parts.map((p) => p.text || '').join('');\n}\n\n/**\n * If SDP (Sensitive Data Protection) filter returns sanitized data,\n * we swap out the data with sanitized data.\n */\nfunction applySdp(\n  messages: MessageData[],\n  targetIndex: number,\n  result: protos.google.cloud.modelarmor.v1.ISanitizationResult,\n  options: ModelArmorOptions\n): { sdpApplied: boolean; messages: MessageData[] } {\n  const sdpFilterResult = result.filterResults?.['sdp']?.sdpFilterResult;\n\n  if (!sdpFilterResult) {\n    return { sdpApplied: false, messages };\n  }\n\n  // If user provided applyDeidentificationResults, we use it to apply\n  // the deidentification results.\n  if (typeof options.applyDeidentificationResults === 'function') {\n    const newMessages = options.applyDeidentificationResults({\n      messages,\n      sdpResult: sdpFilterResult,\n    });\n    if (!newMessages) {\n      return { sdpApplied: false, messages };\n    }\n    const sdpApplied = !!sdpFilterResult.deidentifyResult?.data?.text;\n    return { sdpApplied, messages: newMessages };\n  }\n\n  // if applyDeidentificationResults is set to true, we use the default/basic\n  // approach to apply the results.\n  if (options.applyDeidentificationResults === true) {\n    const deidentifyResult = sdpFilterResult.deidentifyResult;\n    if (deidentifyResult && deidentifyResult.data?.text) {\n      const targetMessage = messages[targetIndex];\n      const nonTextParts = targetMessage.content.filter((p) => !p.text);\n      const newContent = [\n        ...nonTextParts,\n        { text: deidentifyResult.data.text },\n      ];\n      const newMessages = [...messages];\n      newMessages[targetIndex] = { ...targetMessage, content: newContent };\n      return {\n        sdpApplied: true,\n        messages: newMessages,\n      };\n    }\n  }\n\n  return { sdpApplied: false, messages };\n}\n\nfunction shouldBlock(\n  result: protos.google.cloud.modelarmor.v1.ISanitizationResult,\n  options: ModelArmorOptions,\n  sdpApplied: boolean\n): boolean {\n  if (result.filterMatchState !== 'MATCH_FOUND') {\n    return false;\n  }\n  // Check if we should block.\n  // If strict SDP enforcement is enabled and SDP was applied, we must block.\n  if (options.strictSdpEnforcement && sdpApplied) {\n    return true;\n  }\n  // Otherwise, check if any active filter matched.\n  if (result.filterResults) {\n    for (const [key, filterResult] of Object.entries(result.filterResults)) {\n      if (options.filters && !options.filters.includes(key)) continue;\n      if (key === 'sdp' && sdpApplied) continue;\n\n      // Look for matchState in the nested object\n      // e.g. filterResult.raiFilterResult.matchState\n      const nestedResult = Object.values(filterResult)[0];\n      if (nestedResult?.matchState === 'MATCH_FOUND') {\n        return true;\n      }\n    }\n  }\n  return false;\n}\n\nasync function sanitizeUserPrompt(\n  req: GenerateRequest,\n  client: ModelArmorClient,\n  options: ModelArmorOptions\n) {\n  let targetMessageIndex = -1;\n  // Find the last user message to sanitize\n  for (let i = req.messages.length - 1; i >= 0; i--) {\n    if (req.messages[i].role === 'user') {\n      targetMessageIndex = i;\n      break;\n    }\n  }\n\n  if (targetMessageIndex !== -1) {\n    const userMessage = req.messages[targetMessageIndex];\n    const promptText = extractText(userMessage.content);\n\n    if (promptText) {\n      await runInNewSpan(\n        { metadata: { name: 'sanitizeUserPrompt' } },\n        async (meta) => {\n          meta.input = {\n            name: options.templateName,\n            userPromptData: {\n              text: promptText,\n            },\n          };\n          const [response] = await client.sanitizeUserPrompt({\n            name: options.templateName,\n            userPromptData: {\n              text: promptText,\n            },\n          });\n          meta.output = response;\n\n          if (response.sanitizationResult) {\n            const result = response.sanitizationResult;\n            const { sdpApplied, messages: modifiedMessages } = applySdp(\n              req.messages,\n              targetMessageIndex,\n              result,\n              options\n            );\n\n            if (\n              sdpApplied ||\n              typeof options.applyDeidentificationResults === 'function'\n            ) {\n              req.messages = modifiedMessages;\n            }\n\n            if (shouldBlock(result, options, sdpApplied)) {\n              throw new GenkitError({\n                status: 'PERMISSION_DENIED',\n                message: 'Model Armor blocked user prompt.',\n                detail: result,\n              });\n            }\n          }\n        }\n      );\n    }\n  }\n}\n\nasync function sanitizeModelResponse(\n  response: GenerateResponseData,\n  client: ModelArmorClient,\n  options: ModelArmorOptions\n) {\n  const usingMessageProp = !!response.message;\n  const candidates = response.message\n    ? [{ index: 0, message: response.message, finishReason: 'stop' }]\n    : response.candidates || [];\n\n  for (const candidate of candidates) {\n    const modelText = extractText(candidate.message.content);\n\n    if (modelText) {\n      await runInNewSpan(\n        { metadata: { name: 'sanitizeModelResponse' } },\n        async (meta) => {\n          meta.input = {\n            name: options.templateName,\n            modelResponseData: {\n              text: modelText,\n            },\n          };\n          const [apiResponse] = await client.sanitizeModelResponse({\n            name: options.templateName,\n            modelResponseData: {\n              text: modelText,\n            },\n          });\n          meta.output = apiResponse;\n\n          if (apiResponse.sanitizationResult) {\n            const result = apiResponse.sanitizationResult;\n            const { sdpApplied, messages: modifiedMessages } = applySdp(\n              [candidate.message],\n              0,\n              result,\n              options\n            );\n\n            if (\n              sdpApplied ||\n              typeof options.applyDeidentificationResults === 'function'\n            ) {\n              candidate.message = modifiedMessages[0];\n            }\n\n            if (shouldBlock(result, options, sdpApplied)) {\n              throw new GenkitError({\n                status: 'PERMISSION_DENIED',\n                message: 'Model Armor blocked model response.',\n                detail: result,\n              });\n            }\n          }\n        }\n      );\n    }\n  }\n\n  if (usingMessageProp && candidates.length > 0) {\n    response.message = candidates[0].message;\n  }\n}\n\n/**\n * Model Middleware that uses Google Cloud Model Armor to sanitize user prompts and model responses.\n */\nexport function modelArmor(options: ModelArmorOptions): ModelMiddleware {\n  const client = options.client || new ModelArmorClient(options.clientOptions);\n  const protectionTarget = options.protectionTarget ?? 'all';\n  const protectUserPrompt =\n    protectionTarget === 'all' || protectionTarget === 'userPrompt';\n  const protectModelResponse =\n    protectionTarget === 'all' || protectionTarget === 'modelResponse';\n\n  return async (req, next) => {\n    // 1. Sanitize User Prompt\n    if (protectUserPrompt) {\n      await sanitizeUserPrompt(req, client, options);\n    }\n\n    // 2. Call Model\n    const response = await next(req);\n\n    // 3. Sanitize Model Response\n    if (protectModelResponse) {\n      await sanitizeModelResponse(response, client, options);\n    }\n\n    return response;\n  };\n}\n"],"mappings":"AAgBA,SAAS,wBAAgC;AACzC,SAAS,mBAAmB;AAQ5B,SAAS,oBAAoB;AA8C7B,SAAS,YAAY,OAAuB;AAC1C,SAAO,MAAM,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE;AAC/C;AAMA,SAAS,SACP,UACA,aACA,QACA,SACkD;AAClD,QAAM,kBAAkB,OAAO,gBAAgB,KAAK,GAAG;AAEvD,MAAI,CAAC,iBAAiB;AACpB,WAAO,EAAE,YAAY,OAAO,SAAS;AAAA,EACvC;AAIA,MAAI,OAAO,QAAQ,iCAAiC,YAAY;AAC9D,UAAM,cAAc,QAAQ,6BAA6B;AAAA,MACvD;AAAA,MACA,WAAW;AAAA,IACb,CAAC;AACD,QAAI,CAAC,aAAa;AAChB,aAAO,EAAE,YAAY,OAAO,SAAS;AAAA,IACvC;AACA,UAAM,aAAa,CAAC,CAAC,gBAAgB,kBAAkB,MAAM;AAC7D,WAAO,EAAE,YAAY,UAAU,YAAY;AAAA,EAC7C;AAIA,MAAI,QAAQ,iCAAiC,MAAM;AACjD,UAAM,mBAAmB,gBAAgB;AACzC,QAAI,oBAAoB,iBAAiB,MAAM,MAAM;AACnD,YAAM,gBAAgB,SAAS,WAAW;AAC1C,YAAM,eAAe,cAAc,QAAQ,OAAO,CAAC,MAAM,CAAC,EAAE,IAAI;AAChE,YAAM,aAAa;AAAA,QACjB,GAAG;AAAA,QACH,EAAE,MAAM,iBAAiB,KAAK,KAAK;AAAA,MACrC;AACA,YAAM,cAAc,CAAC,GAAG,QAAQ;AAChC,kBAAY,WAAW,IAAI,EAAE,GAAG,eAAe,SAAS,WAAW;AACnE,aAAO;AAAA,QACL,YAAY;AAAA,QACZ,UAAU;AAAA,MACZ;AAAA,IACF;AAAA,EACF;AAEA,SAAO,EAAE,YAAY,OAAO,SAAS;AACvC;AAEA,SAAS,YACP,QACA,SACA,YACS;AACT,MAAI,OAAO,qBAAqB,eAAe;AAC7C,WAAO;AAAA,EACT;AAGA,MAAI,QAAQ,wBAAwB,YAAY;AAC9C,WAAO;AAAA,EACT;AAEA,MAAI,OAAO,eAAe;AACxB,eAAW,CAAC,KAAK,YAAY,KAAK,OAAO,QAAQ,OAAO,aAAa,GAAG;AACtE,UAAI,QAAQ,WAAW,CAAC,QAAQ,QAAQ,SAAS,GAAG,EAAG;AACvD,UAAI,QAAQ,SAAS,WAAY;AAIjC,YAAM,eAAe,OAAO,OAAO,YAAY,EAAE,CAAC;AAClD,UAAI,cAAc,eAAe,eAAe;AAC9C,eAAO;AAAA,MACT;AAAA,IACF;AAAA,EACF;AACA,SAAO;AACT;AAEA,eAAe,mBACb,KACA,QACA,SACA;AACA,MAAI,qBAAqB;AAEzB,WAAS,IAAI,IAAI,SAAS,SAAS,GAAG,KAAK,GAAG,KAAK;AACjD,QAAI,IAAI,SAAS,CAAC,EAAE,SAAS,QAAQ;AACnC,2BAAqB;AACrB;AAAA,IACF;AAAA,EACF;AAEA,MAAI,uBAAuB,IAAI;AAC7B,UAAM,cAAc,IAAI,SAAS,kBAAkB;AACnD,UAAM,aAAa,YAAY,YAAY,OAAO;AAElD,QAAI,YAAY;AACd,YAAM;AAAA,QACJ,EAAE,UAAU,EAAE,MAAM,qBAAqB,EAAE;AAAA,QAC3C,OAAO,SAAS;AACd,eAAK,QAAQ;AAAA,YACX,MAAM,QAAQ;AAAA,YACd,gBAAgB;AAAA,cACd,MAAM;AAAA,YACR;AAAA,UACF;AACA,gBAAM,CAAC,QAAQ,IAAI,MAAM,OAAO,mBAAmB;AAAA,YACjD,MAAM,QAAQ;AAAA,YACd,gBAAgB;AAAA,cACd,MAAM;AAAA,YACR;AAAA,UACF,CAAC;AACD,eAAK,SAAS;AAEd,cAAI,SAAS,oBAAoB;AAC/B,kBAAM,SAAS,SAAS;AACxB,kBAAM,EAAE,YAAY,UAAU,iBAAiB,IAAI;AAAA,cACjD,IAAI;AAAA,cACJ;AAAA,cACA;AAAA,cACA;AAAA,YACF;AAEA,gBACE,cACA,OAAO,QAAQ,iCAAiC,YAChD;AACA,kBAAI,WAAW;AAAA,YACjB;AAEA,gBAAI,YAAY,QAAQ,SAAS,UAAU,GAAG;AAC5C,oBAAM,IAAI,YAAY;AAAA,gBACpB,QAAQ;AAAA,gBACR,SAAS;AAAA,gBACT,QAAQ;AAAA,cACV,CAAC;AAAA,YACH;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;AAEA,eAAe,sBACb,UACA,QACA,SACA;AACA,QAAM,mBAAmB,CAAC,CAAC,SAAS;AACpC,QAAM,aAAa,SAAS,UACxB,CAAC,EAAE,OAAO,GAAG,SAAS,SAAS,SAAS,cAAc,OAAO,CAAC,IAC9D,SAAS,cAAc,CAAC;AAE5B,aAAW,aAAa,YAAY;AAClC,UAAM,YAAY,YAAY,UAAU,QAAQ,OAAO;AAEvD,QAAI,WAAW;AACb,YAAM;AAAA,QACJ,EAAE,UAAU,EAAE,MAAM,wBAAwB,EAAE;AAAA,QAC9C,OAAO,SAAS;AACd,eAAK,QAAQ;AAAA,YACX,MAAM,QAAQ;AAAA,YACd,mBAAmB;AAAA,cACjB,MAAM;AAAA,YACR;AAAA,UACF;AACA,gBAAM,CAAC,WAAW,IAAI,MAAM,OAAO,sBAAsB;AAAA,YACvD,MAAM,QAAQ;AAAA,YACd,mBAAmB;AAAA,cACjB,MAAM;AAAA,YACR;AAAA,UACF,CAAC;AACD,eAAK,SAAS;AAEd,cAAI,YAAY,oBAAoB;AAClC,kBAAM,SAAS,YAAY;AAC3B,kBAAM,EAAE,YAAY,UAAU,iBAAiB,IAAI;AAAA,cACjD,CAAC,UAAU,OAAO;AAAA,cAClB;AAAA,cACA;AAAA,cACA;AAAA,YACF;AAEA,gBACE,cACA,OAAO,QAAQ,iCAAiC,YAChD;AACA,wBAAU,UAAU,iBAAiB,CAAC;AAAA,YACxC;AAEA,gBAAI,YAAY,QAAQ,SAAS,UAAU,GAAG;AAC5C,oBAAM,IAAI,YAAY;AAAA,gBACpB,QAAQ;AAAA,gBACR,SAAS;AAAA,gBACT,QAAQ;AAAA,cACV,CAAC;AAAA,YACH;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAEA,MAAI,oBAAoB,WAAW,SAAS,GAAG;AAC7C,aAAS,UAAU,WAAW,CAAC,EAAE;AAAA,EACnC;AACF;AAKO,SAAS,WAAW,SAA6C;AACtE,QAAM,SAAS,QAAQ,UAAU,IAAI,iBAAiB,QAAQ,aAAa;AAC3E,QAAM,mBAAmB,QAAQ,oBAAoB;AACrD,QAAM,oBACJ,qBAAqB,SAAS,qBAAqB;AACrD,QAAM,uBACJ,qBAAqB,SAAS,qBAAqB;AAErD,SAAO,OAAO,KAAK,SAAS;AAE1B,QAAI,mBAAmB;AACrB,YAAM,mBAAmB,KAAK,QAAQ,OAAO;AAAA,IAC/C;AAGA,UAAM,WAAW,MAAM,KAAK,GAAG;AAG/B,QAAI,sBAAsB;AACxB,YAAM,sBAAsB,UAAU,QAAQ,OAAO;AAAA,IACvD;AAEA,WAAO;AAAA,EACT;AACF;","names":[]}

package/lib/telemetry/action.d.mts

@@ -0,0 +1,27 @@
+import { ReadableSpan } from '@opentelemetry/sdk-trace-base';
+import { Telemetry } from '../metrics.mjs';
+import '@opentelemetry/api';
+
+/**
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+declare class ActionTelemetry implements Telemetry {
+    tick(span: ReadableSpan, logInputAndOutput: boolean, projectId?: string): void;
+    private writeLog;
+}
+declare const actionTelemetry: ActionTelemetry;
+
+export { actionTelemetry };

package/lib/telemetry/action.d.ts

@@ -0,0 +1,27 @@
+import { ReadableSpan } from '@opentelemetry/sdk-trace-base';
+import { Telemetry } from '../metrics.js';
+import '@opentelemetry/api';
+
+/**
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+declare class ActionTelemetry implements Telemetry {
+    tick(span: ReadableSpan, logInputAndOutput: boolean, projectId?: string): void;
+    private writeLog;
+}
+declare const actionTelemetry: ActionTelemetry;
+
+export { actionTelemetry };

package/lib/telemetry/action.js

@@ -0,0 +1,92 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var action_exports = {};
+__export(action_exports, {
+  actionTelemetry: () => actionTelemetry
+});
+module.exports = __toCommonJS(action_exports);
+var import_logging = require("genkit/logging");
+var import_tracing = require("genkit/tracing");
+var import_utils = require("../utils.js");
+class ActionTelemetry {
+  tick(span, logInputAndOutput, projectId) {
+    if (!logInputAndOutput) {
+      return;
+    }
+    const attributes = span.attributes;
+    const actionName = attributes["genkit:name"] || "<unknown>";
+    const subtype = attributes["genkit:metadata:subtype"];
+    if (subtype === "tool" || actionName === "generate") {
+      const path = attributes["genkit:path"] || "<unknown>";
+      const input = (0, import_utils.truncate)(attributes["genkit:input"]);
+      const output = (0, import_utils.truncate)(attributes["genkit:output"]);
+      const sessionId = attributes["genkit:sessionId"];
+      const threadName = attributes["genkit:threadName"];
+      let featureName = (0, import_utils.extractOuterFeatureNameFromPath)(path);
+      if (!featureName || featureName === "<unknown>") {
+        featureName = actionName;
+      }
+      if (input) {
+        this.writeLog(
+          span,
+          "Input",
+          featureName,
+          path,
+          input,
+          projectId,
+          sessionId,
+          threadName
+        );
+      }
+      if (output) {
+        this.writeLog(
+          span,
+          "Output",
+          featureName,
+          path,
+          output,
+          projectId,
+          sessionId,
+          threadName
+        );
+      }
+    }
+  }
+  writeLog(span, tag, featureName, qualifiedPath, content, projectId, sessionId, threadName) {
+    const path = (0, import_utils.truncatePath)((0, import_tracing.toDisplayPath)(qualifiedPath));
+    const sharedMetadata = {
+      ...(0, import_utils.createCommonLogAttributes)(span, projectId),
+      path,
+      qualifiedPath,
+      featureName,
+      sessionId,
+      threadName
+    };
+    import_logging.logger.logStructured(`${tag}[${path}, ${featureName}]`, {
+      ...sharedMetadata,
+      content
+    });
+  }
+}
+const actionTelemetry = new ActionTelemetry();
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  actionTelemetry
+});
+//# sourceMappingURL=action.js.map

package/lib/telemetry/action.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/telemetry/action.ts"],"sourcesContent":["/**\n * Copyright 2024 Google LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { ReadableSpan } from '@opentelemetry/sdk-trace-base';\nimport { logger } from 'genkit/logging';\nimport { toDisplayPath } from 'genkit/tracing';\nimport { type Telemetry } from '../metrics.js';\nimport {\n  createCommonLogAttributes,\n  extractOuterFeatureNameFromPath,\n  truncate,\n  truncatePath,\n} from '../utils.js';\n\nclass ActionTelemetry implements Telemetry {\n  tick(\n    span: ReadableSpan,\n    logInputAndOutput: boolean,\n    projectId?: string\n  ): void {\n    if (!logInputAndOutput) {\n      return;\n    }\n    const attributes = span.attributes;\n    const actionName = (attributes['genkit:name'] as string) || '<unknown>';\n    const subtype = attributes['genkit:metadata:subtype'] as string;\n\n    if (subtype === 'tool' || actionName === 'generate') {\n      const path = (attributes['genkit:path'] as string) || '<unknown>';\n      const input = truncate(attributes['genkit:input'] as string);\n      const output = truncate(attributes['genkit:output'] as string);\n      const sessionId = attributes['genkit:sessionId'] as string;\n      const threadName = attributes['genkit:threadName'] as string;\n      let featureName = extractOuterFeatureNameFromPath(path);\n      if (!featureName || featureName === '<unknown>') {\n        featureName = actionName;\n      }\n\n      if (input) {\n        this.writeLog(\n          span,\n          'Input',\n          featureName,\n          path,\n          input,\n          projectId,\n          sessionId,\n          threadName\n        );\n      }\n      if (output) {\n        this.writeLog(\n          span,\n          'Output',\n          featureName,\n          path,\n          output,\n          projectId,\n          sessionId,\n          threadName\n        );\n      }\n    }\n  }\n\n  private writeLog(\n    span: ReadableSpan,\n    tag: string,\n    featureName: string,\n    qualifiedPath: string,\n    content: string,\n    projectId?: string,\n    sessionId?: string,\n    threadName?: string\n  ) {\n    const path = truncatePath(toDisplayPath(qualifiedPath));\n    const sharedMetadata = {\n      ...createCommonLogAttributes(span, projectId),\n      path,\n      qualifiedPath,\n      featureName,\n      sessionId,\n      threadName,\n    };\n    logger.logStructured(`${tag}[${path}, ${featureName}]`, {\n      ...sharedMetadata,\n      content,\n    });\n  }\n}\n\nconst actionTelemetry = new ActionTelemetry();\nexport { actionTelemetry };\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAiBA,qBAAuB;AACvB,qBAA8B;AAE9B,mBAKO;AAEP,MAAM,gBAAqC;AAAA,EACzC,KACE,MACA,mBACA,WACM;AACN,QAAI,CAAC,mBAAmB;AACtB;AAAA,IACF;AACA,UAAM,aAAa,KAAK;AACxB,UAAM,aAAc,WAAW,aAAa,KAAgB;AAC5D,UAAM,UAAU,WAAW,yBAAyB;AAEpD,QAAI,YAAY,UAAU,eAAe,YAAY;AACnD,YAAM,OAAQ,WAAW,aAAa,KAAgB;AACtD,YAAM,YAAQ,uBAAS,WAAW,cAAc,CAAW;AAC3D,YAAM,aAAS,uBAAS,WAAW,eAAe,CAAW;AAC7D,YAAM,YAAY,WAAW,kBAAkB;AAC/C,YAAM,aAAa,WAAW,mBAAmB;AACjD,UAAI,kBAAc,8CAAgC,IAAI;AACtD,UAAI,CAAC,eAAe,gBAAgB,aAAa;AAC/C,sBAAc;AAAA,MAChB;AAEA,UAAI,OAAO;AACT,aAAK;AAAA,UACH;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,QACF;AAAA,MACF;AACA,UAAI,QAAQ;AACV,aAAK;AAAA,UACH;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,SACN,MACA,KACA,aACA,eACA,SACA,WACA,WACA,YACA;AACA,UAAM,WAAO,+BAAa,8BAAc,aAAa,CAAC;AACtD,UAAM,iBAAiB;AAAA,MACrB,OAAG,wCAA0B,MAAM,SAAS;AAAA,MAC5C;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AACA,0BAAO,cAAc,GAAG,GAAG,IAAI,IAAI,KAAK,WAAW,KAAK;AAAA,MACtD,GAAG;AAAA,MACH;AAAA,IACF,CAAC;AAAA,EACH;AACF;AAEA,MAAM,kBAAkB,IAAI,gBAAgB;","names":[]}

package/lib/telemetry/action.mjs

@@ -0,0 +1,73 @@
+import { logger } from "genkit/logging";
+import { toDisplayPath } from "genkit/tracing";
+import {
+  createCommonLogAttributes,
+  extractOuterFeatureNameFromPath,
+  truncate,
+  truncatePath
+} from "../utils.js";
+class ActionTelemetry {
+  tick(span, logInputAndOutput, projectId) {
+    if (!logInputAndOutput) {
+      return;
+    }
+    const attributes = span.attributes;
+    const actionName = attributes["genkit:name"] || "<unknown>";
+    const subtype = attributes["genkit:metadata:subtype"];
+    if (subtype === "tool" || actionName === "generate") {
+      const path = attributes["genkit:path"] || "<unknown>";
+      const input = truncate(attributes["genkit:input"]);
+      const output = truncate(attributes["genkit:output"]);
+      const sessionId = attributes["genkit:sessionId"];
+      const threadName = attributes["genkit:threadName"];
+      let featureName = extractOuterFeatureNameFromPath(path);
+      if (!featureName || featureName === "<unknown>") {
+        featureName = actionName;
+      }
+      if (input) {
+        this.writeLog(
+          span,
+          "Input",
+          featureName,
+          path,
+          input,
+          projectId,
+          sessionId,
+          threadName
+        );
+      }
+      if (output) {
+        this.writeLog(
+          span,
+          "Output",
+          featureName,
+          path,
+          output,
+          projectId,
+          sessionId,
+          threadName
+        );
+      }
+    }
+  }
+  writeLog(span, tag, featureName, qualifiedPath, content, projectId, sessionId, threadName) {
+    const path = truncatePath(toDisplayPath(qualifiedPath));
+    const sharedMetadata = {
+      ...createCommonLogAttributes(span, projectId),
+      path,
+      qualifiedPath,
+      featureName,
+      sessionId,
+      threadName
+    };
+    logger.logStructured(`${tag}[${path}, ${featureName}]`, {
+      ...sharedMetadata,
+      content
+    });
+  }
+}
+const actionTelemetry = new ActionTelemetry();
+export {
+  actionTelemetry
+};
+//# sourceMappingURL=action.mjs.map

package/lib/telemetry/action.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/telemetry/action.ts"],"sourcesContent":["/**\n * Copyright 2024 Google LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport type { ReadableSpan } from '@opentelemetry/sdk-trace-base';\nimport { logger } from 'genkit/logging';\nimport { toDisplayPath } from 'genkit/tracing';\nimport { type Telemetry } from '../metrics.js';\nimport {\n  createCommonLogAttributes,\n  extractOuterFeatureNameFromPath,\n  truncate,\n  truncatePath,\n} from '../utils.js';\n\nclass ActionTelemetry implements Telemetry {\n  tick(\n    span: ReadableSpan,\n    logInputAndOutput: boolean,\n    projectId?: string\n  ): void {\n    if (!logInputAndOutput) {\n      return;\n    }\n    const attributes = span.attributes;\n    const actionName = (attributes['genkit:name'] as string) || '<unknown>';\n    const subtype = attributes['genkit:metadata:subtype'] as string;\n\n    if (subtype === 'tool' || actionName === 'generate') {\n      const path = (attributes['genkit:path'] as string) || '<unknown>';\n      const input = truncate(attributes['genkit:input'] as string);\n      const output = truncate(attributes['genkit:output'] as string);\n      const sessionId = attributes['genkit:sessionId'] as string;\n      const threadName = attributes['genkit:threadName'] as string;\n      let featureName = extractOuterFeatureNameFromPath(path);\n      if (!featureName || featureName === '<unknown>') {\n        featureName = actionName;\n      }\n\n      if (input) {\n        this.writeLog(\n          span,\n          'Input',\n          featureName,\n          path,\n          input,\n          projectId,\n          sessionId,\n          threadName\n        );\n      }\n      if (output) {\n        this.writeLog(\n          span,\n          'Output',\n          featureName,\n          path,\n          output,\n          projectId,\n          sessionId,\n          threadName\n        );\n      }\n    }\n  }\n\n  private writeLog(\n    span: ReadableSpan,\n    tag: string,\n    featureName: string,\n    qualifiedPath: string,\n    content: string,\n    projectId?: string,\n    sessionId?: string,\n    threadName?: string\n  ) {\n    const path = truncatePath(toDisplayPath(qualifiedPath));\n    const sharedMetadata = {\n      ...createCommonLogAttributes(span, projectId),\n      path,\n      qualifiedPath,\n      featureName,\n      sessionId,\n      threadName,\n    };\n    logger.logStructured(`${tag}[${path}, ${featureName}]`, {\n      ...sharedMetadata,\n      content,\n    });\n  }\n}\n\nconst actionTelemetry = new ActionTelemetry();\nexport { actionTelemetry };\n"],"mappings":"AAiBA,SAAS,cAAc;AACvB,SAAS,qBAAqB;AAE9B;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAEP,MAAM,gBAAqC;AAAA,EACzC,KACE,MACA,mBACA,WACM;AACN,QAAI,CAAC,mBAAmB;AACtB;AAAA,IACF;AACA,UAAM,aAAa,KAAK;AACxB,UAAM,aAAc,WAAW,aAAa,KAAgB;AAC5D,UAAM,UAAU,WAAW,yBAAyB;AAEpD,QAAI,YAAY,UAAU,eAAe,YAAY;AACnD,YAAM,OAAQ,WAAW,aAAa,KAAgB;AACtD,YAAM,QAAQ,SAAS,WAAW,cAAc,CAAW;AAC3D,YAAM,SAAS,SAAS,WAAW,eAAe,CAAW;AAC7D,YAAM,YAAY,WAAW,kBAAkB;AAC/C,YAAM,aAAa,WAAW,mBAAmB;AACjD,UAAI,cAAc,gCAAgC,IAAI;AACtD,UAAI,CAAC,eAAe,gBAAgB,aAAa;AAC/C,sBAAc;AAAA,MAChB;AAEA,UAAI,OAAO;AACT,aAAK;AAAA,UACH;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,QACF;AAAA,MACF;AACA,UAAI,QAAQ;AACV,aAAK;AAAA,UACH;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,UACA;AAAA,QACF;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,SACN,MACA,KACA,aACA,eACA,SACA,WACA,WACA,YACA;AACA,UAAM,OAAO,aAAa,cAAc,aAAa,CAAC;AACtD,UAAM,iBAAiB;AAAA,MACrB,GAAG,0BAA0B,MAAM,SAAS;AAAA,MAC5C;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AACA,WAAO,cAAc,GAAG,GAAG,IAAI,IAAI,KAAK,WAAW,KAAK;AAAA,MACtD,GAAG;AAAA,MACH;AAAA,IACF,CAAC;AAAA,EACH;AACF;AAEA,MAAM,kBAAkB,IAAI,gBAAgB;","names":[]}

package/lib/telemetry/defaults.d.mts

@@ -0,0 +1,30 @@
+import { GcpTelemetryConfigOptions, GcpTelemetryConfig } from '../types.mjs';
+import '@opentelemetry/auto-instrumentations-node';
+import '@opentelemetry/instrumentation';
+import '@opentelemetry/sdk-trace-base';
+import 'google-auth-library';
+
+/**
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/** Consolidated defaults for telemetry configuration. */
+declare const TelemetryConfigs: {
+    defaults: (overrides?: GcpTelemetryConfigOptions) => GcpTelemetryConfig;
+    developmentDefaults: (overrides?: GcpTelemetryConfigOptions) => GcpTelemetryConfig;
+    productionDefaults: (overrides?: GcpTelemetryConfigOptions) => GcpTelemetryConfig;
+};
+
+export { TelemetryConfigs };

package/lib/telemetry/defaults.d.ts

@@ -0,0 +1,30 @@
+import { GcpTelemetryConfigOptions, GcpTelemetryConfig } from '../types.js';
+import '@opentelemetry/auto-instrumentations-node';
+import '@opentelemetry/instrumentation';
+import '@opentelemetry/sdk-trace-base';
+import 'google-auth-library';
+
+/**
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/** Consolidated defaults for telemetry configuration. */
+declare const TelemetryConfigs: {
+    defaults: (overrides?: GcpTelemetryConfigOptions) => GcpTelemetryConfig;
+    developmentDefaults: (overrides?: GcpTelemetryConfigOptions) => GcpTelemetryConfig;
+    productionDefaults: (overrides?: GcpTelemetryConfigOptions) => GcpTelemetryConfig;
+};
+
+export { TelemetryConfigs };

package/lib/telemetry/defaults.js

@@ -0,0 +1,70 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var defaults_exports = {};
+__export(defaults_exports, {
+  TelemetryConfigs: () => TelemetryConfigs
+});
+module.exports = __toCommonJS(defaults_exports);
+var import_sdk_trace_base = require("@opentelemetry/sdk-trace-base");
+var import_genkit = require("genkit");
+const TelemetryConfigs = {
+  defaults: (overrides = {}) => {
+    return (0, import_genkit.isDevEnv)() ? TelemetryConfigs.developmentDefaults(overrides) : TelemetryConfigs.productionDefaults(overrides);
+  },
+  developmentDefaults: (overrides = {}) => {
+    const defaults = {
+      sampler: new import_sdk_trace_base.AlwaysOnSampler(),
+      autoInstrumentation: true,
+      autoInstrumentationConfig: {
+        "@opentelemetry/instrumentation-dns": { enabled: false }
+      },
+      instrumentations: [],
+      metricExportIntervalMillis: 5e3,
+      metricExportTimeoutMillis: 5e3,
+      disableMetrics: false,
+      disableTraces: false,
+      exportInputAndOutput: !overrides.disableLoggingInputAndOutput,
+      export: !!overrides.forceDevExport
+      // false
+    };
+    return { ...defaults, ...overrides };
+  },
+  productionDefaults: (overrides = {}) => {
+    const defaults = {
+      sampler: new import_sdk_trace_base.AlwaysOnSampler(),
+      autoInstrumentation: true,
+      autoInstrumentationConfig: {
+        "@opentelemetry/instrumentation-dns": { enabled: false }
+      },
+      instrumentations: [],
+      metricExportIntervalMillis: 3e5,
+      metricExportTimeoutMillis: 3e5,
+      disableMetrics: false,
+      disableTraces: false,
+      exportInputAndOutput: !overrides.disableLoggingInputAndOutput,
+      export: true
+    };
+    return { ...defaults, ...overrides };
+  }
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  TelemetryConfigs
+});
+//# sourceMappingURL=defaults.js.map

package/lib/telemetry/defaults.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/telemetry/defaults.ts"],"sourcesContent":["/**\n * Copyright 2024 Google LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { AlwaysOnSampler } from '@opentelemetry/sdk-trace-base';\nimport { isDevEnv } from 'genkit';\nimport type {\n  GcpTelemetryConfig,\n  GcpTelemetryConfigOptions,\n} from '../types.js';\n\n/** Consolidated defaults for telemetry configuration. */\n\nexport const TelemetryConfigs = {\n  defaults: (overrides: GcpTelemetryConfigOptions = {}): GcpTelemetryConfig => {\n    return isDevEnv()\n      ? TelemetryConfigs.developmentDefaults(overrides)\n      : TelemetryConfigs.productionDefaults(overrides);\n  },\n\n  developmentDefaults: (\n    overrides: GcpTelemetryConfigOptions = {}\n  ): GcpTelemetryConfig => {\n    const defaults = {\n      sampler: new AlwaysOnSampler(),\n      autoInstrumentation: true,\n      autoInstrumentationConfig: {\n        '@opentelemetry/instrumentation-dns': { enabled: false },\n      },\n      instrumentations: [],\n      metricExportIntervalMillis: 5_000,\n      metricExportTimeoutMillis: 5_000,\n      disableMetrics: false,\n      disableTraces: false,\n      exportInputAndOutput: !overrides.disableLoggingInputAndOutput,\n      export: !!overrides.forceDevExport, // false\n    };\n    return { ...defaults, ...overrides };\n  },\n\n  productionDefaults: (\n    overrides: GcpTelemetryConfigOptions = {}\n  ): GcpTelemetryConfig => {\n    const defaults = {\n      sampler: new AlwaysOnSampler(),\n      autoInstrumentation: true,\n      autoInstrumentationConfig: {\n        '@opentelemetry/instrumentation-dns': { enabled: false },\n      },\n      instrumentations: [],\n      metricExportIntervalMillis: 300_000,\n      metricExportTimeoutMillis: 300_000,\n      disableMetrics: false,\n      disableTraces: false,\n      exportInputAndOutput: !overrides.disableLoggingInputAndOutput,\n      export: true,\n    };\n    return { ...defaults, ...overrides };\n  },\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAgBA,4BAAgC;AAChC,oBAAyB;AAQlB,MAAM,mBAAmB;AAAA,EAC9B,UAAU,CAAC,YAAuC,CAAC,MAA0B;AAC3E,eAAO,wBAAS,IACZ,iBAAiB,oBAAoB,SAAS,IAC9C,iBAAiB,mBAAmB,SAAS;AAAA,EACnD;AAAA,EAEA,qBAAqB,CACnB,YAAuC,CAAC,MACjB;AACvB,UAAM,WAAW;AAAA,MACf,SAAS,IAAI,sCAAgB;AAAA,MAC7B,qBAAqB;AAAA,MACrB,2BAA2B;AAAA,QACzB,sCAAsC,EAAE,SAAS,MAAM;AAAA,MACzD;AAAA,MACA,kBAAkB,CAAC;AAAA,MACnB,4BAA4B;AAAA,MAC5B,2BAA2B;AAAA,MAC3B,gBAAgB;AAAA,MAChB,eAAe;AAAA,MACf,sBAAsB,CAAC,UAAU;AAAA,MACjC,QAAQ,CAAC,CAAC,UAAU;AAAA;AAAA,IACtB;AACA,WAAO,EAAE,GAAG,UAAU,GAAG,UAAU;AAAA,EACrC;AAAA,EAEA,oBAAoB,CAClB,YAAuC,CAAC,MACjB;AACvB,UAAM,WAAW;AAAA,MACf,SAAS,IAAI,sCAAgB;AAAA,MAC7B,qBAAqB;AAAA,MACrB,2BAA2B;AAAA,QACzB,sCAAsC,EAAE,SAAS,MAAM;AAAA,MACzD;AAAA,MACA,kBAAkB,CAAC;AAAA,MACnB,4BAA4B;AAAA,MAC5B,2BAA2B;AAAA,MAC3B,gBAAgB;AAAA,MAChB,eAAe;AAAA,MACf,sBAAsB,CAAC,UAAU;AAAA,MACjC,QAAQ;AAAA,IACV;AACA,WAAO,EAAE,GAAG,UAAU,GAAG,UAAU;AAAA,EACrC;AACF;","names":[]}

package/lib/telemetry/defaults.mjs

@@ -0,0 +1,46 @@
+import { AlwaysOnSampler } from "@opentelemetry/sdk-trace-base";
+import { isDevEnv } from "genkit";
+const TelemetryConfigs = {
+  defaults: (overrides = {}) => {
+    return isDevEnv() ? TelemetryConfigs.developmentDefaults(overrides) : TelemetryConfigs.productionDefaults(overrides);
+  },
+  developmentDefaults: (overrides = {}) => {
+    const defaults = {
+      sampler: new AlwaysOnSampler(),
+      autoInstrumentation: true,
+      autoInstrumentationConfig: {
+        "@opentelemetry/instrumentation-dns": { enabled: false }
+      },
+      instrumentations: [],
+      metricExportIntervalMillis: 5e3,
+      metricExportTimeoutMillis: 5e3,
+      disableMetrics: false,
+      disableTraces: false,
+      exportInputAndOutput: !overrides.disableLoggingInputAndOutput,
+      export: !!overrides.forceDevExport
+      // false
+    };
+    return { ...defaults, ...overrides };
+  },
+  productionDefaults: (overrides = {}) => {
+    const defaults = {
+      sampler: new AlwaysOnSampler(),
+      autoInstrumentation: true,
+      autoInstrumentationConfig: {
+        "@opentelemetry/instrumentation-dns": { enabled: false }
+      },
+      instrumentations: [],
+      metricExportIntervalMillis: 3e5,
+      metricExportTimeoutMillis: 3e5,
+      disableMetrics: false,
+      disableTraces: false,
+      exportInputAndOutput: !overrides.disableLoggingInputAndOutput,
+      export: true
+    };
+    return { ...defaults, ...overrides };
+  }
+};
+export {
+  TelemetryConfigs
+};
+//# sourceMappingURL=defaults.mjs.map

package/lib/telemetry/defaults.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/telemetry/defaults.ts"],"sourcesContent":["/**\n * Copyright 2024 Google LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\n */\n\nimport { AlwaysOnSampler } from '@opentelemetry/sdk-trace-base';\nimport { isDevEnv } from 'genkit';\nimport type {\n  GcpTelemetryConfig,\n  GcpTelemetryConfigOptions,\n} from '../types.js';\n\n/** Consolidated defaults for telemetry configuration. */\n\nexport const TelemetryConfigs = {\n  defaults: (overrides: GcpTelemetryConfigOptions = {}): GcpTelemetryConfig => {\n    return isDevEnv()\n      ? TelemetryConfigs.developmentDefaults(overrides)\n      : TelemetryConfigs.productionDefaults(overrides);\n  },\n\n  developmentDefaults: (\n    overrides: GcpTelemetryConfigOptions = {}\n  ): GcpTelemetryConfig => {\n    const defaults = {\n      sampler: new AlwaysOnSampler(),\n      autoInstrumentation: true,\n      autoInstrumentationConfig: {\n        '@opentelemetry/instrumentation-dns': { enabled: false },\n      },\n      instrumentations: [],\n      metricExportIntervalMillis: 5_000,\n      metricExportTimeoutMillis: 5_000,\n      disableMetrics: false,\n      disableTraces: false,\n      exportInputAndOutput: !overrides.disableLoggingInputAndOutput,\n      export: !!overrides.forceDevExport, // false\n    };\n    return { ...defaults, ...overrides };\n  },\n\n  productionDefaults: (\n    overrides: GcpTelemetryConfigOptions = {}\n  ): GcpTelemetryConfig => {\n    const defaults = {\n      sampler: new AlwaysOnSampler(),\n      autoInstrumentation: true,\n      autoInstrumentationConfig: {\n        '@opentelemetry/instrumentation-dns': { enabled: false },\n      },\n      instrumentations: [],\n      metricExportIntervalMillis: 300_000,\n      metricExportTimeoutMillis: 300_000,\n      disableMetrics: false,\n      disableTraces: false,\n      exportInputAndOutput: !overrides.disableLoggingInputAndOutput,\n      export: true,\n    };\n    return { ...defaults, ...overrides };\n  },\n};\n"],"mappings":"AAgBA,SAAS,uBAAuB;AAChC,SAAS,gBAAgB;AAQlB,MAAM,mBAAmB;AAAA,EAC9B,UAAU,CAAC,YAAuC,CAAC,MAA0B;AAC3E,WAAO,SAAS,IACZ,iBAAiB,oBAAoB,SAAS,IAC9C,iBAAiB,mBAAmB,SAAS;AAAA,EACnD;AAAA,EAEA,qBAAqB,CACnB,YAAuC,CAAC,MACjB;AACvB,UAAM,WAAW;AAAA,MACf,SAAS,IAAI,gBAAgB;AAAA,MAC7B,qBAAqB;AAAA,MACrB,2BAA2B;AAAA,QACzB,sCAAsC,EAAE,SAAS,MAAM;AAAA,MACzD;AAAA,MACA,kBAAkB,CAAC;AAAA,MACnB,4BAA4B;AAAA,MAC5B,2BAA2B;AAAA,MAC3B,gBAAgB;AAAA,MAChB,eAAe;AAAA,MACf,sBAAsB,CAAC,UAAU;AAAA,MACjC,QAAQ,CAAC,CAAC,UAAU;AAAA;AAAA,IACtB;AACA,WAAO,EAAE,GAAG,UAAU,GAAG,UAAU;AAAA,EACrC;AAAA,EAEA,oBAAoB,CAClB,YAAuC,CAAC,MACjB;AACvB,UAAM,WAAW;AAAA,MACf,SAAS,IAAI,gBAAgB;AAAA,MAC7B,qBAAqB;AAAA,MACrB,2BAA2B;AAAA,QACzB,sCAAsC,EAAE,SAAS,MAAM;AAAA,MACzD;AAAA,MACA,kBAAkB,CAAC;AAAA,MACnB,4BAA4B;AAAA,MAC5B,2BAA2B;AAAA,MAC3B,gBAAgB;AAAA,MAChB,eAAe;AAAA,MACf,sBAAsB,CAAC,UAAU;AAAA,MACjC,QAAQ;AAAA,IACV;AACA,WAAO,EAAE,GAAG,UAAU,GAAG,UAAU;AAAA,EACrC;AACF;","names":[]}