diff --git a/src/blackbox/chat.py b/src/blackbox/chat.py index 7ddc52b..0ba8c68 100644 --- a/src/blackbox/chat.py +++ b/src/blackbox/chat.py @@ -30,19 +30,20 @@ class Chat(Blackbox): @logging_time(logger=logger) def processing(self, *args, **kwargs) -> str: - data = args[0] - - user_model_name = data.get("model_name") - user_context = data.get("context") - user_question = data.get("question") - user_template = data.get("template") - user_temperature = data.get("temperature") - user_top_p = data.get("top_p") - user_n = data.get("n") - user_max_tokens = data.get("max_tokens") - user_stop = data.get("stop") - user_frequency_penalty = data.get("frequency_penalty") - user_presence_penalty = data.get("presence_penalty") + settings: dict = args[0] + if settings is None: + settings = {} + user_model_name = settings.get("model_name") + user_context = settings.get("context") + user_question = settings.get("question") + user_template = settings.get("template") + user_temperature = settings.get("temperature") + user_top_p = settings.get("top_p") + user_n = settings.get("n") + user_max_tokens = settings.get("max_tokens") + user_stop = settings.get("stop") + user_frequency_penalty = settings.get("frequency_penalty") + user_presence_penalty = settings.get("presence_penalty") if user_context == None: user_context = []