Implement managed browser security, individual agents and observability

apps/inspector/sandbox_worker.py

@@ -44,11 +44,15 @@ def main():
     data = base64.b64decode(request['data'], validate=True)
     if len(data)>MAX_BYTES: raise ValueError('Document exceeds inspection budget')
     route = request['route']
-    if route == '/inspect': result = inspect_content(data, request['policy'])
-    elif route == '/scan-download': result = scan_download(data, request['policy'])
-    elif route == '/reference-document': result = reference_document(data)
-    elif route == '/embed': result = embedding_features(data)
-    else: raise ValueError('Unknown inspection operation')
+    from stage_timings import collect, metadata
+    with collect(request.get('policy', {})) as timings:
+        if route == '/inspect': result = inspect_content(data, request['policy'])
+        elif route == '/scan-download': result = scan_download(data, request['policy'])
+        elif route == '/reference-document': result = reference_document(data)
+        elif route == '/embed': result = embedding_features(data)
+        else: raise ValueError('Unknown inspection operation')
+    diagnostic = metadata(timings)
+    if diagnostic is not None: result['_diagnosticStages'] = diagnostic
     response.write(json.dumps(result, allow_nan=False))
     response.flush()
     response.close()