File tree Expand file tree Collapse file tree 1 file changed +14
-3
lines changed Expand file tree Collapse file tree 1 file changed +14
-3
lines changed Original file line number Diff line number Diff line change @@ -204,7 +204,11 @@ message TokenInfo {
204
204
message BatchedTokenizeRequest {
205
205
string model_id = 1 ;
206
206
repeated TokenizeRequest requests = 2 ;
207
- bool return_tokens = 3 ; //TBD
207
+ bool return_tokens = 3 ;
208
+ bool return_offsets = 4 ;
209
+
210
+ // Zero means don't truncate.
211
+ uint32 truncate_input_tokens = 5 ;
208
212
}
209
213
210
214
message BatchedTokenizeResponse {
@@ -216,10 +220,17 @@ message TokenizeRequest {
216
220
}
217
221
218
222
message TokenizeResponse {
223
+ message Offset {
224
+ uint32 start = 1 ;
225
+ uint32 end = 2 ;
226
+ }
227
+
219
228
uint32 token_count = 1 ;
220
- repeated string tokens = 2 ; // if include_tokens = true
221
229
222
- // We'll possibly add more later
230
+ // if return_tokens = true
231
+ repeated string tokens = 2 ;
232
+ // if return_tokens = true
233
+ repeated Offset offsets = 3 ;
223
234
}
224
235
225
236
You can’t perform that action at this time.
0 commit comments