大修_execute_request炸程序的问题
This commit is contained in:
@@ -18,7 +18,7 @@ heartflow_config = LogConfig(
|
||||
logger = get_module_logger("heartflow", config=heartflow_config)
|
||||
|
||||
|
||||
class CuttentState:
|
||||
class CurrentState:
|
||||
def __init__(self):
|
||||
self.willing = 0
|
||||
self.current_state_info = ""
|
||||
@@ -34,7 +34,7 @@ class Heartflow:
|
||||
def __init__(self):
|
||||
self.current_mind = "你什么也没想"
|
||||
self.past_mind = []
|
||||
self.current_state: CuttentState = CuttentState()
|
||||
self.current_state: CurrentState = CurrentState()
|
||||
self.llm_model = LLM_request(
|
||||
model=global_config.llm_heartflow, temperature=0.6, max_tokens=1000, request_type="heart_flow"
|
||||
)
|
||||
@@ -102,7 +102,11 @@ class Heartflow:
|
||||
current_thinking_info = self.current_mind
|
||||
mood_info = self.current_state.mood
|
||||
related_memory_info = "memory"
|
||||
sub_flows_info = await self.get_all_subheartflows_minds()
|
||||
try:
|
||||
sub_flows_info = await self.get_all_subheartflows_minds()
|
||||
except Exception as e:
|
||||
logger.error(f"获取子心流的想法失败: {e}")
|
||||
return
|
||||
|
||||
schedule_info = bot_schedule.get_current_num_task(num=4, time_info=True)
|
||||
|
||||
@@ -111,26 +115,29 @@ class Heartflow:
|
||||
prompt += f"{personality_info}\n"
|
||||
prompt += f"你想起来{related_memory_info}。"
|
||||
prompt += f"刚刚你的主要想法是{current_thinking_info}。"
|
||||
prompt += f"你还有一些小想法,因为你在参加不同的群聊天,是你正在做的事情:{sub_flows_info}\n"
|
||||
prompt += f"你还有一些小想法,因为你在参加不同的群聊天,这是你正在做的事情:{sub_flows_info}\n"
|
||||
prompt += f"你现在{mood_info}。"
|
||||
prompt += "现在你接下去继续思考,产生新的想法,但是要基于原有的主要想法,不要分点输出,"
|
||||
prompt += "输出连贯的内心独白,不要太长,但是记得结合上述的消息,关注新内容:"
|
||||
|
||||
reponse, reasoning_content = await self.llm_model.generate_response_async(prompt)
|
||||
try:
|
||||
response, reasoning_content = await self.llm_model.generate_response_async(prompt)
|
||||
except Exception as e:
|
||||
logger.error(f"内心独白获取失败: {e}")
|
||||
return
|
||||
self.update_current_mind(response)
|
||||
|
||||
self.update_current_mind(reponse)
|
||||
|
||||
self.current_mind = reponse
|
||||
self.current_mind = response
|
||||
logger.info(f"麦麦的总体脑内状态:{self.current_mind}")
|
||||
# logger.info("麦麦想了想,当前活动:")
|
||||
# await bot_schedule.move_doing(self.current_mind)
|
||||
|
||||
for _, subheartflow in self._subheartflows.items():
|
||||
subheartflow.main_heartflow_info = reponse
|
||||
subheartflow.main_heartflow_info = response
|
||||
|
||||
def update_current_mind(self, reponse):
|
||||
def update_current_mind(self, response):
|
||||
self.past_mind.append(self.current_mind)
|
||||
self.current_mind = reponse
|
||||
self.current_mind = response
|
||||
|
||||
async def get_all_subheartflows_minds(self):
|
||||
sub_minds = ""
|
||||
@@ -167,9 +174,9 @@ class Heartflow:
|
||||
prompt += """现在请你总结这些聊天内容,注意关注聊天内容对原有的想法的影响,输出连贯的内心独白
|
||||
不要太长,但是记得结合上述的消息,要记得你的人设,关注新内容:"""
|
||||
|
||||
reponse, reasoning_content = await self.llm_model.generate_response_async(prompt)
|
||||
response, reasoning_content = await self.llm_model.generate_response_async(prompt)
|
||||
|
||||
return reponse
|
||||
return response
|
||||
|
||||
def create_subheartflow(self, subheartflow_id):
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user