From 6cf58765388b85329769cb78405c5e5ff74dc414 Mon Sep 17 00:00:00 2001
From: Andrei Betlen <abetlen@gmail.com>
Date: Wed, 12 Apr 2023 14:06:04 -0400
Subject: [PATCH] Deprecate generate method

---
 llama_cpp/llama.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/llama_cpp/llama.py b/llama_cpp/llama.py
index 3e13776..69f7680 100644
--- a/llama_cpp/llama.py
+++ b/llama_cpp/llama.py
@@ -3,6 +3,7 @@ import sys
 import uuid
 import time
 import math
+import warnings
 import multiprocessing
 from typing import List, Optional, Union, Generator, Sequence, Iterator
 from collections import deque
@@ -239,6 +240,11 @@ class Llama:
         Yields:
             The generated tokens.
         """
+        warnings.warn(
+            "Llama.generate is deprecated and will be removed in v0.2.0",
+            DeprecationWarning,
+            stacklevel=2,
+        )
         assert self.ctx is not None
         self.reset()
         while True: