From 6d93d9ecc294d03d515a94e37f9eaa21fbcbbc42 Mon Sep 17 00:00:00 2001
From: ExtReMLapin <3909752+ExtReMLapin@users.noreply.github.com>
Date: Mon, 12 Aug 2024 08:10:40 +0200
Subject: [PATCH] Added back from_file method to LlamaGrammar

---
 llama_cpp/llama_grammar.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/llama_cpp/llama_grammar.py b/llama_cpp/llama_grammar.py
index bc2a9cc45..4cd52c2d5 100644
--- a/llama_cpp/llama_grammar.py
+++ b/llama_cpp/llama_grammar.py
@@ -1,6 +1,7 @@
 """Python implementation of llama grammar parser directly translated from C++ source file in vendor/llama.cpp/common/grammar-parser.cpp."""
 
 # flake8: noqa
+from pathlib import Path
 import sys
 import ctypes
 import enum
@@ -893,6 +894,23 @@ def from_string(cls, grammar: str, verbose: bool = True) -> "LlamaGrammar":
         if verbose:
             print_grammar(file=sys.stdout, state=parsed_grammar)
         return cls(parsed_grammar)
+    
+    @classmethod
+    def from_file(cls, file: Union[str, Path], verbose: bool = True) -> "LlamaGrammar":
+        try:
+            with open(file) as f:
+                grammar = f.read()
+        except Exception as err:
+            raise Exception(
+                f"{cls.from_file.__name__}: error reading grammar file: {err}"
+            )
+
+        if grammar:
+            return cls.from_string(grammar, verbose=verbose)
+
+        raise ValueError(
+            f"{cls.from_file.__name__}: error parsing grammar file: params_grammer is empty"
+        )
 
     @classmethod
     def from_json_schema(cls, json_schema: str, verbose: bool = True) -> "LlamaGrammar":