@grwnd/pi-governance 3.0.0 → 3.0.1

package/dist/extensions/index.js

@@ -3103,6 +3103,7 @@ var piGovernance = (pi) => {
     );
   });
   pi.on("tool_call", async (event, _ctx) => {
+    if (!audit || !policyEngine || !identity) return void 0;
     const { toolName, input } = event;
     const params = summarizeParams(toolName, input);
     const baseRecord = {
@@ -3319,6 +3320,7 @@ var piGovernance = (pi) => {
     return void 0;
   });
   pi.on("tool_result", async (event, _ctx) => {
+    if (!audit || !identity) return;
     if (dlpScanner && dlpMasker && event.output) {
       const result = dlpScanner.scan(event.output);
       if (result.hasMatches) {
@@ -3370,15 +3372,16 @@ var piGovernance = (pi) => {
   });
   pi.on("session_shutdown", async (_event, _ctx) => {
     configWatcher?.stop();
+    if (!audit) return;
     await audit.log({
       sessionId,
       event: "session_end",
-      userId: identity.userId,
-      role: identity.role,
-      orgUnit: identity.orgUnit,
+      userId: identity?.userId,
+      role: identity?.role,
+      orgUnit: identity?.orgUnit,
       metadata: {
         stats: { ...stats },
-        budget: { used: budgetTracker.used(), remaining: budgetTracker.remaining() },
+        budget: budgetTracker ? { used: budgetTracker.used(), remaining: budgetTracker.remaining() } : void 0,
         summary: Object.fromEntries(audit.getSummary())
       }
     });