feat: mypy for all type check (#10921)
This commit is contained in:
parent
c91e8b1737
commit
56e15d09a9
584 changed files with 3975 additions and 2826 deletions
|
|
@ -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":
|
||||
|
|
|
|||
|
|
@ -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:
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue