From 2721f7bfe0d40d3ece5a639799aee7caa258b42e Mon Sep 17 00:00:00 2001
From: Mitchell Shibilski-Unkel
 <133730558+MitchellShibilski-Unkel@users.noreply.github.com>
Date: Wed, 20 Mar 2024 18:08:11 -0600
Subject: [PATCH] Update pyai.py

---
 pyai.py | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/pyai.py b/pyai.py
index d33100b..dcf4013 100644
--- a/pyai.py
+++ b/pyai.py
@@ -34,6 +34,22 @@ def __init__(self, text: str):
         self._present = ["is", "has"]
         self._future = ["will", "shall"]
 
+    def setTokensTo(self, letters: bool, *words: bool, **sentences: bool):
+        self.tokens = []
+
+        if letters:
+            tokens = iter(self.sentences)
+            for t in tokens:
+                self.tokens.append(t)
+        elif words:
+            for t in self.words:
+                self.tokens.append(t)
+        elif sentences:
+            for t in self.sentences:
+                self.tokens.append(t)
+        else:
+            self.tokens.append("ERROR")
+
     def getTense(self):
         self.past = False
         self.present = False
@@ -54,4 +70,7 @@ def getWords(self):
         return self.words
 
     def getSentences(self):
-        return self.sentences
\ No newline at end of file
+        return self.sentences
+    
+    def getTokens(self):
+        return self.tokens