def get_location_by_id(self, memid: str) -> "LocationNode":
"""Given memory ID, retrieve LocationNode
Args:
memid (string): Memory ID
"""
return LocationNode(self, memid)
def add_location(self, xyz: XYZ) -> str:
"""Create a new LocationNode
Args:
xyz (tuple): XYZ coordinates
"""
return LocationNode.create(self, xyz)
def test_get_node_from_memid(self):
self.memory = AgentMemory()
joe_memid = PlayerNode.create(
self.memory, Player(10, "joe", Pos(1, 0, 1), Look(0, 0)))
assert self.memory.get_node_from_memid(joe_memid) == "Player"
loc_memid = LocationNode.create(self.memory, (0, 0, 0))
assert self.memory.get_node_from_memid(loc_memid) == "Location"
def test_check_memid_exists(self):
self.memory = AgentMemory()
chat_memid = ChatNode.create(self.memory,
speaker="dfghjk123445",
chat="hi there!")
loc_memid = LocationNode.create(self.memory, (0, 2, 0))
assert self.memory.check_memid_exists(chat_memid, "Chats") == True
assert self.memory.check_memid_exists(loc_memid,
"ReferenceObjects") == True
def __init__(self, agent, task_data):
super().__init__(agent)
assert task_data.get("target") is not None
loc = self.target_to_memory(task_data["target"])
loc_memid = LocationNode.create(agent.memory, loc)
TripleNode.create(agent.memory,
subj=self.memid,
pred_text="task_reference_object",
obj=loc_memid)
def test_tag(self):
self.memory = AgentMemory()
sheep_memid = NamedAbstractionNode.create(self.memory, "sheep")
loc_memid = LocationNode.create(self.memory, (1, 2, 3))
self.memory.tag(subj_memid=sheep_memid, tag_text="furry")
self.memory.add_triple(subj=sheep_memid,
pred_text="has_home_location",
obj=loc_memid)
self.memory.add_triple(subj=sheep_memid,
pred_text="has_fur_color",
obj_text="white")
assert len(self.memory.get_triples(subj=sheep_memid)) == 3
assert len(
self.memory.get_triples(subj=sheep_memid,
pred_text="has_home_location")) == 1