@pauly4010/evalai-sdk 1.8.0 → 1.9.1

package/dist/cli/manifest.js

@@ -0,0 +1,282 @@
+"use strict";
+/**
+ * TICKET 2 — Evaluation Manifest Generation
+ *
+ * Goal: turn discovery output into a stable, versioned, machine-consumable artifact
+ * that becomes the input to run / impact / diff.
+ *
+ * This is the compiler output that everything else consumes.
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MANIFEST_SCHEMA_VERSION = exports.SDK_VERSION = void 0;
+exports.generateManifest = generateManifest;
+exports.writeManifest = writeManifest;
+exports.readManifest = readManifest;
+exports.readLock = readLock;
+const crypto = __importStar(require("node:crypto"));
+const fs = __importStar(require("node:fs/promises"));
+const path = __importStar(require("node:path"));
+const version_1 = require("../version");
+Object.defineProperty(exports, "SDK_VERSION", { enumerable: true, get: function () { return version_1.SDK_VERSION; } });
+/**
+ * Manifest schema version
+ */
+exports.MANIFEST_SCHEMA_VERSION = 1;
+/**
+ * Generate evaluation manifest from discovery results
+ */
+async function generateManifest(specs, projectRoot, projectName, executionMode) {
+    const generatedAt = Math.floor(Date.now() / 1000);
+    const namespace = generateNamespace(projectRoot);
+    // Process spec files and specs
+    const specFiles = [];
+    const processedSpecs = [];
+    // Group specs by file
+    const specsByFile = new Map();
+    for (const spec of specs) {
+        const normalizedPath = normalizePath(spec.file, projectRoot);
+        if (!specsByFile.has(normalizedPath)) {
+            specsByFile.set(normalizedPath, []);
+        }
+        specsByFile.get(normalizedPath)?.push(spec);
+    }
+    // Process each file
+    for (const [filePath, fileSpecs] of specsByFile) {
+        const absolutePath = path.join(projectRoot, filePath);
+        const fileHash = await hashFile(absolutePath);
+        specFiles.push({
+            filePath,
+            fileHash,
+            specCount: fileSpecs.length,
+        });
+        // Process individual specs
+        for (const spec of fileSpecs) {
+            const processedSpec = await processSpec(spec, filePath, projectRoot);
+            processedSpecs.push(processedSpec);
+        }
+    }
+    return {
+        schemaVersion: exports.MANIFEST_SCHEMA_VERSION,
+        generatedAt,
+        project: {
+            name: projectName,
+            root: ".",
+            namespace,
+        },
+        runtime: {
+            mode: executionMode.mode,
+            sdkVersion: version_1.SDK_VERSION,
+        },
+        specFiles,
+        specs: processedSpecs,
+    };
+}
+/**
+ * Process individual specification
+ */
+async function processSpec(spec, filePath, projectRoot) {
+    const absolutePath = path.join(projectRoot, filePath);
+    const content = await fs.readFile(absolutePath, "utf-8");
+    // Extract position from AST analysis (simplified for now)
+    const position = extractPosition(content, spec.name);
+    // Extract dependencies from content
+    const dependsOn = extractDependencies(content);
+    // Generate suite path from tags or file structure
+    const suitePath = generateSuitePath(spec.tags, filePath);
+    return {
+        id: spec.id,
+        name: spec.name,
+        suitePath,
+        filePath: normalizePath(spec.file, projectRoot),
+        position,
+        tags: spec.tags,
+        dependsOn,
+    };
+}
+/**
+ * Extract position from content (simplified implementation)
+ */
+function extractPosition(content, specName) {
+    const lines = content.split("\n");
+    const specPattern = new RegExp(`defineEval\\s*\\(\\s*["'\`]${specName}["'\`]`, "g");
+    let match = null;
+    let line = 1;
+    let column = 1;
+    for (let i = 0; i < lines.length; i++) {
+        const lineContent = lines[i];
+        specPattern.lastIndex = 0;
+        match = specPattern.exec(lineContent);
+        if (match) {
+            line = i + 1;
+            column = match.index + 1;
+            break;
+        }
+    }
+    return { line, column };
+}
+/**
+ * Extract dependencies from content
+ */
+function extractDependencies(content) {
+    const dependsOn = {
+        prompts: [],
+        datasets: [],
+        tools: [],
+        code: [],
+    };
+    // Extract from dependsOn option if present
+    const dependsOnMatch = content.match(/dependsOn\s*:\s*({[^}]+})/s);
+    if (dependsOnMatch) {
+        try {
+            // Use JSON.parse instead of eval for safety
+            const deps = JSON.parse(dependsOnMatch[1]);
+            return {
+                prompts: deps.prompts || [],
+                datasets: deps.datasets || [],
+                tools: deps.tools || [],
+                code: deps.code || [],
+            };
+        }
+        catch (_error) {
+            // If parsing fails, return empty dependencies
+            return {
+                prompts: [],
+                datasets: [],
+                tools: [],
+                code: [],
+            };
+        }
+    }
+    // Simple extraction as fallback
+    const patterns = {
+        prompts: /["']([^"']*\.md)["']/g,
+        datasets: /["']([^"']*\.json)["']/g,
+        tools: /["']([^"']*\.ts)["']/g,
+        code: /import.*from\s*["']([^"']+)["']/g,
+    };
+    for (const [type, pattern] of Object.entries(patterns)) {
+        let match;
+        match = pattern.exec(content);
+        while (match !== null) {
+            dependsOn[type].push(match[1]);
+            match = pattern.exec(content);
+        }
+    }
+    return dependsOn;
+}
+/**
+ * Generate suite path from tags or file structure
+ */
+function generateSuitePath(tags, filePath) {
+    // Use tags as primary suite path
+    if (tags.length > 0) {
+        return [tags[0]];
+    }
+    // Fall back to file structure
+    const parts = filePath.split("/");
+    if (parts.length > 1) {
+        return [parts[0]];
+    }
+    return ["general"];
+}
+/**
+ * Generate namespace from project root
+ */
+function generateNamespace(projectRoot) {
+    const hash = crypto.createHash("sha256");
+    hash.update(projectRoot);
+    return hash.digest("hex").slice(0, 8);
+}
+/**
+ * Normalize path to POSIX format
+ */
+function normalizePath(filePath, projectRoot) {
+    const relativePath = path.relative(projectRoot, filePath);
+    return relativePath.replace(/\\/g, "/");
+}
+/**
+ * Hash file content
+ */
+async function hashFile(filePath) {
+    const content = await fs.readFile(filePath, "utf-8");
+    const hash = crypto.createHash("sha256");
+    hash.update(content);
+    return `sha256:${hash.digest("hex")}`;
+}
+/**
+ * Write manifest to disk
+ */
+async function writeManifest(manifest, projectRoot) {
+    const evalaiDir = path.join(projectRoot, ".evalai");
+    // Ensure .evalai directory exists
+    await fs.mkdir(evalaiDir, { recursive: true });
+    // Write manifest.json
+    const manifestPath = path.join(evalaiDir, "manifest.json");
+    await fs.writeFile(manifestPath, JSON.stringify(manifest, null, 2), "utf-8");
+    // Write lock file
+    const lock = {
+        generatedAt: manifest.generatedAt,
+        fileHashes: Object.fromEntries(manifest.specFiles.map((f) => [f.filePath, f.fileHash])),
+    };
+    const lockPath = path.join(evalaiDir, "manifest.lock.json");
+    await fs.writeFile(lockPath, JSON.stringify(lock, null, 2), "utf-8");
+}
+/**
+ * Read existing manifest
+ */
+async function readManifest(projectRoot) {
+    const manifestPath = path.join(projectRoot, ".evalai", "manifest.json");
+    try {
+        const content = await fs.readFile(manifestPath, "utf-8");
+        return JSON.parse(content);
+    }
+    catch (_error) {
+        return null;
+    }
+}
+/**
+ * Read existing lock file
+ */
+async function readLock(projectRoot) {
+    const lockPath = path.join(projectRoot, ".evalai", "manifest.lock.json");
+    try {
+        const content = await fs.readFile(lockPath, "utf-8");
+        return JSON.parse(content);
+    }
+    catch (_error) {
+        return null;
+    }
+}

