From 4877dd21331b354c8a03c59eb47787d8b15378d5 Mon Sep 17 00:00:00 2001 From: Robert Haase Date: Fri, 17 May 2024 13:22:35 +0200 Subject: [PATCH] delete chat history when initializing --- src/bia_bob/_machinery.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/bia_bob/_machinery.py b/src/bia_bob/_machinery.py index 1079265..c77e243 100644 --- 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