Skip to content

Commit a556f43

Browse files
feat: add PredictPair to proto (#225)
1 parent 68d63ed commit a556f43

File tree

2 files changed

+265
-417
lines changed

2 files changed

+265
-417
lines changed

proto/tei.proto

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,9 @@ service Embed {
1919

2020
service Predict {
2121
rpc Predict (PredictRequest) returns (PredictResponse);
22+
rpc PredictPair (PredictPairRequest) returns (PredictResponse);
2223
rpc PredictStream (stream PredictRequest) returns (stream PredictResponse);
24+
rpc PredictPairStream (stream PredictPairRequest) returns (stream PredictResponse);
2325
}
2426

2527
service Rerank {
@@ -113,6 +115,12 @@ message PredictRequest {
113115
bool raw_scores = 3;
114116
}
115117

118+
message PredictPairRequest {
119+
repeated string inputs = 1;
120+
bool truncate = 2;
121+
bool raw_scores = 3;
122+
}
123+
116124
message Prediction {
117125
float score = 1;
118126
string label = 2;

0 commit comments

Comments
 (0)