feat: mypy for all type check (#10921)

This commit is contained in:
yihong
2024-12-24 18:38:51 +08:00
committed by GitHub
parent c91e8b1737
commit 56e15d09a9
584 changed files with 3975 additions and 2826 deletions

View File

@@ -66,6 +66,8 @@ class DatasetConfigManager:
dataset_configs = config.get("dataset_configs")
else:
dataset_configs = {"retrieval_model": "multiple"}
if dataset_configs is None:
return None
query_variable = config.get("dataset_query_variable")
if dataset_configs["retrieval_model"] == "single":

View File

@@ -94,7 +94,7 @@ class ModelConfigManager:
config["model"]["completion_params"]
)
return config, ["model"]
return dict(config), ["model"]
@classmethod
def validate_model_completion_params(cls, cp: dict) -> dict:

View File

@@ -7,10 +7,10 @@ class OpeningStatementConfigManager:
:param config: model config args
"""
# opening statement
opening_statement = config.get("opening_statement")
opening_statement = config.get("opening_statement", "")
# suggested questions
suggested_questions_list = config.get("suggested_questions")
suggested_questions_list = config.get("suggested_questions", [])
return opening_statement, suggested_questions_list