provider(base): add chunk (#13)

This commit is contained in:
0xMRTT 2023-05-06 00:41:22 +02:00
parent e306d8dd2d
commit b2fd2566ab
Signed by: 0xmrtt
GPG key ID: 19C1449A774028BD

View file

@ -71,3 +71,10 @@ class BavarderProvider:
def load(self, data):
raise NotImplementedError()
def chunk(self, prompt, n=4000):
if len(prompt) > n:
print("Chuncking prompt")
prompt = [(prompt[i:i+n]) for i in range(0, len(prompt), n)]
return prompt