Merge pull request #24 from google/token-getters

add chat_template::{bos_token,eos_token}()
google · Jan 20, 2025 · b8437df · b8437df
2 parents 51a4a07 + 84f63ef
commit b8437df
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/include/minja/chat-template.hpp b/include/minja/chat-template.hpp
@@ -113,6 +113,8 @@ class chat_template {
     }
 
     const std::string & source() const { return source_; }
+    const std::string & bos_token() const { return bos_token_; }
+    const std::string & eos_token() const { return eos_token_; }
     bool supports_tools() const { return supports_tools_; }
     bool supports_parallel_tool_calls() const { return supports_parallel_tool_calls_; }