fix actions/roles ser&deser

This commit is contained in:
better629 2023-11-30 19:31:26 +08:00
parent a01766ae72
commit a6510c44fc
3 changed files with 12 additions and 11 deletions

View file

@ -117,19 +117,18 @@ class SearchAndSummarize(Action):
@root_validator
def validate_engine_and_run_func(cls, values):
engine = values.get('engine')
search_func = values.get('search_func')
engine = values.get("engine")
search_func = values.get("search_func")
config = Config()
if engine is None:
engine = config.search_engine
config_data = {
'engine': engine,
'run_func': search_func
}
search_engine = SearchEngine(**config_data)
try:
search_engine = SearchEngine(engine=engine, run_func=search_func)
except pydantic.ValidationError:
search_engine = None
values['search_engine'] = search_engine
values["search_engine"] = search_engine
return values
async def run(self, context: list[Message], system_text=SEARCH_AND_SUMMARIZE_SYSTEM) -> str: