add config
This commit is contained in:
@@ -17,6 +17,7 @@ from src.classes.typings import ACTION_NAME, ACTION_PARAMS, ACTION_PAIR
|
||||
from src.classes.ai import AI, RuleAI, LLMAI
|
||||
from src.classes.persona import Persona, personas_by_id
|
||||
from src.utils.id_generator import get_avatar_id
|
||||
from src.utils.config import CONFIG
|
||||
|
||||
class Gender(Enum):
|
||||
MALE = "male"
|
||||
@@ -62,8 +63,10 @@ class Avatar:
|
||||
在Avatar创建后自动初始化tile和AI
|
||||
"""
|
||||
self.tile = self.world.map.get_tile(self.pos_x, self.pos_y)
|
||||
self.ai = LLMAI(self)
|
||||
# self.ai = RuleAI(self)
|
||||
if CONFIG.ai.mode == "llm":
|
||||
self.ai = LLMAI(self)
|
||||
else:
|
||||
self.ai = RuleAI(self)
|
||||
|
||||
def __str__(self) -> str:
|
||||
"""
|
||||
|
||||
@@ -30,3 +30,5 @@ personas_by_name[p4.name] = p4
|
||||
personas_by_name[p5.name] = p5
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user