diff --git a/network-poc/native-node/src/inference.rs b/network-poc/native-node/src/inference.rs index d2ea366..3915d60 100644 --- a/network-poc/native-node/src/inference.rs +++ b/network-poc/native-node/src/inference.rs @@ -207,8 +207,8 @@ impl LlmEngine { // Stop-sekvenssit: katkaistaan kun malli alkaa selittää let lower = generated_text.to_lowercase(); - if lower.contains("\n###") || lower.contains("\nexplanation") || lower.contains("\nnote:") || lower.contains("\noutput:") || lower.contains("\n```\n\n") { - for stop in &["\n###", "\nExplanation", "\nNote:", "\nOutput:", "\n```\n\n"] { + if lower.contains("\n###") || lower.contains("\nexplanation") || lower.contains("\nnote:") || lower.contains("\noutput:") || lower.contains("\n```\n\n") || lower.contains("\n// example") || lower.contains("\n# example") { + for stop in &["\n###", "\nExplanation", "\nNote:", "\nOutput:", "\n```\n\n", "\n// Example", "\n// example", "\n# Example", "\n# example"] { if let Some(pos) = generated_text.find(stop) { generated_text.truncate(pos); } diff --git a/network-poc/node/src/qwen_coder.rs b/network-poc/node/src/qwen_coder.rs index 8e2c85a..363bf15 100644 --- a/network-poc/node/src/qwen_coder.rs +++ b/network-poc/node/src/qwen_coder.rs @@ -342,8 +342,8 @@ pub async fn run_coder_inference(prompt: String, ws: Rc>, use // Stop-sekvenssit: katkaistaan kun malli alkaa selittää let lower = generated_text.to_lowercase(); - if lower.contains("\n###") || lower.contains("\nexplanation") || lower.contains("\nnote:") || lower.contains("\noutput:") || lower.contains("\n```\n\n") { - for stop in &["\n###", "\nExplanation", "\nNote:", "\nOutput:", "\n```\n\n"] { + if lower.contains("\n###") || lower.contains("\nexplanation") || lower.contains("\nnote:") || lower.contains("\noutput:") || lower.contains("\n```\n\n") || lower.contains("\n// example") || lower.contains("\n# example") { + for stop in &["\n###", "\nExplanation", "\nNote:", "\nOutput:", "\n```\n\n", "\n// Example", "\n// example", "\n# Example", "\n# example"] { if let Some(pos) = generated_text.find(stop) { generated_text.truncate(pos); }