fix: udpate predict method for url/base64 (#26)

instill-ai · Feb 24, 2022 · 9bd38a3 · 9bd38a3
1 parent 9640542
commit 9bd38a3
Showing 1 changed file with 18 additions and 2 deletions.
diff --git a/model/model.proto b/model/model.proto
@@ -49,9 +49,9 @@ service Model {
   rpc PredictModelByUpload (stream PredictModelRequest) returns (google.protobuf.Struct) {}
 
   // This method handle request with file in body such as url/base64 encode
-  rpc PredictModel (PredictModelRequest) returns (google.protobuf.Struct) {
+  rpc PredictModel (PredictModelImageRequest) returns (google.protobuf.Struct) {
     option (google.api.http) = {
-      post: "/models/{name}/outputs"
+      post: "/models/{name}/versions/{version}/outputs"
       body: "*"
     };
   }
@@ -121,6 +121,22 @@ message PredictModelRequest {
     bytes content    = 3 [(google.api.field_behavior) = REQUIRED];
 }
 
+message ImageRequest {
+    // image url
+    string url       = 1;
+    // base 64 byte array of image content
+    string base64    = 2;
+}
+
+message PredictModelImageRequest {
+    // model name
+    string name                    = 1 [(google.api.field_behavior) = REQUIRED];
+    // model version
+    int32 version                  = 2 [(google.api.field_behavior) = REQUIRED];
+    // image content request
+    repeated ImageRequest contents = 3 [(google.api.field_behavior) = REQUIRED];
+}
+
 message ListModelRequest {}
 
 message ListModelResponse {