use bot_in.type instead of bot.type

Author: lwj2015

kubechat/views/main.py

@@ -640,7 +640,7 @@ async def create_bot(request, bot_in: BotIn):
     memory = config.get("memory", False)
     model = config.get("model")
     llm_config = config.get("llm")
-    valid, msg = validate_bot_config(model, llm_config, bot, memory)
+    valid, msg = validate_bot_config(model, llm_config, bot_in.type, memory)
     if not valid:
         return fail(HTTPStatus.BAD_REQUEST, msg)
     await bot.asave()
@@ -715,7 +715,7 @@ async def update_bot(request, bot_id, bot_in: BotIn):
     model = new_config.get("model")
     memory = new_config.get("memory", False)
     llm_config = new_config.get("llm")
-    valid, msg = validate_bot_config(model, llm_config, bot, memory)
+    valid, msg = validate_bot_config(model, llm_config, bot_in.type, memory)
     if not valid:
         return fail(HTTPStatus.BAD_REQUEST, msg)
     old_config = json.loads(bot.config)

kubechat/views/utils.py

@@ -82,7 +82,7 @@ def validate_source_connect_config(config: Dict) -> (bool, str):
     return True, ""
 
 
-def validate_bot_config(model, config: Dict, bot, memory) -> (bool, str):
+def validate_bot_config(model, config: Dict, type, memory) -> (bool, str):
     try:
         Predictor.from_model(model, PredictorType.CUSTOM_LLM, **config)
     except Exception as e:
@@ -91,9 +91,9 @@ def validate_bot_config(model, config: Dict, bot, memory) -> (bool, str):
     try:
         # validate the prompt
         prompt_template = config.get("prompt_template", None)
-        if prompt_template and bot.type == BotType.KNOWLEDGE:
+        if prompt_template and type == BotType.KNOWLEDGE:
             PromptTemplate(template=prompt_template, input_variables=["query", "context"])
-        elif prompt_template and bot.type == BotType.COMMON:
+        elif prompt_template and type == BotType.COMMON:
             PromptTemplate(template=prompt_template, input_variables=["query"])
             # pass
     except ValidationError:
@@ -103,9 +103,9 @@ def validate_bot_config(model, config: Dict, bot, memory) -> (bool, str):
         try:
             # validate the memory prompt
             prompt_template = config.get("memory_prompt_template", None)
-            if prompt_template and bot.type == BotType.KNOWLEDGE:
+            if prompt_template and type == BotType.KNOWLEDGE:
                 PromptTemplate(template=prompt_template, input_variables=["query", "context"])
-            elif prompt_template and bot.type == BotType.COMMON:
+            elif prompt_template and type == BotType.COMMON:
                 PromptTemplate(template=prompt_template, input_variables=["query"])
                 # pass
         except ValidationError: