diff --git a/config/config.yaml b/config/config.yaml
index 444f55efd..e76deac50 100644
--- a/config/config.yaml
+++ b/config/config.yaml
@@ -7,7 +7,7 @@
## Or, you can configure OPENAI_PROXY to access official OPENAI_API_BASE.
OPENAI_API_BASE: "https://api.openai.com/v1"
#OPENAI_PROXY: "http://127.0.0.1:8118"
-#OPENAI_API_KEY: "YOUR_API_KEY"
+#OPENAI_API_KEY: "YOUR OPENAI KEY"
OPENAI_API_MODEL: "gpt-4"
MAX_TOKENS: 1500
RPM: 10
diff --git a/examples/__init__.py b/examples/__init__.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/examples/st_game/actions/inner_voice_action.py b/examples/st_game/actions/inner_voice_action.py
new file mode 100644
index 000000000..082921554
--- /dev/null
+++ b/examples/st_game/actions/inner_voice_action.py
@@ -0,0 +1,36 @@
+import re
+from examples.st_game.actions.st_action import STAction
+from examples.st_game.memory.agent_memory import BasicMemory
+from metagpt.logs import logger
+
+class AgentWhisperThoughtAction(STAction):
+
+ def __init__(self, name="AgentWhisperThoughtAction", context: list[BasicMemory] = None, llm=None):
+ super().__init__(name, context, llm)
+
+ def _func_validate(self, llm_resp: str, prompt: str) -> bool:
+ try:
+ self._func_cleanup(llm_resp, prompt)
+ return True
+ except:
+ return False
+
+ def _func_cleanup(self, llm_resp: str, prompt: str = "") -> list:
+ return llm_resp.split('"')[0].strip()
+
+ def _func_fail_default_resp(self) -> str:
+ pass
+
+ async def run(self, role: "STRole", statements: str, test_input=None, verbose=False) -> str:
+ def create_prompt_input(role: "STRole", statements, test_input=None):
+ prompt_input = [role.scratch.name, statements]
+ return prompt_input
+
+ prompt_input = create_prompt_input(role, statements)
+ prompt = self.generate_prompt_with_tmpl_filename(prompt_input,
+ "whisper_inner_thought_v1.txt")
+
+ output = await self._run_v1(prompt)
+ logger.info(f"Run action: {self.__class__.__name__} with result: {output}")
+ return output
+
diff --git a/examples/st_game/actions/run_reflect_action.py b/examples/st_game/actions/run_reflect_action.py
index ab83a22ee..d29b654f8 100644
--- a/examples/st_game/actions/run_reflect_action.py
+++ b/examples/st_game/actions/run_reflect_action.py
@@ -3,6 +3,7 @@
# @Desc : Integration Reflect Action
import re
+
from metagpt.logs import logger
from examples.st_game.actions.st_action import STAction
@@ -23,16 +24,18 @@ class AgentFocusPt(STAction):
return False
def _func_cleanup(self, llm_resp: str, prompt: str = "") -> str:
- llm_resp = "1) " + llm_resp.strip()
- ret = []
- for i in llm_resp.split("\n"):
- ret += [i.split(") ")[-1]]
- return ret
+ try:
+ """
+ Cleanup handling has been completed for run_v2
+ """
+ return llm_resp
+ except Exception as exp:
+ logger.error(f"{self.__class__.__name__} with error {exp}")
def _func_fail_default_resp(self) -> str:
pass
- async def run(self, role: "STRole", statements: str, n: int, test_input=None) -> str:
+ def run(self, role: "STRole", statements: str, n: int, test_input=None) -> str:
def create_prompt_input(role: "STRole", statements, n, test_input=None):
prompt_input = [statements, str(n)]
return prompt_input
@@ -43,9 +46,9 @@ class AgentFocusPt(STAction):
example_output = '["What should Jane do for lunch", "Does Jane like strawberry", "Who is Jane"]'
special_instruction = "Output must be a list of str."
- output = await self._run_v2(prompt,
- example_output,
- special_instruction)
+ output = self._run_v2(prompt,
+ example_output,
+ special_instruction)
logger.info(f"Run action: {self.__class__.__name__} with result: {output}")
return output
@@ -63,23 +66,26 @@ class AgentInsightAndGuidance(STAction):
except:
return False
- def _func_cleanup(self, llm_resp: str, prompt: str = "") -> str:
- llm_resp = "1. " + llm_resp.strip()
- ret = dict()
- for i in llm_resp.split("\n"):
- row = i.split(". ")[-1]
- thought = row.split("(because of ")[0].strip()
- evi_raw = row.split("(because of ")[1].split(")")[0].strip()
- evi_raw = re.findall(r'\d+', evi_raw)
- evi_raw = [int(i.strip()) for i in evi_raw]
- ret[thought] = evi_raw
- return ret
+ def _func_cleanup(self, llm_resp: str, prompt: str = "") -> dict:
+ try:
+ llm_resp = "1. " + llm_resp.strip()
+ ret = dict()
+ for i in llm_resp.split("\n"):
+ row = i.split(". ")[-1]
+ thought = row.split("(because of ")[0].strip()
+ evi_raw = row.split("(because of ")[1].split(")")[0].strip()
+ evi_raw = re.findall(r'\d+', evi_raw)
+ evi_raw = [int(i.strip()) for i in evi_raw]
+ ret[thought] = evi_raw
+ return ret
+ except Exception as exp:
+ logger.error(f"{self.__class__.__name__} with error {exp}")
def _func_fail_default_resp(self) -> str:
pass
- async def run(self, role: "STRole", statements: str, n: int, test_input=None) -> str:
- def create_prompt_input(role: "STRole", statements, n, test_input=None):
+ def run(self, role: "STRole", statements: str, n: int, test_input=None) -> dict:
+ def create_prompt_input(role, statements, n, test_input=None):
prompt_input = [statements, str(n)]
return prompt_input
@@ -87,7 +93,7 @@ class AgentInsightAndGuidance(STAction):
prompt = self.generate_prompt_with_tmpl_filename(prompt_input,
"insight_and_evidence_v1.txt")
- output = await self._run_v1(prompt)
+ output = self._run_v1(prompt)
logger.info(f"Run action: {self.__class__.__name__} with result: {output}")
return output
@@ -106,31 +112,37 @@ class AgentEventTriple(STAction):
return False
return True
- def _func_cleanup(self, llm_resp: str, prompt: str = "") -> str:
- cr = llm_resp.strip()
- cr = [i.strip() for i in cr.split(")")[0].split(",")]
- return cr
+ def _func_cleanup(self, llm_resp: str, prompt: str = "") -> list:
+ try:
+ cr = llm_resp.strip()
+ cr = [i.strip() for i in cr.split(")")[0].split(",")]
+ if len(cr) != 2:
+ return cr[-2:]
+ return cr
+ except Exception as exp:
+ logger.error(f"{self.__class__.__name__} with error {exp}")
def _func_fail_default_resp(self) -> str:
pass
- async def run(self, statements: str, role: "STRole", verbose=False) -> str:
+ def run(self, statements: str, role: "STRole", verbose=False) -> tuple:
def create_prompt_input(statements, role):
if "(" in statements:
statements = statements.split("(")[-1].split(")")[0]
- prompt_input = [role._rc.scratch.name,
+ prompt_input = [role.scratch.name,
statements,
- role._rc.scratch.name]
+ role.scratch.name]
return prompt_input
prompt_input = create_prompt_input(statements, role)
prompt = self.generate_prompt_with_tmpl_filename(prompt_input,
"generate_event_triple_v1.txt")
- output = await self._run_v1(prompt)
+ output = self._run_v1(prompt)
+ output = (role.scratch.name, output[0], output[1])
logger.info(f"Run action: {self.__class__.__name__} with result: {output}")
- return output[0]
+ return output
# Run GPT Prompt Event Poignancy
@@ -145,18 +157,21 @@ class AgentEventPoignancy(STAction):
except:
return False
- def _func_cleanup(self, llm_resp: str, prompt: str = "") -> str:
- llm_resp = int(llm_resp.strip())
- return llm_resp
+ def _func_cleanup(self, llm_resp: str, prompt: str = "") -> int:
+ try:
+ llm_resp = int(llm_resp.strip())
+ return llm_resp
+ except Exception as exp:
+ logger.error(f"{self.__class__.__name__} with error {exp}")
def _func_fail_default_resp(self) -> str:
pass
- async def run(self, role: "STRole", statements: str, test_input=None, verbose=False) -> str:
+ def run(self, role: "STRole", statements: str, test_input=None, verbose=False) -> str:
def create_prompt_input(role: "STRole", statements: str, test_input=None):
- prompt_input = [role._rc.scratch.name,
- role._rc.scratch.get_str_iss(),
- role._rc.scratch.name,
+ prompt_input = [role.scratch.name,
+ role.scratch.get_str_iss(),
+ role.scratch.name,
statements]
return prompt_input
@@ -166,9 +181,9 @@ class AgentEventPoignancy(STAction):
example_output = "5" # ########
special_instruction = "The output should ONLY contain ONE integer value on the scale of 1 to 10."
- output = await self._run_v2(prompt,
- example_output,
- special_instruction)
+ output = self._run_v2(prompt,
+ example_output,
+ special_instruction)
logger.info(f"Run action: {self.__class__.__name__} with result: {output}")
return output
@@ -186,18 +201,21 @@ class AgentChatPoignancy(STAction):
except:
return False
- def _func_cleanup(self, llm_resp: str, prompt: str = "") -> str:
- llm_resp = int(llm_resp.strip())
- return llm_resp
+ def _func_cleanup(self, llm_resp: str, prompt: str = "") -> int:
+ try:
+ llm_resp = int(llm_resp.strip())
+ return llm_resp
+ except Exception as exp:
+ logger.error(f"{self.__class__.__name__} with error {exp}")
def _func_fail_default_resp(self) -> str:
pass
- async def run(self, role: "STRole", statements: str, test_input=None, verbose=False) -> str:
+ def run(self, role: "STRole", statements: str, test_input=None, verbose=False) -> str:
def create_prompt_input(role: "STRole", statements, test_input=None):
- prompt_input = [role._rc.scratch.name,
- role._rc.scratch.get_str_iss(),
- role._rc.scratch.name,
+ prompt_input = [role.scratch.name,
+ role.scratch.get_str_iss(),
+ role.scratch.name,
statements]
return prompt_input
@@ -207,9 +225,9 @@ class AgentChatPoignancy(STAction):
example_output = "5" # ########
special_instruction = "The output should ONLY contain ONE integer value on the scale of 1 to 10."
- output = await self._run_v2(prompt,
- example_output,
- special_instruction)
+ output = self._run_v2(prompt,
+ example_output,
+ special_instruction)
logger.info(f"Run action: {self.__class__.__name__} with result: {output}")
return output
@@ -228,24 +246,27 @@ class AgentPlanThoughtOnConvo(STAction):
return False
def _func_cleanup(self, llm_resp: str, prompt: str = "") -> str:
- return llm_resp.split('"')[0].strip()
+ try:
+ return llm_resp.split('"')[0].strip()
+ except Exception as exp:
+ logger.error(f"{self.__class__.__name__} with error {exp}")
def _func_fail_default_resp(self) -> str:
pass
- async def run(self, role: "STRole", statements: str, test_input=None, verbose=False) -> str:
+ def run(self, role: "STRole", statements: str, test_input=None, verbose=False) -> str:
def create_prompt_input(role, statements, test_input=None):
prompt_input = [statements,
- role._rc.scratch.name,
- role._rc.scratch.name,
- role._rc.scratch.name]
+ role.scratch.name,
+ role.scratch.name,
+ role.scratch.name]
return prompt_input
prompt_input = create_prompt_input(role, statements)
prompt = self.generate_prompt_with_tmpl_filename(prompt_input,
"planning_thought_on_convo_v1.txt")
- output = await self._run_v1(prompt)
+ output = self._run_v1(prompt)
logger.info(f"Run action: {self.__class__.__name__} with result: {output}")
return output
@@ -264,17 +285,20 @@ class AgentMemoryOnConvo(STAction):
return False
def _func_cleanup(self, llm_resp: str, prompt: str = "") -> str:
- return llm_resp.split('"')[0].strip()
+ try:
+ return llm_resp.split('"')[0].strip()
+ except Exception as exp:
+ logger.error(f"{self.__class__.__name__} with error {exp}")
def _func_fail_default_resp(self) -> str:
pass
- async def run(self, role: "STRole", statements: str, test_input=None, verbose=False) -> str:
+ def run(self, role: "STRole", statements: str, test_input=None, verbose=False) -> str:
def create_prompt_input(role, statements, test_input=None):
prompt_input = [statements,
- role._rc.scratch.name,
- role._rc.scratch.name,
- role._rc.scratch.name]
+ role.scratch.name,
+ role.scratch.name,
+ role.scratch.name]
return prompt_input
prompt_input = create_prompt_input(role, statements)
@@ -282,9 +306,9 @@ class AgentMemoryOnConvo(STAction):
"memo_on_convo_v1.txt")
example_output = 'Jane Doe was interesting to talk to.'
special_instruction = 'The output should ONLY contain a string that summarizes anything interesting that the agent may have noticed'
- output = await self._run_v2(prompt,
- example_output,
- special_instruction)
+ output = self._run_v2(prompt,
+ example_output,
+ special_instruction)
logger.info(f"Run action: {self.__class__.__name__} with result: {output}")
return output
diff --git a/examples/st_game/actions/st_action.py b/examples/st_game/actions/st_action.py
index 60f73f36d..e5c048b4e 100644
--- a/examples/st_game/actions/st_action.py
+++ b/examples/st_game/actions/st_action.py
@@ -56,7 +56,7 @@ class STAction(Action):
def _ask(self, prompt: str, system_msgs: Optional[list[str]] = None) -> str:
return self.llm.ask(prompt)
- def _run_v1(self, prompt: str, retry: int = 3) -> str:
+ def _run_v1(self, prompt: str, retry: int = 3) :
"""
same with `gpt_structure.safe_generate_response`
default post-preprocess operations of LLM response
diff --git a/examples/st_game/memory/agent_memory.py b/examples/st_game/memory/agent_memory.py
index 60aa4ae81..869f94a4d 100644
--- a/examples/st_game/memory/agent_memory.py
+++ b/examples/st_game/memory/agent_memory.py
@@ -7,6 +7,7 @@ from datetime import datetime
from metagpt.memory.memory import Memory
from metagpt.schema import Message
+from metagpt.logs import logger
class BasicMemory(Message):
@@ -33,7 +34,7 @@ class BasicMemory(Message):
self.memory_count: int = memory_count # 第几个记忆,实际数值与Memory相等
self.type_count: int = type_count # 第几种记忆,类型为整数(具体不太理解如何生成的)
self.memory_type: str = memory_type # 记忆类型,包含 event,thought,chat三种类型
- self.depth: str = depth # 记忆深度,类型为整数
+ self.depth: int = depth # 记忆深度,类型为整数
self.created: datetime = created # 创建时间
self.expiration: datetime = expiration # 记忆失效时间,默认为空()
@@ -62,10 +63,10 @@ class BasicMemory(Message):
memory_dict[node_id] = dict()
memory_dict[node_id]["node_count"] = self.memory_count
memory_dict[node_id]["type_count"] = self.type_count
- memory_dict[node_id]["type"] = self.type
+ memory_dict[node_id]["type"] = self.memory_type
memory_dict[node_id]["depth"] = self.depth
- memory_dict[node_id]["cmemory_dicteated"] = self.created.strftime('%Y-%m-%d %H:%M:%S')
+ memory_dict[node_id]["created"] = self.created.strftime('%Y-%m-%d %H:%M:%S')
memory_dict[node_id]["expiration"] = None
if self.expiration:
memory_dict[node_id]["expiration"] = (self.expiration
@@ -75,7 +76,7 @@ class BasicMemory(Message):
memory_dict[node_id]["predicate"] = self.predicate
memory_dict[node_id]["object"] = self.object
- memory_dict[node_id]["description"] = self.description
+ memory_dict[node_id]["description"] = self.content
memory_dict[node_id]["embedding_key"] = self.embedding_key
memory_dict[node_id]["poignancy"] = self.poignancy
memory_dict[node_id]["keywords"] = list(self.keywords)
@@ -102,7 +103,7 @@ class AgentMemory(Memory):
"""
super(AgentMemory, self).__init__()
self.id_to_node = dict() # TODO jiayi add
- self.storage: list[BasicMemory] = [] # 重写Stroage,存储BasicMemory所有节点
+ self.storage: list[BasicMemory] = [] # 重写Storage,存储BasicMemory所有节点
self.event_list = [] # 存储event记忆
self.thought_list = [] # 存储thought记忆
self.chat_list = [] # chat-related memory
@@ -114,6 +115,9 @@ class AgentMemory(Memory):
self.kw_strength_event = dict() # 关键词影响存储
self.kw_strength_thought = dict()
+ self.memory_saved = None
+ self.embeddings = None
+
# self.load(memory_saved)
def set_mem_path(self, memory_saved: str):
@@ -122,13 +126,14 @@ class AgentMemory(Memory):
def save(self, memory_saved: str):
"""
- 将MemormyBasic类存储为Nodes.json形式。复现GA中的Kw Strength.json形式
+ 将MemoryBasic类存储为Nodes.json形式。复现GA中的Kw Strength.json形式
这里添加一个路径即可
+ TODO 这里在存储时候进行倒序存储,之后需要验证(test_memory通过)
"""
-
memory_json = dict()
for i in range(len(self.storage)):
- memory_node = self.storage[i]
+ memory_node = self.storage[len(self.storage)-i-1]
+ memory_node = memory_node.save_to_dict()
memory_json.update(memory_node)
with open(memory_saved + "/nodes.json", "w") as outfile:
json.dump(memory_json, outfile)
@@ -152,16 +157,13 @@ class AgentMemory(Memory):
node_id = f"node_{str(count + 1)}"
node_details = memory_load[node_id]
node_type = node_details["type"]
- created = datetime.datetime.strptime(node_details["created"],
+ created = datetime.strptime(node_details["created"],
'%Y-%m-%d %H:%M:%S')
expiration = None
if node_details["expiration"]:
- expiration = datetime.datetime.strptime(node_details["expiration"],
+ expiration = datetime.strptime(node_details["expiration"],
'%Y-%m-%d %H:%M:%S')
- if node_details["cause_by"]:
- cause_by = node_details["cause_by"]
-
s = node_details["subject"]
p = node_details["predicate"]
o = node_details["object"]
@@ -172,16 +174,16 @@ class AgentMemory(Memory):
poignancy = node_details["poignancy"]
keywords = set(node_details["keywords"])
filling = node_details["filling"]
-
+ if node_type == "thought":
+ self.add_thought(created, expiration, s, p, o,
+ description, keywords, poignancy, embedding_pair, filling)
if node_type == "event":
self.add_event(created, expiration, s, p, o,
description, keywords, poignancy, embedding_pair, filling)
- elif node_type == "chat":
+ if node_type == "chat":
self.add_chat(created, expiration, s, p, o,
- description, keywords, poignancy, embedding_pair, filling, cause_by)
- elif node_type == "thought":
- self.add_thought(created, expiration, s, p, o,
- description, keywords, poignancy, embedding_pair, filling)
+ description, keywords, poignancy, embedding_pair, filling)
+
strength_keywords_load = json.load(open(memory_saved + "/kw_strength.json"))
if strength_keywords_load["kw_strength_event"]:
@@ -194,29 +196,30 @@ class AgentMemory(Memory):
Add a new message to storage, while updating the index
重写add方法,修改原有的Message类为BasicMemory类,并添加不同的记忆类型添加方式
"""
- if memory_basic in self.storage:
+ if memory_basic.memory_id in self.storage:
return
self.storage.append(memory_basic)
- if memory_basic.cause_by:
- self.index[memory_basic.cause_by][0:0] = [memory_basic]
+ if memory_basic.memory_type == "chat":
+ self.chat_list[0:0] = [memory_basic]
return
- if memory_basic.type == "thought":
+ if memory_basic.memory_type == "thought":
self.thought_list[0:0] = [memory_basic]
return
- if memory_basic.type == "event":
+ if memory_basic.memory_type == "event":
self.event_list[0:0] = [memory_basic]
+ return
def add_chat(self, created, expiration, s, p, o,
content, keywords, poignancy,
embedding_pair, filling,
- cause_by):
+ cause_by = ''):
"""
- 调用add方法,初始化chat,在创建的时候就需要调用embeeding函数
+ 调用add方法,初始化chat,在创建的时候就需要调用embedding函数
"""
memory_count = len(self.storage) + 1
type_count = len(self.thought_list) + 1
memory_type = "chat"
- memory_id = f"memory_{str(memory_count)}"
+ memory_id = f"node_{str(memory_count)}"
depth = 1
memory_node = BasicMemory(memory_id, memory_count, type_count, memory_type, depth,
@@ -246,8 +249,8 @@ class AgentMemory(Memory):
"""
memory_count = len(self.storage) + 1
type_count = len(self.thought_list) + 1
- memory_type = "event"
- memory_id = f"memory_{str(memory_count)}"
+ memory_type = "thought"
+ memory_id = f"node_{str(memory_count)}"
depth = 1
try:
@@ -255,6 +258,7 @@ class AgentMemory(Memory):
depth_list = [memory_node.depth for memory_node in self.storage if memory_node.memory_id in filling]
depth += max(depth_list)
except Exception as exp:
+ logger.warning(f"filling init occur {exp}")
pass
memory_node = BasicMemory(memory_id, memory_count, type_count, memory_type, depth,
@@ -291,7 +295,7 @@ class AgentMemory(Memory):
memory_count = len(self.storage) + 1
type_count = len(self.event_list) + 1
memory_type = "event"
- memory_id = f"memory_{str(memory_count)}"
+ memory_id = f"node_{str(memory_count)}"
depth = 0
if "(" in content:
@@ -330,7 +334,7 @@ class AgentMemory(Memory):
ret_set.add(e_node.summary())
return ret_set
- def get_last_chat(self, target_role_name: str) -> str:
+ def get_last_chat(self, target_role_name: str):
if target_role_name.lower() in self.chat_keywords:
return self.chat_keywords[target_role_name.lower()][0]
else:
diff --git a/examples/st_game/memory/retrieve.py b/examples/st_game/memory/retrieve.py
index 9f19a41d4..9478b1f9f 100644
--- a/examples/st_game/memory/retrieve.py
+++ b/examples/st_game/memory/retrieve.py
@@ -3,17 +3,17 @@
# @Desc : Retrieve函数实现
import datetime
-from typing import Union
from numpy import dot
from numpy.linalg import norm
-from examples.st_game.memory.agent_memory import AgentMemory, BasicMemory
+from examples.st_game.memory.agent_memory import BasicMemory
from examples.st_game.utils.utils import get_embedding
+from metagpt.logs import logger
-def agent_retrieve(agent_memory: AgentMemory, curr_time: datetime.datetime, memory_forget: float, query: str,
- topk: int = 4) -> list[BasicMemory]:
+def agent_retrieve(agent_memory, curr_time: datetime.datetime, memory_forget: float, query: str, nodes: list[BasicMemory],
+ topk: int = 4, ) -> list[BasicMemory]:
"""
Retrieve需要集合Role使用,原因在于Role才具有AgentMemory,scratch
逻辑:Role调用该函数,self._rc.AgentMemory,self._rc.scratch.curr_time,self._rc.scratch.memory_forget
@@ -27,13 +27,14 @@ def agent_retrieve(agent_memory: AgentMemory, curr_time: datetime.datetime, memo
"relevance": 搜索结果
}
"""
- memories = agent_memory.storage
+ memories = nodes
+ agent_memory_embedding = agent_memory.embeddings
memories = sorted(memories, key=lambda memory_node: memory_node.last_accessed, reverse=True)
score_list = []
score_list = extract_importance(memories, score_list)
score_list = extract_recency(curr_time, memory_forget, score_list)
- score_list = extract_relevance(query, score_list)
+ score_list = extract_relevance(agent_memory_embedding,query, score_list)
score_list = normalize_score_floats(score_list, 0, 1)
total_dict = {}
@@ -43,31 +44,37 @@ def agent_retrieve(agent_memory: AgentMemory, curr_time: datetime.datetime, memo
score_list[i]['recency'] * gw[1] +
score_list[i]['relevance'] * gw[2]
)
- total_dict[score_list[i]['memory']] = total_score
+ total_dict[score_list[i]['memory'].memory_id] = total_score
result = top_highest_x_values(total_dict, topk)
return result # 返回的是一个BasicMemory列表
-def new_agent_retrieve(strole: "STRole", focus_points: list, n_count=30):
+def new_agent_retrieve(role, focus_points: list, n_count=30) -> dict:
"""
- 输入为Strole,关注点列表,返回记忆数量
+ 输入为role,关注点列表,返回记忆数量
输出为字典,键为focus_point,值为对应的记忆列表
"""
retrieved = dict()
for focal_pt in focus_points:
nodes = [[i.last_accessed, i]
- for i in strole._rc.memory.event_list + strole._rc.memory.thought_list
+ for i in role.memory.event_list + role.memory.thought_list
if "idle" not in i.embedding_key]
nodes = sorted(nodes, key=lambda x: x[0])
nodes = [i for created, i in nodes]
- results = agent_retrieve(strole._rc.memory, strole._rc.scratch.curr_time, strole._rc.scratch.recency_decay,
- focal_pt, n_count)
+ results = agent_retrieve(role.memory, role.scratch.curr_time, role.scratch.recency_decay,
+ focal_pt, nodes, n_count)
+ final_result = []
for n in results:
- n.last_accessed = strole._rc.scratch.curr_time
+ for i in role.memory.storage:
+ if i.memory_id == n:
+ i.last_accessed = role.scratch.curr_time
+ final_result.append(i)
- retrieved[focal_pt] = results
+ retrieved[focal_pt] = final_result
+
+ return retrieved
def top_highest_x_values(d, x):
@@ -91,14 +98,15 @@ def extract_importance(memories, score_list):
return score_list
-def extract_relevance(query, score_list):
+def extract_relevance(agent_memory_embedding,query, score_list):
"""
抽取相关性
"""
query_embedding = get_embedding(query)
# 进行
for i in range(len(score_list)):
- result = cos_sim(score_list[i]["memory"].embedding_key, query_embedding)
+ node_embedding = agent_memory_embedding[score_list[i]["memory"].embedding_key]
+ result = cos_sim(node_embedding, query_embedding)
score_list[i]['relevance'] = result
return score_list
diff --git a/examples/st_game/prompts/insight_and_evidence_v1.txt b/examples/st_game/prompts/insight_and_evidence_v1.txt
index 579c81637..35671023b 100644
--- a/examples/st_game/prompts/insight_and_evidence_v1.txt
+++ b/examples/st_game/prompts/insight_and_evidence_v1.txt
@@ -8,5 +8,5 @@ Variables:
Input:
!!
-What !! high-level insights can you infer from the above statements? (example format: insight (because of 1, 5, 3))
+What !! high-level insights can you infer from the above statements? Please ensure it includes 'because of' and generates according to the example format.(example format: insight (because of 1, 5, 3)) .
1.
\ No newline at end of file
diff --git a/examples/st_game/prompts/whisper_inner_thought_v1.txt b/examples/st_game/prompts/whisper_inner_thought_v1.txt
new file mode 100644
index 000000000..b1ed50aaa
--- /dev/null
+++ b/examples/st_game/prompts/whisper_inner_thought_v1.txt
@@ -0,0 +1,11 @@
+whisper_inner_thought_v1.txt
+
+Variables:
+!! -- init persona name
+!! -- whisper
+
+###
+Translate the following thought into a statement about !!.
+
+Thought: "!!"
+Statement: "
\ No newline at end of file
diff --git a/examples/st_game/reflect/ga_prompt_generator.py b/examples/st_game/reflect/ga_prompt_generator.py
deleted file mode 100644
index 81ed5e7f7..000000000
--- a/examples/st_game/reflect/ga_prompt_generator.py
+++ /dev/null
@@ -1,64 +0,0 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
-# author: didi
-# Date:9.25
-
-import openai
-from metagpt.llm import DEFAULT_LLM
-# 直接调用Prompt生成
-# ga的prompt构建格式和metagpt完全不同。没有办法融合。
-
-
-# 特殊指令加入Prompt生成
-
-
-async def final_response(prompt, special_instruction, example_output=None):
- """
- 通过将特殊指令加入Prompt生成最终的响应。
-
- 参数:
- - prompt:要生成响应的提示文本。
- - special_instruction:要加入Prompt的特殊指令。
- - example_output(可选):示例输出的JSON字符串。
-
- 返回:
- 生成的最终响应。
-
- """
- prompt = '"""\n' + prompt + '\n"""\n'
- prompt += f"Output the response to the prompt above in json. {special_instruction}\n"
- if example_output:
- prompt += "Example output json:\n"
- prompt += '{"output": "' + str(example_output) + '"}'
- return await DEFAULT_LLM.aask(prompt)
-
-# prompt填充模板
-
-
-def prompt_generate(curr_input, prompt_lib_file):
- """
- Takes in the current input (e.g. comment that you want to classifiy) and
- the path to a prompt file. The prompt file contains the raw str prompt that
- will be used, which contains the following substr: !! -- this
- function replaces this substr with the actual curr_input to produce the
- final promopt that will be sent to the GPT3 server.
- ARGS:
- curr_input: the input we want to feed in (IF THERE ARE MORE THAN ONE
- INPUT, THIS CAN BE A LIST.)
- prompt_lib_file: the path to the promopt file.
- RETURNS:
- a str prompt that will be sent to OpenAI's GPT server.
- """
- if isinstance(curr_input, str):
- curr_input = [curr_input]
- curr_input = [str(i) for i in curr_input]
-
- f = open(prompt_lib_file, "r")
- prompt = f.read()
- f.close()
- for count, i in enumerate(curr_input):
- prompt = prompt.replace(f"!!", i)
- if "###" in prompt:
- prompt = prompt.split(
- "###")[1]
- return prompt.strip()
diff --git a/examples/st_game/reflect/reflect.py b/examples/st_game/reflect/reflect.py
index 6c19cf3fc..62f835e1a 100644
--- a/examples/st_game/reflect/reflect.py
+++ b/examples/st_game/reflect/reflect.py
@@ -5,7 +5,6 @@
import datetime
from metagpt.logs import logger
-
from examples.st_game.utils.utils import get_embedding
from examples.st_game.actions.run_reflect_action import (
AgentFocusPt, AgentInsightAndGuidance, AgentEventTriple,
@@ -17,14 +16,14 @@ from examples.st_game.actions.run_reflect_action import (
def generate_focal_points(role: "STRole", n=3):
nodes = [
[i.last_accessed, i] for i in
- role._rc.memory.event_list + role._rc.memory.thought_list
+ role.memory.event_list + role.memory.thought_list
if "idle" not in i.embedding_key
]
nodes = sorted(nodes, key=lambda x: x[0])
nodes = [i for _, i in nodes]
statements = ""
- for node in nodes[-1 * role._rc.scratch.importance_ele_n:]:
+ for node in nodes[-1 * role.scratch.importance_ele_n:]:
statements += node.embedding_key + "\n"
run_focal_pt = AgentFocusPt()
return run_focal_pt.run(role, statements, n)
@@ -38,12 +37,14 @@ def generate_insights_and_evidence(role, nodes, n=5):
ret = run_insight_and_guidance.run(role, statements, n)
logger.info(ret)
+
try:
for thought, evi_raw in ret.items():
- evidence_node_id = [nodes[i].node_id for i in evi_raw]
+ evidence_node_id = [nodes[i].memory_id for i in evi_raw]
ret[thought] = evidence_node_id
return ret
- except:
+ except Exception as exp:
+ logger.error(f"generate_insights_and_evidence error:{exp}")
return {"this is blank": "node_1"}
@@ -59,7 +60,8 @@ def generate_action_event_triple(act_desp, role):
"🧈🍞"
"""
run_event_triple = AgentEventTriple()
- return AgentEventTriple(act_desp, role)
+ result = run_event_triple.run(act_desp, role)
+ return result
def generate_poig_score(role: "STRole", event_type, description):
@@ -68,11 +70,11 @@ def generate_poig_score(role: "STRole", event_type, description):
if event_type == "event" or event_type == "thought":
run_event_poignancy = AgentEventPoignancy()
- return run_event_poignancy.run(role, description)[0]
+ return run_event_poignancy.run(role, description)
elif event_type == "chat":
run_chat_poignancy = AgentChatPoignancy()
return run_chat_poignancy.run(role,
- role._rc.scratch.act_description)[0]
+ role.scratch.act_description)
def generate_planning_thought_on_convo(role, all_utt):
@@ -98,7 +100,7 @@ def run_reflect(role: "STRole"):
"""
# Reflection requires certain focal points. Generate that first.
focal_points = generate_focal_points(role, 3)
- # Retrieve the relevant Nodes object for each of the focal points.
+ # Retrieve the relevant Nodesobject for each of the focal points.
# has keys of focal points, and values of the associated Nodes.
retrieved = role.retrieve(focal_points)
@@ -106,25 +108,25 @@ def run_reflect(role: "STRole"):
# agent's memory.
for focal_pt, nodes in retrieved.items():
xx = [i.embedding_key for i in nodes]
- for xxx in xx: logger.info(xxx)
+ for xxx in xx: logger.info(f"Nodes retrieved for {focal_pt} are {xxx}.")
thoughts = generate_insights_and_evidence(role, nodes, 5)
# 生成的是字典类型
for thought, evidence in thoughts.items():
created = role.scratch.curr_time
expiration = created + datetime.timedelta(days=30)
- s, p, o = generate_action_event_triple(thought, role)
+ s, p, o = generate_action_event_triple("(" + thought + ")", role)
keywords = set([s, p, o])
thought_poignancy = generate_poig_score(role, "thought", thought)
thought_embedding_pair = (thought, get_embedding(thought))
- role._rc.memory.add_thought(
+ role.memory.add_thought(
created, expiration, s, p, o, thought, keywords,
thought_poignancy, thought_embedding_pair, evidence
)
+ logger.info(f"add thought memory: {thought}")
-# Done
def reflection_trigger(role: "STRole"):
"""
Given the current role, determine whether the role should run a
@@ -139,14 +141,10 @@ def reflection_trigger(role: "STRole"):
True if we are running a new reflection.
False otherwise.
"""
- logger.info(
- role._rc.scratch.name, "role.scratch.importance_trigger_curr::",
- role._rc.scratch.importance_trigger_curr
- )
- logger.info(role._rc.scratch.importance_trigger_max)
+ logger.info(f"{role.scratch.name} role.scratch.importance_trigger_curr:: {role.scratch.importance_trigger_curr}"),
- if (role._rc.scratch.importance_trigger_curr <= 0 and
- [] != role._rc.memory.seq_event + role._rc.memory.seq_thought):
+ if (role.scratch.importance_trigger_curr <= 0 and
+ [] != role.memory.event_list + role.memory.thought_list):
return True
return False
@@ -161,13 +159,12 @@ def reset_reflection_counter(role: "STRole"):
Output:
None
"""
- role_imt_max = role._rc.scratch.importance_trigger_max
- role._rc.scratch.importance_trigger_curr = role_imt_max
- role._rc.scratch.importance_ele_n = 0
+ role_imt_max = role.scratch.importance_trigger_max
+ role.scratch.importance_trigger_curr = role_imt_max
+ role.scratch.importance_ele_n = 0
-# Question 1 chat函数
-def reflect(role: "STRole"):
+def role_reflect(role: "STRole"):
"""
The main reflection module for the role. We first check if the trigger
conditions are met, and if so, run the reflection and reset any of the
@@ -182,42 +179,41 @@ def reflect(role: "STRole"):
run_reflect(role)
reset_reflection_counter(role)
- if role._rc.scratch.chatting_end_time:
- if role._rc.scratch.curr_time + datetime.timedelta(0, 10) == role._rc.scratch.chatting_end_time:
+ if role.scratch.chatting_end_time:
+ if role.scratch.curr_time + datetime.timedelta(0, 10) == role.scratch.chatting_end_time:
all_utt = ""
- if role._rc.scratch.chat:
- for row in role._rc.scratch.chat:
+ if role.scratch.chat:
+ for row in role.scratch.chat:
all_utt += f"{row[0]}: {row[1]}\n"
- # Question memory添加对话函数
- evidence = [role._rc.memory.get_last_chat(role._rc.scratch.chatting_with).memory_id]
+ evidence = [role.memory.get_last_chat(role.scratch.chatting_with).memory_id]
planning_thought = generate_planning_thought_on_convo(role, all_utt)
- planning_thought = f"For {role._rc.scratch.name}'s planning: {planning_thought}"
+ planning_thought = f"For {role.scratch.name}'s planning: {planning_thought}"
- created = role._rc.scratch.curr_time
+ created = role.scratch.curr_time
expiration = created + datetime.timedelta(days=30)
s, p, o = generate_action_event_triple(planning_thought, role)
keywords = set([s, p, o])
thought_poignancy = generate_poig_score(role, "thought", planning_thought)
thought_embedding_pair = (planning_thought, get_embedding(planning_thought))
- role._rc.memory.add_thought(
+ role.memory.add_thought(
created, expiration, s, p, o, planning_thought, keywords,
thought_poignancy, thought_embedding_pair, evidence
)
memo_thought = generate_memo_on_convo(role, all_utt)
- memo_thought = f"{role._rc.scratch.name} {memo_thought}"
+ memo_thought = f"{role.scratch.name} {memo_thought}"
- created = role._rc.scratch.curr_time
+ created = role.scratch.curr_time
expiration = created + datetime.timedelta(days=30)
s, p, o = generate_action_event_triple(memo_thought, role)
keywords = set([s, p, o])
thought_poignancy = generate_poig_score(role, "thought", memo_thought)
thought_embedding_pair = (memo_thought, get_embedding(memo_thought))
- role._rc.memory.add_thought(
+ role.memory.add_thought(
created, expiration, s, p, o, memo_thought, keywords,
thought_poignancy, thought_embedding_pair, evidence
)
diff --git a/examples/st_game/reflect/st_reflect.py b/examples/st_game/reflect/st_reflect.py
deleted file mode 100644
index f4b4c448c..000000000
--- a/examples/st_game/reflect/st_reflect.py
+++ /dev/null
@@ -1,92 +0,0 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
-# @Desc : st's reflection execution
-
-import asyncio
-import json
-import time
-from metagpt.logs import logger
-
-from examples.st_game.prompts.wrapper_prompt import special_response_generate
-from examples.st_game.memory.agent_memory import BasicMemory
-
-
-async def agent_reflect(memories_list):
- """
- 代理反思函数:生成关注点并生成洞察和证据
-
- """
- A = await generate_focus_point(memories_list)
-
- for i in A:
- B = await generate_insights_and_evidence(memories_list, question=i)
-
-
-async def generate_focus_point(memories_list: list[BasicMemory], n=3):
- """
- 生成关注点函数:根据记忆列表生成关注点
- """
- wait_sorted_mem = [[i.accessed_time, i] for i in memories_list]
- sorted_memories = sorted(wait_sorted_mem, key=lambda x: x[0])
- memorys = [i for created, i in sorted_memories]
- statements = ''
- for i in memorys:
- statements += i.description + "\n"
- prompt = '''
- {statements}
- Given only the information above, what are {num_question} most salient high-level questions we can answer about the subjects grounded in the statements?
- '''
- example_output = '["What should Jane do for lunch", "Does Jane like strawberry", "Who is Jane"]'
- out = await final_response(prompt.format(statements=statements, num_question=n),
- "Output must be a list of str.", example_output)
- try:
- poi_dict = json.loads(out)
- return poi_dict['output']
- except ValueError:
- print(out)
- logger.error('无法返回正常结果')
- return out
-
-
-async def generate_insights_and_evidence(memories_list: list[BasicMemory], question: str, n=5):
- """
- 生成洞察和证据函数:根据问题生成洞察和证据
- """
- memories_list = await agent_retrieve(agent, question, 50, 10)
- statements = ""
- for count, mem in enumerate(memories_list):
- statements += f'{str(count)}. {mem.description}\n'
- prompt = '''
- Input:
- {statements}
-
- What {n} high-level insights can you infer from the above statements?
- You should return a list of list[str,list]. The first element is the insight you have found. The second element is the
- '''
-
- ret = final_response(prompt.format(
- question=question, statements=statements, n=n), "['insightA',[1,2,3]]")
- try:
- insight_list = json.loads(ret)
- for insight, index in insight_list:
- agent.memory_list.append(BasicMemory(
- time.time(), None, insight, None, None))
- return insight_list
- except:
- logger.error('我们无法获得想要的返回。')
- return ret
-
-
-""" if __name__ == "__main__":
- # 例子,构建John Agent,实现retrive
- John_iss = "John Lin is a pharmacy shopkeeper at the Willow Market and Pharmacy who loves to help people. He is always looking for ways to make the process of getting medication easier for his customers; John Lin is living with his wife, Mei Lin, who is a college professor, and son, Eddy Lin, who is a student studying music theory; John Lin loves his family very much; John Lin has known the old couple next-door, Sam Moore and Jennifer Moore, for a few years; John Lin thinks Sam Moore is a kind and nice man; John Lin knows his neighbor, Yuriko Yamamoto, well; John Lin knows of his neighbors, Tamara Taylor and Carmen Ortiz, but has not met them before; John Lin and Tom Moreno are colleagues at The Willows Market and Pharmacy; John Lin and Tom Moreno are friends and like to discuss local politics together; John Lin knows the Moreno family somewhat well — the husband Tom Moreno and the wife Jane Moreno."
- John = AgentMemory(
- "John", John_iss, memory_path="agent_memories/John_memory.json")
-
- # John的相关信息:{'Had a friendly chat with Yuriko about her garden.': 2.4992317730827667, 'Helped Mrs. Moore carry groceries into her house.': 1.957656720441911, 'Discussed local politics with Tom Moreno.': 1.9458268038234035}
- asyncio.run(agent_reflect(John))
- '''
- 这里是输出,list形式,返回给记忆。
- [['The pharmacy is a friendly and helpful community.', [0, 2, 9, 12]], ['The pharmacy is a place where people come for more than just medication.', [3, 5, 13, 14]], ['The pharmacy is a place where people come for advice and conversation.', [0, 2, 6, 9, 12]], ['The pharmacy is a place where people come for assistance with daily tasks.', [3, 5, 13, 14]], ['The pharmacy is a place where people come for political discussions.', [1]]]
- '''
- """
diff --git a/examples/st_game/roles/st_role.py b/examples/st_game/roles/st_role.py
index 1521d0336..ca1bac6e0 100644
--- a/examples/st_game/roles/st_role.py
+++ b/examples/st_game/roles/st_role.py
@@ -34,6 +34,9 @@ from examples.st_game.utils.utils import get_embedding, path_finder
from examples.st_game.utils.const import collision_block_id, STORAGE_PATH
from examples.st_game.reflect.reflect import generate_poig_score
from examples.st_game.utils.mg_ga_transform import save_movement, get_role_environment
+from examples.st_game.actions.inner_voice_action import AgentWhisperThoughtAction
+from examples.st_game.actions.run_reflect_action import AgentEventTriple
+from examples.st_game.reflect.reflect import role_reflect
class STRoleContext(RoleContext):
@@ -101,6 +104,10 @@ class STRole(Role):
def scratch(self):
return self._rc.scratch
+ @property
+ def memory(self):
+ return self._rc.memory
+
def load_from(self, folder: Path):
"""
load role data from `storage/{simulation_name}/personas/{role_name}
@@ -120,15 +127,31 @@ class STRole(Role):
observed = self._rc.env.memory.get_by_actions(self._rc.watch)
self._rc.news = self._rc.memory.remember(observed)
if len(self._rc.news) == 1 and self._rc.news[0].cause_by == UserRequirement:
- # add inner voice
- # TODO
+ self.add_inner_voice(self._rc.news[0].content)
logger.warning(f"Role: {self.name} add inner voice: {self._rc.news[0].content}")
return 1 # always return 1 to execute role's `_react`
- def add_inner_voice(self):
+ def add_inner_voice(self, whisper):
# TODO
- pass
+ def generate_inner_thought(role: STRole, whisper):
+ run_whisper_thought = AgentWhisperThoughtAction()
+ inner_thought = run_whisper_thought.run(self, whisper)
+ return inner_thought
+
+ whisper = input("Enter Input: ")
+ thought = generate_inner_thought(whisper)
+
+ created = self._rc.scratch.curr_time
+ expiration = self._rc.scratch.curr_time + datetime.timedelta(days=30)
+ run_event_triple = AgentEventTriple()
+ s, p, o = run_event_triple.run(thought, self)
+ keywords = set([s, p, o])
+ thought_poignancy = generate_poig_score(self, "event", whisper)
+ thought_embedding_pair = (thought, get_embedding(thought))
+ self._rc.memory.add_thought(created, expiration, s, p, o,
+ thought, keywords, thought_poignancy,
+ thought_embedding_pair, None)
async def observe(self) -> list[BasicMemory]:
# TODO observe info from maze_env
@@ -271,7 +294,7 @@ class STRole(Role):
return ret_events
- async def retrieve(self, focus_points, n=30):
+ def retrieve(self, focus_points, n=30) -> dict:
# TODO retrieve memories from agent_memory
retrieve_memories = new_agent_retrieve(self, focus_points, n)
return retrieve_memories
@@ -286,11 +309,11 @@ class STRole(Role):
# TODO re-add result into memory
pass
- async def reflect(self):
+ def reflect(self):
# TODO reflection if meet reflect condition
-
+ result = role_reflect(self)
# TODO re-add result to memory
- pass
+ # 已封装到Reflect函数之中
def execute(self, plan: str):
"""
diff --git a/examples/st_game/tests/test_agent_memory.py b/examples/st_game/tests/test_agent_memory.py
new file mode 100644
index 000000000..9596f97d4
--- /dev/null
+++ b/examples/st_game/tests/test_agent_memory.py
@@ -0,0 +1,73 @@
+import pytest
+import os
+import time
+from datetime import datetime, timedelta
+from metagpt.logs import logger
+from examples.st_game.memory.agent_memory import AgentMemory
+from examples.st_game.utils.const import STORAGE_PATH
+from examples.st_game.memory.retrieve import agent_retrieve
+
+"""
+memory测试思路
+1. Basic Memory测试
+2. Agent Memory测试
+ 2.1 Load & Save方法测试; Load方法中使用了add方法,验证Load即可验证所有add
+ 2.2 Get方法测试
+"""
+memory_easy_storage_path = os.path.join(STORAGE_PATH,"July1_the_ville_isabella_maria_klaus-step-3-4/personas/Isabella Rodriguez/bootstrap_memory/associative_memory")
+memroy_chat_storage_path = os.path.join(STORAGE_PATH,"July1_the_ville_isabella_maria_klaus-step-3-11/personas/Isabella Rodriguez/bootstrap_memory/associative_memory")
+memory_save_easy_test_path = os.path.join(STORAGE_PATH,"July1_the_ville_isabella_maria_klaus-step-3-4/personas/Isabella Rodriguez/bootstrap_memory/test_memory")
+memory_save_chat_test_path = os.path.join(STORAGE_PATH,"July1_the_ville_isabella_maria_klaus-step-3-11/personas/Isabella Rodriguez/bootstrap_memory/test_memory")
+class TestAgentMemory:
+ @pytest.fixture
+ def agent_memory(self):
+ # 创建一个AgentMemory实例并返回,可以在所有测试用例中共享
+ test_agent_memory = AgentMemory()
+ test_agent_memory.set_mem_path(memroy_chat_storage_path)
+ return test_agent_memory
+
+ def test_load(self,agent_memory):
+ logger.info(f"存储路径为:{agent_memory.memory_saved}")
+ logger.info(f"存储记忆条数为:{len(agent_memory.storage)}")
+ logger.info(f"kw_strength为{agent_memory.kw_strength_event},{agent_memory.kw_strength_thought}")
+ logger.info(f"embeeding.json条数为{len(agent_memory.embeddings)}")
+
+ assert agent_memory.embeddings != None
+
+ def test_save(self,agent_memory):
+ try:
+ agent_memory.save(memory_save_chat_test_path)
+ logger.info("成功存储")
+ except:
+ pass
+
+ def test_summary_function(self, agent_memory):
+ logger.info(f"event长度为{len(agent_memory.event_list)}")
+ logger.info(f"thought长度为{len(agent_memory.thought_list)}")
+ logger.info(f"chat长度为{len(agent_memory.chat_list)}")
+ result1 = agent_memory.get_summarized_latest_events(4)
+ logger.info(f"总结最近事件结果为:{result1}")
+ def test_get_last_chat_function(self,agent_memory):
+ result2 = agent_memory.get_last_chat("customers")
+ logger.info(f"上一次对话是{result2}")
+
+ def test_retrieve_function(self,agent_memory):
+ focus_points = ["who i love?"]
+ retrieved = dict()
+ for focal_pt in focus_points:
+ nodes = [[i.last_accessed, i]
+ for i in agent_memory.event_list + agent_memory.thought_list
+ if "idle" not in i.embedding_key]
+ nodes = sorted(nodes, key=lambda x: x[0])
+ nodes = [i for created, i in nodes]
+ results = agent_retrieve(agent_memory, datetime.now()-timedelta(days=120), 0.99,
+ focal_pt, nodes, 5)
+ final_result = []
+ for n in results:
+ for i in agent_memory.storage:
+ if i.memory_id == n:
+ i.last_accessed = datetime.now()-timedelta(days=120)
+ final_result.append(i)
+
+ retrieved[focal_pt] = final_result
+ logger.info(f"检索结果为{retrieved}")
\ No newline at end of file
diff --git a/examples/st_game/tests/test_basic_memory.py b/examples/st_game/tests/test_basic_memory.py
new file mode 100644
index 000000000..79184ceae
--- /dev/null
+++ b/examples/st_game/tests/test_basic_memory.py
@@ -0,0 +1,66 @@
+from datetime import datetime, timedelta
+from metagpt.logs import logger
+from examples.st_game.memory.agent_memory import BasicMemory
+import pytest
+
+"""
+memory测试思路
+1. Basic Memory测试
+2. Agent Memory测试
+ 2.1 Load & Save方法测试
+ 2.2 Add方法测试
+ 2.3 Get方法测试
+"""
+
+# Create some sample BasicMemory instances
+memory1 = BasicMemory(
+ memory_id="1",
+ memory_count=1,
+ type_count=1,
+ memory_type="event",
+ depth=1,
+ created=datetime.now(),
+ expiration=datetime.now() + timedelta(days=30),
+ subject="Subject1",
+ predicate="Predicate1",
+ object="Object1",
+ content="This is content 1",
+ embedding_key="embedding_key_1",
+ poignancy=1,
+ keywords=["keyword1", "keyword2"],
+ filling=["memory_id_2"]
+)
+memory2 = BasicMemory(
+ memory_id="2",
+ memory_count=2,
+ type_count=2,
+ memory_type="thought",
+ depth=2,
+ created=datetime.now(),
+ expiration=datetime.now() + timedelta(days=30),
+ subject="Subject2",
+ predicate="Predicate2",
+ object="Object2",
+ content="This is content 2",
+ embedding_key="embedding_key_2",
+ poignancy=2,
+ keywords=["keyword3", "keyword4"],
+ filling=[]
+)
+
+@pytest.fixture
+def basic_mem_set():
+ basic_mem2 = memory2
+ yield basic_mem2
+
+def test_basic_mem_function(basic_mem_set):
+ a, b, c = basic_mem_set.summary()
+ logger.info(f"{a}{b}{c}")
+ assert a == "Subject2"
+
+def test_basic_mem_save(basic_mem_set):
+ result = basic_mem_set.save_to_dict()
+ logger.info(f"save结果为{result}")
+
+if __name__ == "__main__":
+ pytest.main()
diff --git a/examples/st_game/tests/test_reflect.py b/examples/st_game/tests/test_reflect.py
new file mode 100644
index 000000000..191e46279
--- /dev/null
+++ b/examples/st_game/tests/test_reflect.py
@@ -0,0 +1,65 @@
+import pytest
+from examples.st_game.roles.st_role import STRole
+from examples.st_game.actions.run_reflect_action import AgentFocusPt, AgentInsightAndGuidance, AgentEventTriple, \
+ AgentEventPoignancy, AgentChatPoignancy, AgentPlanThoughtOnConvo, AgentMemoryOnConvo
+from metagpt.logs import logger
+
+
+class TestReflectFunction:
+ @pytest.fixture
+ def init_agent(self):
+ """
+ init STRole form local json, set sim_code(path),curr_time & start_date
+ """
+ role = STRole(sim_code="July1_the_ville_isabella_maria_klaus-step-3-11", start_date='February 13, 2023',
+ curr_time='February 13, 2023, 14:53:10')
+ return role
+
+ def test_function_focus_and_insight_action(self, init_agent):
+ """
+ test for AgentFocusPt & AgentInsightAndGuidance
+ """
+ logger.info(f"{__name__}函数启动")
+ run_focus = AgentFocusPt()
+ statements = ""
+ run_focus.run(init_agent, statements, n=3)
+
+ """
+ 这里有通过测试的结果,但是更多时候LLM生成的结果缺少了because of;考虑修改一下prompt
+ result = {'Klaus Mueller and Maria Lopez have a close relationship because they have been friends for a long time and have a strong bond': [1, 2, 5, 9, 11, 14], 'Klaus Mueller has a crush on Maria Lopez': [8, 15, 24], 'Klaus Mueller is academically inclined and actively researching a topic': [13, 20], 'Klaus Mueller is socially active and acquainted with Isabella Rodriguez': [17, 21, 22], 'Klaus Mueller is organized and prepared': [19]}
+ """
+ run_insight = AgentInsightAndGuidance()
+ statements = "[user: Klaus Mueller has a close relationship with Maria Lopez, user:s Mueller and Maria Lopez have a close relationship, user: Klaus Mueller has a close relationship with Maria Lopez, user: Klaus Mueller has a close relationship with Maria Lopez, user: Klaus Mueller and Maria Lopez have a strong relationship, user: Klaus Mueller is a dormmate of Maria Lopez., user: Klaus Mueller and Maria Lopez have a strong bond, user: Klaus Mueller has a crush on Maria Lopez, user: Klaus Mueller and Maria Lopez have been friends for more than 2 years., user: Klaus Mueller has a close relationship with Maria Lopez, user: Klaus Mueller Maria Lopez is heading off to college., user: Klaus Mueller and Maria Lopez have a close relationship, user: Klaus Mueller is actively researching a topic, user: Klaus Mueller is close friends and classmates with Maria Lopez., user: Klaus Mueller is socially active, user: Klaus Mueller has a crush on Maria Lopez., user: Klaus Mueller and Maria Lopez have been friends for a long time, user: Klaus Mueller is academically inclined, user: For Klaus Mueller's planning: should remember to ask Maria Lopez about her research paper, as she found it interesting that he mentioned it., user: Klaus Mueller is acquainted with Isabella Rodriguez, user: Klaus Mueller is organized and prepared, user: Maria Lopez is conversing about conversing about Maria's research paper mentioned by Klaus, user: Klaus Mueller is conversing about conversing about Maria's research paper mentioned by Klaus, user: Klaus Mueller is a student, user: Klaus Mueller is a student, user: Klaus Mueller is conversing about two friends named Klaus Mueller and Maria Lopez discussing their morning plans and progress on a research paper before Maria heads off to college., user: Klaus Mueller is socially active, user: Klaus Mueller is socially active, user: Klaus Mueller is socially active and acquainted with Isabella Rodriguez, user: Klaus Mueller has a crush on Maria Lopez]"
+ run_insight.run(init_agent, statements, n=5)
+
+ def test_event_triple_action(self, init_agent):
+ """
+ test for AgentEventTriple
+ error:遇到复杂情况时(多个Object)无法正确归纳,
+ solution: 限制MaxTokens为15
+ """
+ run_triple = AgentEventTriple()
+ statements = "(Klaus Mueller is academically inclined)"
+ run_triple.run(statements, init_agent)
+
+ def test_poignancy_action(self, init_agent):
+ """
+ test for AgentEventPoignancy, AgentChatPoignancy
+ done in reflect test
+ """
+ pass
+
+ def test_convo_action(self, init_agent):
+ """
+ test for AgentPlanThoughtOnConvo, AgentMemoryOnConvo
+ TODO Undone
+ """
+ pass
+
+ def test_reflect_function(self, init_agent):
+ """
+ test for reflection
+ modify importance_trigger_curr to trigger reflect
+ """
+ init_agent.scratch.importance_trigger_curr = -1
+ init_agent.reflect()