Update utils/helper.py

utils/helper.py

@@ -3,10 +3,11 @@ from openai import OpenAI
 
 
 class ChatBot:
-    def __init__(self, protocol: str = "You are a helpful assistant."):
+    def __init__(self, protocol: str = "You are a helpful assistant.", conversation: list[dict]):
         self.client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
         self.protocol = protocol
-        self.history = [{"role": "system", "content": self.protocol}]
+        self.conversation = conversation
+        self.history = [{"role": "system", "content": self.protocol}] + self.conversation
 
     def generate_response(self, prompt: str) -> str:
         self.history.append({"role": "user", "content": prompt})