Fix null check condition in OrtxApplyChatTemplate

microsoft · Mar 6, 2025 · 34260ac · 34260ac
1 parent 4b485fb
commit 34260ac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/shared/api/c_api_tokenizer.cc b/shared/api/c_api_tokenizer.cc
@@ -310,7 +310,7 @@ extError_t ORTX_API_CALL OrtxDetokenizeCached(const OrtxTokenizer* tokenizer, Or
 extError_t ORTX_API_CALL OrtxApplyChatTemplate(const OrtxTokenizer* tokenizer, const char* template_str,
                                                const char* input, OrtxStringArray** output,
                                                bool add_generation_prompt) {
-  if (tokenizer == nullptr || output == nullptr) {
+  if (tokenizer == nullptr && template_str == nullptr) {
     ReturnableStatus::last_error_message_ = "both tokenizer and template_str are null, no template to apply";
     return kOrtxErrorInvalidArgument;
   }