package/dist/cli/migrate.d.ts

@@ -0,0 +1,41 @@
+/**
+ * COMPAT-203: Config → DSL migration generator (file-based)
+ *
+ * CLI command: evalai migrate config --in evalai.config.json --out eval/legacy.spec.ts
+ * Generates defineEval() calls with comments and TODOs for manual completion
+ */
+import { Command } from "commander";
+/**
+ * Migration options
+ */
+interface MigrateOptions {
+    /** Input config file path */
+    input: string;
+    /** Output DSL file path */
+    output: string;
+    /** Include detailed comments */
+    verbose?: boolean;
+    /** Generate helper functions */
+    helpers?: boolean;
+    /** Preserve original test IDs */
+    preserveIds?: boolean;
+    /** Include provenance metadata */
+    provenance?: boolean;
+}
+/**
+ * Main migration function
+ */
+export declare function migrateConfig(options: MigrateOptions): Promise<void>;
+/**
+ * CLI command definition
+ */
+export declare function createMigrateCommand(): Command;
+/**
+ * Validate config file structure
+ */
+export declare function validateConfigFile(filePath: string): Promise<boolean>;
+/**
+ * Show migration preview without writing files
+ */
+export declare function previewMigration(filePath: string): Promise<void>;
+export {};

package/dist/cli/migrate.js

