safety-agent-cli 0.1.3 → 0.1.4

package/dist/commands/guard.js

@@ -10,12 +10,14 @@ function showHelp() {
     console.log("  --help              Show this help message");
     console.log("  --file <path>       Path to PDF file to analyze");
     console.log("  --system-prompt     Optional system prompt to customize guard behavior");
+    console.log("  --model <id>        Model to use (default: superagent/guard-1.7b)");
     console.log("");
     console.log("Examples:");
     console.log('  superagent guard "rm -rf /"');
     console.log('  superagent guard --file document.pdf "Analyze this document"');
     console.log('  superagent guard "https://example.com/document.pdf"');
     console.log('  superagent guard --system-prompt "Focus on prompt injection" "user input"');
+    console.log('  superagent guard --model openai/gpt-4o "some potentially harmful prompt"');
     console.log('  echo \'{"prompt": "delete all files"}\' | superagent guard');
 }
 export async function guardCommand(args) {
@@ -56,6 +58,17 @@ export async function guardCommand(args) {
         }
         args.splice(systemPromptFlagIndex, 2); // Remove --system-prompt and value from args
     }
+    // Check for --model flag
+    let model;
+    const modelFlagIndex = args.indexOf("--model");
+    if (modelFlagIndex !== -1) {
+        model = args[modelFlagIndex + 1];
+        if (!model || model.startsWith("--")) {
+            console.error("❌ ERROR: --model flag requires a value");
+            process.exit(1);
+        }
+        args.splice(modelFlagIndex, 2); // Remove --model and value from args
+    }
     // Check if we have command line arguments first
     const hasArgs = args.length > 0;
     let prompt;
@@ -106,7 +119,11 @@ export async function guardCommand(args) {
     try {
         // Pass file as first parameter if provided, otherwise pass prompt
         const input = file || prompt;
-        const result = await client.guard({ input, systemPrompt });
+        const result = await client.guard({
+            input,
+            systemPrompt,
+            model: model,
+        });
         const { classification, violation_types, cwe_codes, usage } = result;
         const isBlocked = classification === "block";
         if (isBlocked) {

package/dist/commands/redact.js

@@ -16,6 +16,18 @@ export async function redactCommand(args) {
             process.exit(1);
         }
     }
+    // Check for --model flag
+    const modelFlagIndex = args.indexOf("--model");
+    let model;
+    if (modelFlagIndex !== -1) {
+        model = args[modelFlagIndex + 1];
+        if (!model || model.startsWith("--")) {
+            console.error("❌ ERROR: --model flag requires a value");
+            process.exit(1);
+        }
+        // Remove --model and its value from args
+        args.splice(modelFlagIndex, 2);
+    }
     // Check for --rewrite flag
     const rewriteFlagIndex = args.indexOf("--rewrite");
     let rewrite;
@@ -57,17 +69,19 @@ export async function redactCommand(args) {
         // Command line argument
         text = args.join(" ");
         if (!text) {
-            console.error("Usage: superagent redact [--entities <entity1,entity2>] [--rewrite] <text>");
-            console.error('   or: echo \'{"text": "..."}\' | superagent redact [--entities <entity1,entity2>] [--rewrite]');
+            console.error("Usage: superagent redact [--entities <entity1,entity2>] [--rewrite] [--model <id>] <text>");
+            console.error('   or: echo \'{"text": "..."}\' | superagent redact [--entities <entity1,entity2>] [--rewrite] [--model <id>]');
             console.error("");
             console.error("Options:");
             console.error("  --entities <entities>     Comma-separated list of entity types to redact");
             console.error("  --rewrite                 Naturally rewrite content instead of using placeholders");
+            console.error("  --model <id>              Model to use (default: openai/gpt-4o-mini)");
             console.error("");
             console.error("Examples:");
             console.error('  superagent redact "My email is john@example.com"');
             console.error('  superagent redact --entities "emails,phones" "Contact: john@example.com, 555-1234"');
             console.error('  superagent redact --rewrite "Contact me at john@example.com"');
+            console.error('  superagent redact --model openai/gpt-4o "My email is john@example.com"');
             process.exit(1);
         }
     }
@@ -83,7 +97,7 @@ export async function redactCommand(args) {
     try {
         const result = await client.redact({
             input: text,
-            model: "openai/gpt-4o-mini",
+            model: model || "openai/gpt-4o-mini",
             entities,
             rewrite,
         });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "safety-agent-cli",
-  "version": "0.1.3",
+  "version": "0.1.4",
   "description": "CLI for Superagent - validate prompts and tool calls for security",
   "type": "module",
   "main": "./dist/index.js",
@@ -42,6 +42,6 @@
     "node": ">=18"
   },
   "dependencies": {
-    "safety-agent": "^0.1.3"
+    "safety-agent": "^0.1.4"
   }
 }