@grwnd/pi-governance 3.0.0 → 3.0.2

package/dist/extensions/index.js

@@ -2981,8 +2981,9 @@ var piGovernance = (pi) => {
     }
     const rulesFileCfg = config.policy?.yaml?.rules_file ?? "./governance-rules.yaml";
     paths.add(resolve(rulesFileCfg));
-    paths.add(resolve(ctx.workingDirectory, ".pi/governance.yaml"));
-    paths.add(resolve(ctx.workingDirectory, "governance-rules.yaml"));
+    const cwd = ctx.workingDirectory ?? process.cwd();
+    paths.add(resolve(cwd, ".pi/governance.yaml"));
+    paths.add(resolve(cwd, "governance-rules.yaml"));
     protectedPaths = paths;
     const chain = createIdentityChain(config.auth);
     identity = await chain.resolve();
@@ -3103,6 +3104,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 +3321,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 +3373,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())
       }
     });