File tree 3 files changed +6
-10
lines changed
3 files changed +6
-10
lines changed Original file line number Diff line number Diff line change @@ -231,11 +231,10 @@ impl TokenizeOutputOverflowedToken
231
231
232
232
#[ no_mangle]
233
233
pub unsafe extern "C" fn allocate_tokenizer (
234
- json_bytes_ptr : * const u8 ,
235
- json_bytes_length : usize ,
236
- ) -> * mut Tokenizer
234
+ json_bytes : ReadOnlyBuffer < u8 > )
235
+ -> * mut Tokenizer
237
236
{
238
- let json_bytes = slice :: from_raw_parts ( json_bytes_ptr , json_bytes_length ) ;
237
+ let json_bytes = json_bytes . as_slice ( ) ;
239
238
240
239
let tokenizer = Tokenizer :: from_bytes ( json_bytes) . unwrap ( ) ;
241
240
Original file line number Diff line number Diff line change @@ -256,14 +256,11 @@ public bool Truncate
256
256
257
257
public Tokenizer ( )
258
258
{
259
- var rawTokenizerData = Config . RawTokenizerData . Buffer ;
259
+ var rawTokenizerData = Config . RawTokenizerData . Buffer . AsReadOnly ( ) ;
260
260
261
261
Allocator = new ( ) ;
262
262
263
- TokenizerHandle = TokenizerNativeMethods . AllocateTokenizer (
264
- rawTokenizerData . Ptr ,
265
- rawTokenizerData . Length
266
- ) ;
263
+ TokenizerHandle = TokenizerNativeMethods . AllocateTokenizer ( rawTokenizerData ) ;
267
264
}
268
265
269
266
[ SkipLocalsInit ]
Original file line number Diff line number Diff line change @@ -9,7 +9,7 @@ internal static unsafe partial class TokenizerNativeMethods
9
9
private const string DLL_NAME = "tokenizers_net" ;
10
10
11
11
[ LibraryImport ( DLL_NAME , EntryPoint = "allocate_tokenizer" ) ]
12
- public static partial nint AllocateTokenizer ( byte * jsonBytesPtr , nuint jsonBytesLength ) ;
12
+ public static partial nint AllocateTokenizer ( ReadOnlyNativeBuffer < byte > jsonBytes ) ;
13
13
14
14
[ LibraryImport ( DLL_NAME , EntryPoint = "free_tokenizer" ) ]
15
15
public static partial void FreeTokenizer ( nint tokenizerHandle ) ;
You can’t perform that action at this time.
0 commit comments