fix: create TriggerPipeline rpc for new model-backend (#27)

instill-ai · Feb 24, 2022 · 3c9f861 · 3c9f861
1 parent 919a751
commit 3c9f861
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/pipeline/pipeline.proto b/pipeline/pipeline.proto
@@ -58,7 +58,13 @@ service Pipeline {
       delete: "/pipelines/{name}"
     };
   }
-  rpc TriggerPipelineByUpload (stream TriggerPipelineRequest) returns (google.protobuf.Struct) {
+  rpc TriggerPipeline (TriggerPipelineRequest) returns (google.protobuf.Struct) {
+    option (google.api.http) = {
+      post: "/pipelines/{name}/outputs"
+      body: "*"
+    };
+  }
+  rpc TriggerPipelineByUpload (stream TriggerPipelineImageRequest) returns (google.protobuf.Struct) {
   }
 }
 
@@ -114,14 +120,22 @@ message DeletePipelineRequest {
 message TriggerPipelineContent {
   string url = 1 [(google.api.field_behavior) = OPTIONAL];
   string base64 = 2 [(google.api.field_behavior) = OPTIONAL];
-  bytes chunk = 3 [(google.api.field_behavior) = OPTIONAL];
 }
 
 message TriggerPipelineRequest {
   string name = 1 [(google.api.field_behavior) = REQUIRED];
   repeated TriggerPipelineContent contents = 2 [(google.api.field_behavior) = REQUIRED];
 }
 
+message TriggerPipelineImageContent {
+  bytes chunk = 1 [(google.api.field_behavior) = REQUIRED];
+}
+
+message TriggerPipelineImageRequest {
+  string name = 1 [(google.api.field_behavior) = REQUIRED];
+  repeated TriggerPipelineImageContent contents = 2 [(google.api.field_behavior) = REQUIRED];
+}
+
 message Scheduler {
   string crontab = 1 [(google.api.field_behavior) = REQUIRED];
 }