Spaces:

Felix5572
/

ChuanhuChatGPT

Sleeping

Tuchuanhuhuhu commited on Aug 25, 2023

Commit

785acc6

1 Parent(s): bedaaff

bugfix: 修复模型类型判断错误的 bug

Files changed (1) hide show

modules/models/base_model.py CHANGED Viewed

@@ -166,7 +166,7 @@ class ModelType(Enum):
             model_type = ModelType.ChuanhuAgent
         elif "palm" in model_name_lower:
             model_type = ModelType.GooglePaLM
-        elif "azure" or "api" in model_name_lower:
             model_type = ModelType.LangchainChat
         else:
             model_type = ModelType.Unknown

             model_type = ModelType.ChuanhuAgent
         elif "palm" in model_name_lower:
             model_type = ModelType.GooglePaLM
+        elif "azure" in model_name_lower or "api" in model_name_lower:
             model_type = ModelType.LangchainChat
         else:
             model_type = ModelType.Unknown