Spaces:
Runtime error
Runtime error
File size: 1,953 Bytes
7b2e5db |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
from research_assistant.constants import CONFIG_FILE_PATH
from research_assistant.entity import (
ArticleSearchConfig,
PlannerConfig,
PlannerParserConfig,
QaToolConfig,
SolverConfig,
SubmittedDateConfig,
articleLoaderConfig,
)
from research_assistant.utils.common import read_yaml
class ConfigurationManager:
def __init__(self, config_filepath=CONFIG_FILE_PATH):
self.config = read_yaml(config_filepath)
def get_article_details_config(self) -> articleLoaderConfig:
config = self.config.article_details
return articleLoaderConfig(
file_path=config.file_path, summary_save_dir=config.summary_save_dir
)
def get_article_search_params(self) -> ArticleSearchConfig:
config = self.config.article_search_details
return ArticleSearchConfig(
search_terms=config.search_terms,
num_results=config.num_results,
date_range=SubmittedDateConfig(
start_date=config.date_range.start_date,
end_date=config.date_range.end_date,
),
sort_by=config.sort_by,
sort_order=config.sort_order,
)
def get_planner_config(self) -> PlannerConfig:
config = self.config.planner
return PlannerConfig(model_name=config.model_name)
def get_qa_tool_config(self) -> QaToolConfig:
config = self.config.qa_tool
return QaToolConfig(model_name=config.model_name)
def get_solver_config(self) -> SolverConfig:
config = self.config.solver
return SolverConfig(model_name=config.model_name)
def get_planner_parser_config(self) -> PlannerParserConfig:
config = self.config.planner_parser
return PlannerParserConfig(
tool_list_model=config.tool_list_model,
argument_list_model=config.argument_list_model,
dependency_list_model=config.dependency_list_model,
)
|