Merge pull request #131 from haesleinhuepf/reset-history

delete chat history when initializing
haesleinhuepf · May 17, 2024 · c4c79dc · c4c79dc
2 parents dbab8e5 + 4877dd2
commit c4c79dc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/bia_bob/_machinery.py b/src/bia_bob/_machinery.py
@@ -117,7 +117,7 @@ def combine_user_input(line, cell):
 
 
 def init_assistant(model=DEFAULT_MODEL, auto_execute:bool = False, variables:dict=None, endpoint=None, api_key=None,
-                   vision_model=DEFAULT_VISION_MODEL):
+                   vision_model=DEFAULT_VISION_MODEL, keep_history:bool=False):
     """Initialises the assistant.
 
     Parameters
@@ -142,6 +142,9 @@ def init_assistant(model=DEFAULT_MODEL, auto_execute:bool = False, variables:dic
     else:
         Context.variables = variables
 
+    if not keep_history:
+        Context.chat = []
+
     endpoint, api_key = correct_endpoint(endpoint, api_key)
 
     Context.endpoint = endpoint