1313#include < iostream>
1414#include < memory>
1515
16+ #include " ../../cpp/json_schema_converter.h"
1617#include " ../../cpp/testing.h"
1718
1819// #include "../../cpp/support/logging.h"
@@ -144,7 +145,18 @@ EMSCRIPTEN_BINDINGS(xgrammar) {
144145 function (" vecIntToView" , &vecIntToView);
145146
146147 // Testing methods
147- function (" _JSONSchemaToEBNF" , &_JSONSchemaToEBNF);
148+ function (
149+ " _JSONSchemaToEBNF" ,
150+ select_overload<std::string (
151+ const std::string&,
152+ bool ,
153+ std::optional<int >,
154+ std::optional<std::pair<std::string, std::string>>,
155+ bool ,
156+ std::optional<int >,
157+ JSONFormat
158+ )>(&JSONSchemaToEBNF)
159+ );
148160 function (" DebugGetMaskedTokensFromBitmask" , &Testing_DebugGetMaskedTokensFromBitmask);
149161
150162 class_<Grammar>(" Grammar" )
@@ -166,7 +178,10 @@ EMSCRIPTEN_BINDINGS(xgrammar) {
166178 .constructor <const TokenizerInfo&, int , bool >()
167179 .function (" CompileJSONSchema" , &GrammarCompiler::CompileJSONSchema)
168180 .function (" CompileBuiltinJSONGrammar" , &GrammarCompiler::CompileBuiltinJSONGrammar)
169- .function (" CompileGrammar" , &GrammarCompiler::CompileGrammar)
181+ .function (
182+ " CompileGrammar" ,
183+ select_overload<CompiledGrammar (const Grammar&)>(&GrammarCompiler::CompileGrammar)
184+ )
170185 .function (" ClearCache" , &GrammarCompiler::ClearCache);
171186
172187 class_<GrammarMatcher>(" GrammarMatcher" )
@@ -179,5 +194,5 @@ EMSCRIPTEN_BINDINGS(xgrammar) {
179194 .function (" Reset" , &GrammarMatcher::Reset)
180195 .function (" FindJumpForwardString" , &GrammarMatcher::FindJumpForwardString)
181196 .function (" Rollback" , &GrammarMatcher::Rollback)
182- .function (" _DebugAcceptString " , &GrammarMatcher::_DebugAcceptString );
197+ .function (" AcceptString " , &GrammarMatcher::AcceptString );
183198}
0 commit comments