full chat implemented

This commit is contained in:
Georg Bauer
2026-07-24 19:59:41 +02:00
parent 36946e6e5f
commit 3154f57a2f
4 changed files with 615 additions and 15 deletions

View File

@@ -209,6 +209,24 @@ impl Tokenizer {
output
}
pub(super) fn encode_continuation(&self, prompt: &str, reasoning: ReasoningMode) -> Vec<i32> {
let mut output = Vec::new();
if self.family == ModelFamily::DeepSeek {
output.push(self.eos);
}
output.push(self.user);
output.extend(self.tokenize(prompt));
output.push(self.assistant);
if reasoning != ReasoningMode::Direct {
output.push(self.think_start);
} else if self.family == ModelFamily::Glm {
output.extend([self.think_start, self.think_end]);
} else {
output.push(self.think_end);
}
output
}
pub(super) fn eos(&self) -> i32 {
self.eos
}