Liu Zhicong
66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
import logging
|
|
|
|
from .agent import AIAgent,AgentPrompt
|
|
|
|
class AIRole:
|
|
def __init__(self) -> None:
|
|
self.agent_instance_id : str = None
|
|
self.role_name : str = None
|
|
self.fullname : str = None
|
|
self.agent_name : str = None
|
|
self.prompt : AgentPrompt = None
|
|
self.introduce : str = None
|
|
self.agent = None
|
|
|
|
def load_from_config(self,config:dict) -> bool:
|
|
name_node = config.get("name")
|
|
if name_node is None:
|
|
logging.error("role name is not found!")
|
|
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!")
|
|
return False
|
|
self.agent_name = agent_id_node
|
|
|
|
prompt_node = config.get("prompt")
|
|
if prompt_node:
|
|
self.prompt = AgentPrompt()
|
|
if self.prompt.load_from_config(prompt_node) is False:
|
|
logging.error("load prompt failed!")
|
|
return False
|
|
|
|
intro_node = config.get("intro")
|
|
if intro_node is not None:
|
|
self.introduce = intro_node
|
|
|
|
def get_intro(self) -> str:
|
|
return self.introduce
|
|
|
|
def get_name(self) -> str:
|
|
return self.role_name
|
|
|
|
def get_prompt(self) -> AgentPrompt:
|
|
return self.prompt
|
|
|
|
class AIRoleGroup:
|
|
def __init__(self) -> None:
|
|
self.roles : dict[str,AIRole] = {}
|
|
|
|
def load_from_config(self,config:dict) -> bool:
|
|
for k,v in config.items():
|
|
role = AIRole()
|
|
if role.load_from_config(v) is False:
|
|
logging.error(f"load role {k} failed!")
|
|
return False
|
|
|
|
self.roles[k] = role
|
|
|
|
return True
|
|
|
|
def get(self,role_name:str) -> AIRole:
|
|
return self.roles.get(role_name)
|
|
|
|
|