ivanwu/jarvis-models
1
0
Fork 0
mirror of https://github.com/BoardWare-Genius/jarvis-models.git synced 2025-12-13 16:53:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: check setting none

Browse Source
This commit is contained in:
superobk
2024-05-24 11:22:47 +08:00
parent 6c0af3ac35
commit 7acac48ba1
1 changed files with 14 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
src/blackbox/chat.py
View File

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