@@ -0,0 +1,349 @@
+"use strict";
+/**
+ * COMPAT-203: Config → DSL migration generator (file-based)
+ *
+ * CLI command: evalai migrate config --in evalai.config.json --out eval/legacy.spec.ts
+ * Generates defineEval() calls with comments and TODOs for manual completion
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.migrateConfig = migrateConfig;
+exports.createMigrateCommand = createMigrateCommand;
+exports.validateConfigFile = validateConfigFile;
+exports.previewMigration = previewMigration;
+const fs = __importStar(require("node:fs/promises"));
+const path = __importStar(require("node:path"));
+const commander_1 = require("commander");
+const testsuite_to_dsl_1 = require("../runtime/adapters/testsuite-to-dsl");
+const testing_1 = require("../testing");
+/**
+ * Read and parse evalai.config.json
+ */
+async function readConfigFile(filePath) {
+    try {
+        const content = await fs.readFile(filePath, "utf-8");
+        return JSON.parse(content);
+    }
+    catch (error) {
+        throw new Error(`Failed to read config file ${filePath}: ${error instanceof Error ? error.message : String(error)}`);
+    }
+}
+/**
+ * Extract TestSuite data from config
+ */
+function extractTestSuitesFromConfig(config) {
+    const suites = [];
+    // Handle different config structures
+    if (config.tests) {
+        // Direct tests array
+        const suite = (0, testing_1.createTestSuite)("config-tests", {
+            cases: config.tests,
+            executor: config.executor,
+            timeout: config.timeout,
+            parallel: config.parallel,
+            stopOnFailure: config.stopOnFailure,
+            retries: config.retries,
+        });
+        suites.push({ name: "config-tests", suite });
+    }
+    if (config.suites) {
+        // Multiple named suites
+        for (const [suiteName, suiteConfig] of Object.entries(config.suites)) {
+            const suite = (0, testing_1.createTestSuite)(suiteName, suiteConfig);
+            suites.push({ name: suiteName, suite });
+        }
+    }
+    if (config.testSuites) {
+        // Alternative property name
+        for (const [suiteName, suiteConfig] of Object.entries(config.testSuites)) {
+            const suite = (0, testing_1.createTestSuite)(suiteName, suiteConfig);
+            suites.push({ name: suiteName, suite });
+        }
+    }
+    return suites;
+}
+/**
+ * Generate DSL file header
+ */
+function generateFileHeader(_config, options) {
+    const timestamp = new Date().toISOString();
+    const inputPath = path.resolve(options.input);
+    const outputPath = path.resolve(options.output);
+    return [
+        `/**`,
+        ` * Auto-generated EvalAI DSL from configuration`,
+        ` * `,
+        ` * Generated at: ${timestamp}`,
+        ` * Source config: ${inputPath}`,
+        ` * Output file: ${outputPath}`,
+        ` * `,
+        ` * This file contains defineEval() specifications migrated from evalai.config.json`,
+        ` * `,
+        ` * ⚠️  IMPORTANT: This is a best-effort migration. Manual review and completion required.`,
+        ` * `,
+        ` * Migration notes:`,
+        ` * - Executors have been converted to async functions`,
+        ` * - Assertions have been converted where possible`,
+        ` * - Complex logic may need manual adaptation`,
+        ` * - Review TODO comments for items requiring attention`,
+        ` */`,
+        ``,
+        `import { defineEval, createResult } from '@pauly4010/evalai-sdk';`,
+        ``,
+    ].join("\n");
+}
+/**
+ * Generate helper functions for the entire file
+ */
+function generateGlobalHelpers(config, _options) {
+    const helpers = [];
+    // Add executor helper if config has executor
+    if (config.executor) {
+        helpers.push([
+            `/**`,
+            ` * Legacy executor function from config`,
+            ` * TODO: Replace with actual executor implementation`,
+            ` */`,
+            `async function legacyExecutor(input: string): Promise<string> {`,
+            `  // Original executor was: ${config.executor.toString()}`,
+            `  // TODO: Implement actual executor logic here`,
+            `  return input; // Placeholder`,
+            `}`,
+            ``,
+        ].join("\n"));
+    }
+    // Add assertion helpers
+    helpers.push([
+        `/**`,
+        ` * Helper function for legacy assertion evaluation`,
+        ` * TODO: Implement actual assertion logic based on original config`,
+        ` */`,
+        `function evaluateAssertions(output: string, expected?: string): boolean {`,
+        `  if (expected !== undefined) {`,
+        `    return output === expected;`,
+        `  }`,
+        `  return output.length > 0;`,
+        `}`,
+        ``,
+    ].join("\n"));
+    // Add evaluation helper
+    helpers.push([
+        `/**`,
+        ` * Legacy test evaluation function`,
+        ` * TODO: Adapt based on your original test logic`,
+        ` */`,
+        `async function evaluateLegacyTest(input: string, expected?: string): Promise<unknown> {`,
+        `  const output = await legacyExecutor(input);`,
+        `  const passed = evaluateAssertions(output, expected);`,
+        `  `,
+        `  return createResult({`,
+        `    pass: passed,`,
+        `    score: passed ? 100 : 0,`,
+        `    metadata: { input, expected },`,
+        `  });`,
+        `}`,
+        ``,
+    ].join("\n"));
+    return helpers.join("\n");
+}
+/**
+ * Generate DSL content for a single suite
+ */
+function generateSuiteDSL(suiteName, suite, options) {
+    const dslCode = (0, testsuite_to_dsl_1.generateDefineEvalCode)(suite, {
+        generateHelpers: options.helpers,
+        preserveIds: options.preserveIds,
+        includeProvenance: options.provenance,
+    });
+    // Add suite-specific comments
+    const header = [
+        `/**`,
+        ` * Test suite: ${suiteName}`,
+        ` * Migrated from evalai.config.json`,
+        ` * `,
+        ` * TODO items for this suite:`,
+        ` * - Review executor implementation`,
+        ` * - Verify assertion logic`,
+        ` * - Test with actual data`,
+        ` */`,
+        ``,
+    ].join("\n");
+    return header + dslCode;
+}
+/**
+ * Generate migration summary
+ */
+function generateSummary(suites, options) {
+    const totalTests = suites.reduce((sum, { suite }) => sum + suite.getTests().length, 0);
+    const totalSuites = suites.length;
+    return [
+        `/**`,
+        ` * Migration Summary`,
+        ` * =================`,
+        ` * `,
+        ` * Total suites migrated: ${totalSuites}`,
+        ` * Total tests migrated: ${totalTests}`,
+        ` * `,
+        ` * Migration options used:`,
+        ` * - Include helpers: ${options.helpers}`,
+        ` * - Preserve IDs: ${options.preserveIds}`,
+        ` * - Include provenance: ${options.provenance}`,
+        ` * `,
+        ` * Next steps:`,
+        ` * 1. Review all TODO comments in this file`,
+        ` * 2. Implement actual executor logic`,
+        ` * 3. Adapt complex assertions`,
+        ` * 4. Test with real data`,
+        ` * 5. Remove evalai.config.json when satisfied`,
+        ` * `,
+        ` * For help with migration, see: https://github.com/pauly7610/ai-evaluation-platform/docs/MIGRATION.md`,
+        ` */`,
+        ``,
+    ].join("\n");
+}
+/**
+ * Main migration function
+ */
+async function migrateConfig(options) {
+    try {
+        // Read input config
+        const config = await readConfigFile(options.input);
+        // Extract test suites
+        const suites = extractTestSuitesFromConfig(config);
+        if (suites.length === 0) {
+            throw new Error("No test suites found in config file. Check config structure.");
+        }
+        // Generate DSL content
+        const content = [
+            generateFileHeader(config, options),
+            generateGlobalHelpers(config, options),
+            ...suites.map(({ name, suite }) => generateSuiteDSL(name, suite, options)),
+            generateSummary(suites, options),
+        ].join("\n");
+        // Ensure output directory exists
+        const outputDir = path.dirname(options.output);
+        await fs.mkdir(outputDir, { recursive: true });
+        // Write output file
+        await fs.writeFile(options.output, content, "utf-8");
+        console.log(`✅ Migration complete!`);
+        console.log(`📁 Output written to: ${path.resolve(options.output)}`);
+        console.log(`📊 Migrated ${suites.length} suites with ${suites.reduce((sum, { suite }) => sum + suite.getTests().length, 0)} tests`);
+        console.log(`\n⚠️  Remember to review TODO comments and test the migration!`);
+    }
+    catch (error) {
+        console.error(`❌ Migration failed: ${error instanceof Error ? error.message : String(error)}`);
+        process.exit(1);
+    }
+}
+/**
+ * CLI command definition
+ */
+function createMigrateCommand() {
+    const command = new commander_1.Command("migrate")
+        .description("Migrate legacy configuration to new DSL format")
+        .command("config")
+        .description("Migrate evalai.config.json to defineEval() specifications")
+        .requiredOption("-i, --in <path>", "Input config file path")
+        .requiredOption("-o, --out <path>", "Output DSL file path")
+        .option("-v, --verbose", "Include detailed comments and logging", false)
+        .option("--no-helpers", "Don't generate helper functions")
+        .option("--no-preserve-ids", "Don't preserve original test IDs")
+        .option("--no-provenance", "Don't include provenance metadata")
+        .action(async (options) => {
+        const migrateOptions = {
+            input: options.in,
+            output: options.out,
+            verbose: options.verbose,
+            helpers: options.helpers !== false,
+            preserveIds: options.preserveIds !== false,
+            provenance: options.provenance !== false,
+        };
+        await migrateConfig(migrateOptions);
+    });
+    return command;
+}
+/**
+ * Validate config file structure
+ */
+async function validateConfigFile(filePath) {
+    try {
+        const config = await readConfigFile(filePath);
+        // Basic validation
+        if (!config || typeof config !== "object") {
+            throw new Error("Config file must contain a valid JSON object");
+        }
+        // Check for test data
+        const hasTests = config.tests || config.suites || config.testSuites;
+        if (!hasTests) {
+            throw new Error("Config file must contain 'tests', 'suites', or 'testSuites' property");
+        }
+        console.log(`✅ Config file ${filePath} appears valid for migration`);
+        return true;
+    }
+    catch (error) {
+        console.error(`❌ Config validation failed: ${error instanceof Error ? error.message : String(error)}`);
+        return false;
+    }
+}
+/**
+ * Show migration preview without writing files
+ */
+async function previewMigration(filePath) {
+    try {
+        const config = await readConfigFile(filePath);
+        const suites = extractTestSuitesFromConfig(config);
+        console.log(`📋 Migration preview for: ${filePath}`);
+        console.log(``);
+        console.log(`Found ${suites.length} test suites:`);
+        console.log(``);
+        for (const { name, suite } of suites) {
+            const tests = suite.getTests();
+            console.log(`  📁 ${name}: ${tests.length} tests`);
+            if (tests.length > 0) {
+                console.log(`     Tests: ${tests
+                    .slice(0, 3)
+                    .map((t) => t.id)
+                    .join(", ")}${tests.length > 3 ? "..." : ""}`);
+            }
+        }
+        console.log(``);
+        console.log(`Total tests to migrate: ${suites.reduce((sum, { suite }) => sum + suite.getTests().length, 0)}`);
+        console.log(``);
+        console.log(`To migrate, run: evalai migrate config --in ${filePath} --out eval/migrated.spec.ts`);
+    }
+    catch (error) {
+        console.error(`❌ Preview failed: ${error instanceof Error ? error.message : String(error)}`);
+    }
+}

package/dist/cli/policy-packs.js

@@ -22,7 +22,10 @@ exports.POLICY_PACKS = {
         1: {
             policyId: "SOC2",
             version: 1,
-            thresholds: { requiredSafetyRate: 0.95, maxFlags: ["SAFETY_RISK", "LOW_PASS_RATE"] },
+            thresholds: {
+                requiredSafetyRate: 0.95,
+                maxFlags: ["SAFETY_RISK", "LOW_PASS_RATE"],
+            },
             rationale: "SOC2 trust criteria for security and availability.",
             checks: ["safety_rate", "flag_restrictions"],
         },
@@ -40,7 +43,10 @@ exports.POLICY_PACKS = {
         1: {
             policyId: "PCI_DSS",
             version: 1,
-            thresholds: { requiredSafetyRate: 0.99, maxFlags: ["SAFETY_RISK", "LOW_PASS_RATE"] },
+            thresholds: {
+                requiredSafetyRate: 0.99,
+                maxFlags: ["SAFETY_RISK", "LOW_PASS_RATE"],
+            },
             rationale: "PCI DSS cardholder data security standards.",
             checks: ["safety_rate", "flag_restrictions"],
         },