diff --git a/llama_cpp/llama.py b/llama_cpp/llama.py index e53c9c8..fb2a541 100644 --- a/llama_cpp/llama.py +++ b/llama_cpp/llama.py @@ -921,6 +921,7 @@ class Llama: grammar: Optional[LlamaGrammar] = None, ) -> Union[Iterator[Completion], Iterator[CompletionChunk]]: assert self.ctx is not None + assert suffix is None or suffix.__class__ is str completion_id: str = f"cmpl-{str(uuid.uuid4())}" created: int = int(time.time())