Create protos/VertexInferenceService.proto
Browse files
protos/VertexInferenceService.proto
ADDED
@@ -0,0 +1,39 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
syntax = "proto3"; // 使用Protobuf版本3
|
2 |
+
|
3 |
+
package runtime.aot.machine_learning.parents.vertex;
|
4 |
+
|
5 |
+
service VertexInferenceService {
|
6 |
+
rpc Predict (Requests) returns (Response);
|
7 |
+
rpc PredictWithStream (Requests) returns (stream Response);
|
8 |
+
}
|
9 |
+
//Global
|
10 |
+
message Messages{
|
11 |
+
int64 unknown = 1; // 也许只是常量1(请求)/3(回应)
|
12 |
+
string message = 2; // 消息
|
13 |
+
}
|
14 |
+
|
15 |
+
// Requests
|
16 |
+
message Requests {
|
17 |
+
string models = 1; // 模型名称
|
18 |
+
Args args = 2;
|
19 |
+
}
|
20 |
+
|
21 |
+
message Args {
|
22 |
+
reserved 1;
|
23 |
+
Messages messages = 2; // 消息,可以视作user
|
24 |
+
string rules = 3; // 规则,可以视作system
|
25 |
+
}
|
26 |
+
|
27 |
+
// Response
|
28 |
+
message Response {
|
29 |
+
int64 response_code = 2; // 返回状态码,200,439等
|
30 |
+
Args1 args = 4;
|
31 |
+
}
|
32 |
+
|
33 |
+
message Args1{
|
34 |
+
Args2 args = 1; // 不知道什么意义的套一层
|
35 |
+
}
|
36 |
+
|
37 |
+
message Args2{
|
38 |
+
Messages args = 2; // 不知道什么意义的套一层
|
39 |
+
}
|