diff --git a/LLama/Exceptions/RuntimeError.cs b/LLama/Exceptions/RuntimeError.cs
index 4db77911e..bf20543ca 100644
--- a/LLama/Exceptions/RuntimeError.cs
+++ b/LLama/Exceptions/RuntimeError.cs
@@ -58,6 +58,25 @@ public LLamaDecodeError(DecodeResult returnCode)
}
}
+///
+/// `llama_decode` return a non-zero status code
+///
+public class MissingTemplateException
+ : RuntimeError
+{
+ ///
+ public MissingTemplateException()
+ : base("llama_chat_apply_template failed: template not found")
+ {
+ }
+
+ ///
+ public MissingTemplateException(string message)
+ : base($"llama_chat_apply_template failed: template not found for '{message}'")
+ {
+ }
+}
+
///
/// `llama_get_logits_ith` returned null, indicating that the index was invalid
///
diff --git a/LLama/LLamaTemplate.cs b/LLama/LLamaTemplate.cs
index 8f9c9daa2..e82cbccb4 100644
--- a/LLama/LLamaTemplate.cs
+++ b/LLama/LLamaTemplate.cs
@@ -2,6 +2,7 @@
using System.Collections.Generic;
using System.Diagnostics;
using System.Text;
+using LLama.Exceptions;
using LLama.Native;
namespace LLama;
@@ -250,6 +251,19 @@ public ReadOnlySpan Apply()
{
// Run templater and discover true length
var outputLength = ApplyInternal(_nativeChatMessages.AsSpan(0, Count), output);
+
+ // if we have a return code of -1, the template was not found.
+ if (outputLength == -1)
+ {
+ if (_customTemplate != null)
+ {
+ throw new MissingTemplateException(Encoding.GetString(_customTemplate));
+ }
+ else
+ {
+ throw new MissingTemplateException();
+ }
+ }
// If length was too big for output buffer run it again
if (outputLength > output.Length)