Update memory_bank/summarize_memory.py

memory_bank/summarize_memory.py

@@ -327,4 +327,29 @@ def summarize_memory(memory_dir, name=None, language='en'):
             person_prompt = summarize_person_prompt(content, user_name, boot_name, language)
 
             if his_flag:
-                his_summary = ll
+                his_summary = llm_client.generate_text_simple(prompt=hisprompt, prompt_num=gen_prompt_num, language=language)
+                memory[user_name]['summary'][date] = {'content': his_summary}
+            
+            if problem_flag:
+                issues_summary = llm_client.generate_text_simple(prompt=issues_prompt, prompt_num=gen_prompt_num, language=language)
+                memory[user_name]['issues'][date] = {'content': issues_summary}
+                print(issues_summary)
+            
+            if person_flag:
+                person_summary = llm_client.generate_text_simple(prompt=person_prompt, prompt_num=gen_prompt_num, language=language)
+                memory[user_name]['personality'][date] = person_summary
+        
+        overall_his_prompt = summarize_overall_prompt(list(memory[user_name]['summary'].items()), language=language)
+        overall_person_prompt = summarize_overall_personality(list(memory[user_name]['personality'].items()), language=language)
+        
+        memory[user_name]['overall_history'] = llm_client.generate_text_simple(prompt=overall_his_prompt, prompt_num=gen_prompt_num, language=language)
+        memory[user_name]['overall_personality'] = llm_client.generate_text_simple(prompt=overall_person_prompt, prompt_num=gen_prompt_num, language=language)
+ 
+    with open(memory_dir, 'w', encoding='utf8') as f:
+        print(f'Successfully updated memory for {name}')
+        json.dump(memory, f, ensure_ascii=False)
+    return memory
+
+
+if __name__ == '__main__':
+    summarize_memory('../memories/eng_memory_cases.json', language='en')