fix: use loadAll()[0] fallback for AmbiguousScopeError

The previous fallback tried load(dbId, 'app') which fails when
the test infrastructure provisions modules with non-'app' scopes
(e.g. 'platform' + 'org'). Using loadAll()[0] picks the first
available instance without assuming a specific scope name.

Also fixes worker.test.ts: invalidateAll() → invalidate() to
match the ModuleLoader API.

Author: pyramation

job/compute-worker/src/compute-log.ts

@@ -47,11 +47,13 @@ export class ComputeLogTracker {
         return;
       }
       if (err instanceof AmbiguousScopeError) {
-        cfg = await this.loader.computeLog.load(this.databaseId, 'app');
+        const all = await this.loader.computeLog.loadAll(this.databaseId);
+        cfg = all[0];
       } else {
         throw err;
       }
     }
+    if (!cfg) return;
 
     const qualifiedTable = `"${cfg.publicSchema}"."${cfg.computeLogTable}"`;
     try {
@@ -91,11 +93,13 @@ export class ComputeLogTracker {
         return 0;
       }
       if (err instanceof AmbiguousScopeError) {
-        cfg = await this.loader.computeLog.load(this.databaseId, 'app');
+        const all = await this.loader.computeLog.loadAll(this.databaseId);
+        cfg = all[0];
       } else {
         throw err;
       }
     }
+    if (!cfg) return 0;
 
     try {
       const result = await this.pool.query(

job/compute-worker/src/discovery.ts

@@ -39,7 +39,8 @@ export class FunctionDiscovery {
       return await this.loader.function.load(this.databaseId, null);
     } catch (err) {
       if (err instanceof AmbiguousScopeError) {
-        return await this.loader.function.load(this.databaseId, 'app');
+        const all = await this.loader.function.loadAll(this.databaseId);
+        return all[0];
       }
       throw err;
     }

job/compute-worker/src/invocation.ts

@@ -40,7 +40,8 @@ export class InvocationTracker {
     } catch (err) {
       if (err instanceof ModuleNotProvisionedError) return null;
       if (err instanceof AmbiguousScopeError) {
-        return await this.loader.invocation.load(dbId, 'app');
+        const all = await this.loader.invocation.loadAll(dbId);
+        return all[0] ?? null;
       }
       throw err;
     }

job/worker/src/compute-meter.ts

@@ -52,7 +52,8 @@ export function logComputeUsage(pool: Pool, entry: MeterEntry): string {
       return await loader.invocation.load(databaseId, entry.scope ?? null);
     } catch (err) {
       if (err instanceof AmbiguousScopeError) {
-        return await loader.invocation.load(databaseId, 'app');
+        const all = await loader.invocation.loadAll(databaseId);
+        return all[0];
       }
       throw err;
     }

job/worker/src/graph-complete.ts

@@ -27,7 +27,8 @@ async function resolveGraph(pool: Pool, databaseId: string, scope?: string | nul
     return await getLoader(pool).graph.load(databaseId, scope ?? null);
   } catch (err) {
     if (err instanceof AmbiguousScopeError) {
-      return await getLoader(pool).graph.load(databaseId, 'app');
+      const all = await getLoader(pool).graph.loadAll(databaseId);
+      return all[0];
     }
     throw err;
   }

job/worker/src/storage-meter.ts

@@ -36,7 +36,8 @@ async function resolveComputeLog(pool: Pool, databaseId: string, scope: string |
     return await getLoader(pool).computeLog.load(databaseId, scope);
   } catch (err) {
     if (err instanceof AmbiguousScopeError) {
-      return await getLoader(pool).computeLog.load(databaseId, 'app');
+      const all = await getLoader(pool).computeLog.loadAll(databaseId);
+      return all[0];
     }
     throw err;
   }

packages/agentic-server/src/inference-meter.ts

@@ -46,7 +46,8 @@ async function resolveComputeLog(pool: Pool, databaseId: string, scope: string |
     return await getLoader(pool).computeLog.load(databaseId, scope);
   } catch (err) {
     if (err instanceof AmbiguousScopeError) {
-      return await getLoader(pool).computeLog.load(databaseId, 'app');
+      const all = await getLoader(pool).computeLog.loadAll(databaseId);
+      return all[0];
     }
     throw err;
   }

packages/functions-test/__tests__/worker.test.ts

@@ -65,7 +65,7 @@ beforeEach(async () => {
   await conn.db.beforeEach();
   mockServer.reset();
   // Invalidate caches so each test gets fresh module config
-  worker.loader.invalidateAll();
+  worker.loader.invalidate();
   worker.discovery.invalidateAll();
 });