Skip to content

Commit 836c026

Browse files
committed
Use installed llama3.2 model in Java Black Belt CLI
1 parent d94fb48 commit 836c026

1 file changed

Lines changed: 3 additions & 8 deletions

File tree

‎bin/BlackBeltCLI.java‎

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ private static Path resolvePrompt() {
4545

4646
private static int runEngine(String input) throws Exception {
4747
String prompt = Files.readString(resolvePrompt(), StandardCharsets.UTF_8);
48-
String model = System.getenv().getOrDefault("BBEA_MODEL", "llama3.1:8b");
48+
String model = System.getenv().getOrDefault("BBEA_MODEL", "llama3.2:latest");
4949
String url = System.getenv().getOrDefault("BBEA_ENGINE_URL", "http://127.0.0.1:11434/api/generate");
5050
String requestJson = "{" +
5151
"\"model\":" + quoteJson(model) + "," +
@@ -85,10 +85,6 @@ private static int interactive() throws Exception {
8585
System.out.println("Commands: help, quit, exit");
8686
continue;
8787
}
88-
if (!line.trim().startsWith("{") || !line.trim().endsWith("}")) {
89-
System.err.println("Audit input must be a JSON object.");
90-
continue;
91-
}
9288
int rc = runEngine(line);
9389
if (rc != 0) System.err.println("Audit failed (exit " + rc + ").");
9490
}
@@ -101,7 +97,7 @@ public static void main(String[] args) throws Exception {
10197
for (int i = 0; i < args.length; i++) {
10298
switch (args[i]) {
10399
case "--help", "-h" -> { usage(); return; }
104-
case "--version" -> { System.out.println("blackbelt-java 1.1 (BBEA CLI v2 transport)"); return; }
100+
case "--version" -> { System.out.println("blackbelt-java 1.2 (BBEA CLI v2 transport)"); return; }
105101
case "--file" -> {
106102
if (++i >= args.length) { usage(); System.exit(2); }
107103
file = args[i];
@@ -119,8 +115,7 @@ public static void main(String[] args) throws Exception {
119115
: Files.readString(Path.of(file), StandardCharsets.UTF_8);
120116
if (input.trim().isEmpty()) { System.err.println("No JSON audit input received."); System.exit(2); }
121117
if (!input.trim().startsWith("{") || !input.trim().endsWith("}")) {
122-
System.err.println("Audit input must be a JSON object."); System.exit(2);
123-
}
118+
System.err.println("Audit input must be a JSON object."); System.exit(2); }
124119
System.exit(runEngine(input));
125120
}
126121
}

0 commit comments

Comments
 (0)