From 463628372d5fe3a0c1e5864aa5fc57deb7387039 Mon Sep 17 00:00:00 2001
From: Clint Herron <hanclinto@gmail.com>
Date: Wed, 13 Mar 2024 14:10:40 -0400
Subject: [PATCH] grammar : handle missing "root" node (#6004)

---
 common/sampling.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/common/sampling.cpp b/common/sampling.cpp
index 823031feb..5a5450982 100644
--- a/common/sampling.cpp
+++ b/common/sampling.cpp
@@ -17,6 +17,13 @@ struct llama_sampling_context * llama_sampling_init(const struct llama_sampling_
             return nullptr;
         }
 
+        // Ensure that there is a "root" node.
+        if (result->parsed_grammar.symbol_ids.find("root") == result->parsed_grammar.symbol_ids.end()) {
+            fprintf(stderr, "%s: grammar does not contain a 'root' symbol\n", __func__);
+            delete result;
+            return nullptr;
+        }
+
         std::vector<const llama_grammar_element *> grammar_rules(result->parsed_grammar.c_rules());
 
         result->grammar = llama_grammar_init(