feat: add profile hint while extracting

Author: gusye1234

src/server/api/memobase_server/controllers/modal/chat/extract.py

@@ -1,5 +1,6 @@
 import asyncio
 from ....env import CONFIG, ContanstTable, TRACE_LOG
+from ....utils import truncate_string
 from ....models.utils import Promise
 from ....models.blob import Blob, BlobType
 from ....models.response import AIUserProfiles, CODE
@@ -64,14 +65,24 @@ async def extract_topics(
                 for p in profiles
             ]
         )
+        already_topic_subtopics_values = {
+            (
+                attribute_unify(p.attributes[ContanstTable.topic]),
+                attribute_unify(p.attributes[ContanstTable.sub_topic]),
+            ): p.content
+            for p in profiles
+        }
         if STRICT_MODE:
             already_topics_subtopics = already_topics_subtopics.intersection(
                 allowed_topic_subtopics
             )
+            already_topic_subtopics_values = {
+                k: already_topic_subtopics_values[k] for k in already_topics_subtopics
+            }
         already_topics_subtopics = sorted(already_topics_subtopics)
         already_topics_prompt = "\n".join(
             [
-                f"- {topic}{CONFIG.llm_tab_separator}{sub_topic}"
+                f"- {topic}{CONFIG.llm_tab_separator}{sub_topic}{CONFIG.llm_tab_separator}{truncate_string(already_topic_subtopics_values[(topic, sub_topic)], 5)}"
                 for topic, sub_topic in already_topics_subtopics
             ]
         )
@@ -99,7 +110,13 @@ async def extract_topics(
     if not p.ok():
         return p
     results = p.data()
-    # print(user_memo)
+    # print(
+    #     PROMPTS[USE_LANGUAGE]["extract"].pack_input(
+    #         already_topics_prompt,
+    #         user_memo,
+    #         strict_mode=STRICT_MODE,
+    #     )
+    # )
     # print("-------------------------------")
     # print(results)
     parsed_facts: AIUserProfiles = parse_string_into_profiles(results)

src/server/api/memobase_server/env.py

@@ -48,6 +48,7 @@ class ProjectStatus:
 class ContanstTable:
     topic = "topic"
     sub_topic = "sub_topic"
+    memo = "memo"
     update_hits = "update_hits"
 
     roleplay_plot_status = "roleplay_plot_status"

src/server/api/memobase_server/prompts/zh_extract_profile.py

@@ -186,8 +186,8 @@ def pack_input(already_input, chat_strs, strict_mode: bool = False):
 如果提取相关的主题/子主题，请考虑使用下面的主题/子主题命名:
 {already_input}
 
-#### 对话
-请注意，不要输出任何关于对话中未提及的主题/子主题的信息:
+#### 备忘录
+请注意，不要输出任何关于备忘录中未提及的主题/子主题的信息:
 {chat_strs}
 """