Workflow is almost done.

This commit is contained in:
Liu Zhicong
2023-09-01 00:39:36 -07:00
parent 27f163ddd3
commit b4990e4c57
11 changed files with 469 additions and 175 deletions
+8 -2
View File
@@ -6,6 +6,7 @@ class AIRole:
def __init__(self) -> None:
self.agent_instance_id : str = None
self.role_name : str = None
self.role_id :str = None # $workflow_id.$sub_workflow_id.$role_name
self.fullname : str = None
self.agent_name : str = None
self.prompt : AgentPrompt = None
@@ -19,6 +20,7 @@ class AIRole:
return False
self.role_name = name_node
agent_id_node = config.get("agent")
if agent_id_node is None:
logging.error("agent id is not found!")
@@ -35,7 +37,10 @@ class AIRole:
intro_node = config.get("intro")
if intro_node is not None:
self.introduce = intro_node
def get_role_id(self) -> str:
return self.role_id
def get_intro(self) -> str:
return self.introduce
@@ -48,6 +53,7 @@ class AIRole:
class AIRoleGroup:
def __init__(self) -> None:
self.roles : dict[str,AIRole] = {}
self.owner_name : str = None
def load_from_config(self,config:dict) -> bool:
for k,v in config.items():
@@ -55,7 +61,7 @@ class AIRoleGroup:
if role.load_from_config(v) is False:
logging.error(f"load role {k} failed!")
return False
role.role_id = self.owner_name + "." + k
self.roles[k] = role
return True