safety-agent-cli 0.1.2 → 0.1.3

package/dist/commands/guard.js

@@ -10,14 +10,12 @@ 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) {
@@ -58,17 +56,6 @@ 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;
@@ -119,11 +106,7 @@ 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,
-            model: model,
-        });
+        const result = await client.guard({ input, systemPrompt });
         const { classification, violation_types, cwe_codes, usage } = result;
         const isBlocked = classification === "block";
         if (isBlocked) {

package/dist/commands/redact.js

@@ -16,18 +16,6 @@ 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;
@@ -69,19 +57,17 @@ export async function redactCommand(args) {
         // Command line argument
         text = args.join(" ");
         if (!text) {
-            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("Usage: superagent redact [--entities <entity1,entity2>] [--rewrite] <text>");
+            console.error('   or: echo \'{"text": "..."}\' | superagent redact [--entities <entity1,entity2>] [--rewrite]');
             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);
         }
     }
@@ -97,7 +83,7 @@ export async function redactCommand(args) {
     try {
         const result = await client.redact({
             input: text,
-            model: model || "openai/gpt-4o-mini",
+            model: "openai/gpt-4o-mini",
             entities,
             rewrite,
         });

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "safety-agent-cli",
-  "version": "0.1.2",
+  "version": "0.1.3",
   "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.0"
+    "safety-agent": "^0.1.3"
   }
 }