fix: check setting none

This commit is contained in:
superobk
2024-05-24 11:22:47 +08:00
parent 6c0af3ac35
commit 7acac48ba1

View File

@ -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 = []