Update schedule_generator.py

This commit is contained in:
SengokuCola
2025-03-26 23:26:32 +08:00
parent 67291f1b49
commit 805bde0646

View File

@@ -117,14 +117,14 @@ class ScheduleGenerator:
prompt = f"你是{self.name}{self.personality}{self.behavior}" prompt = f"你是{self.name}{self.personality}{self.behavior}"
prompt += f"你昨天的日程是:{self.yesterday_schedule_text}\n" prompt += f"你昨天的日程是:{self.yesterday_schedule_text}\n"
prompt += f"请为你生成{date_str}{weekday})的日程安排,结合你的个人特点和行为习惯\n" prompt += f"请为你生成{date_str}{weekday})的日程安排,结合你的个人特点和行为习惯\n"
prompt += "推测你的日程安排包括你一天都在做什么有什么发现和思考具体一些详细一些需要1500字以上精确到每半个小时记得写明时间\n" prompt += "推测你的日程安排,包括你一天都在做什么,从起床到睡眠,有什么发现和思考具体一些详细一些需要1500字以上精确到每半个小时记得写明时间\n"
prompt += "直接返回你的日程,不要输出其他内容:" prompt += "直接返回你的日程,从起床到睡觉,不要输出其他内容:"
return prompt return prompt
def construct_doing_prompt(self,time: datetime.datetime,mind_thinking: str = ""): def construct_doing_prompt(self,time: datetime.datetime,mind_thinking: str = ""):
now_time = time.strftime("%H:%M") now_time = time.strftime("%H:%M")
if self.today_done_list: if self.today_done_list:
previous_doings = self.get_current_num_task(10, True) previous_doings = self.get_current_num_task(5, True)
# print(previous_doings) # print(previous_doings)
else: else:
previous_doings = "你没做什么事情" previous_doings = "你没做什么事情"
@@ -132,9 +132,9 @@ class ScheduleGenerator:
prompt = f"你是{self.name}{self.personality}{self.behavior}" prompt = f"你是{self.name}{self.personality}{self.behavior}"
prompt += f"你今天的日程是:{self.today_schedule_text}\n" prompt += f"你今天的日程是:{self.today_schedule_text}\n"
prompt += f"你之前做了的事情是:{previous_doings},从之前到现在已经过去了{self.schedule_doing_update_interval/60}分钟了\n"
if mind_thinking: if mind_thinking:
prompt += f"你脑子里在想:{mind_thinking}\n" prompt += f"你脑子里在想:{mind_thinking}\n"
prompt += f"你之前做了的事情是:{previous_doings},从之前到现在已经过去了{self.schedule_doing_update_interval/60}分钟了\n"
prompt += f"现在是{now_time},结合你的个人特点和行为习惯," prompt += f"现在是{now_time},结合你的个人特点和行为习惯,"
prompt += "推测你现在做什么,具体一些,详细一些\n" prompt += "推测你现在做什么,具体一些,详细一些\n"
prompt += "直接返回你在做的事情,不要输出其他内容:" prompt += "直接返回你在做的事情,不要输出其他内容:"
@@ -196,7 +196,7 @@ class ScheduleGenerator:
else: else:
doing_prompt = self.construct_doing_prompt(current_time) doing_prompt = self.construct_doing_prompt(current_time)
print(doing_prompt) # print(doing_prompt)
doing_response,_ = await self.llm_scheduler_doing.generate_response_async(doing_prompt) doing_response,_ = await self.llm_scheduler_doing.generate_response_async(doing_prompt)
self.today_done_list.append((current_time,doing_response)) self.today_done_list.append((current_time,doing_response))