Add feature: Agent/Role enable_function config.

This commit is contained in:
Liu Zhicong
2023-09-21 00:39:39 -07:00
parent 080a8918fe
commit 97f403080e
4 changed files with 26 additions and 5 deletions
+4
View File
@@ -12,6 +12,7 @@ class AIRole:
self.prompt : AgentPrompt = None
self.introduce : str = None
self.agent = None
self.enable_function_list : list[str] = None
def load_from_config(self,config:dict) -> bool:
name_node = config.get("name")
@@ -37,6 +38,9 @@ class AIRole:
intro_node = config.get("intro")
if intro_node is not None:
self.introduce = intro_node
if config.get("enable_function") is not None:
self.enable_function_list = config["enable_function"]
def get_role_id(self) -> str:
return self.role_id