Skip to content

Commit d5524a7

Browse files
authored
Fix modelset and scriptset add missing argument SOURCE and BLOB (#4)
1 parent adbd2a8 commit d5524a7

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

src/main/java/com/redislabs/redisai/Keyword.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
public enum Keyword implements ProtocolCommand{
77

8-
TENSOR, VALUES, INPUTS, OUTPUTS;
8+
TENSOR, VALUES, INPUTS, OUTPUTS, BLOB, SOURCE;
99

1010
private final byte[] raw;
1111

src/main/java/com/redislabs/redisai/RedisAI.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ public boolean setModel(String key, Backend backend, Device devive, String[] inp
116116
for(String output: outputs) {
117117
args.add(SafeEncoder.encode(output));
118118
}
119-
119+
args.add(Keyword.BLOB.getRaw());
120120
args.add(Files.readAllBytes(Paths.get(modelPath)));
121121

122122
return sendCommand(conn, Command.MODEL_SET, args.toArray(new byte[args.size()][]))
@@ -155,6 +155,7 @@ public boolean setScript(String key, Device device, String script){
155155
args.add(SafeEncoder.encode(key));
156156
args.add(device.getRaw());
157157

158+
args.add(Keyword.SOURCE.getRaw());
158159
args.add(SafeEncoder.encode(script));
159160

160161
return sendCommand(conn, Command.SCRIPT_SET, args.toArray(new byte[args.size()][]))

0 commit comments

Comments
 (0)