We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 68d63ed commit a556f43Copy full SHA for a556f43
proto/tei.proto
@@ -19,7 +19,9 @@ service Embed {
19
20
service Predict {
21
rpc Predict (PredictRequest) returns (PredictResponse);
22
+ rpc PredictPair (PredictPairRequest) returns (PredictResponse);
23
rpc PredictStream (stream PredictRequest) returns (stream PredictResponse);
24
+ rpc PredictPairStream (stream PredictPairRequest) returns (stream PredictResponse);
25
}
26
27
service Rerank {
@@ -113,6 +115,12 @@ message PredictRequest {
113
115
bool raw_scores = 3;
114
116
117
118
+message PredictPairRequest {
119
+ repeated string inputs = 1;
120
+ bool truncate = 2;
121
+ bool raw_scores = 3;
122
+}
123
+
124
message Prediction {
125
float score = 1;
126
string label = 2;
0 commit comments