Update utils/helper.py

utils/helper.py

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