This commit is contained in:
SengokuCola
2025-04-06 15:57:42 +08:00
parent 6393bee020
commit 61d35063d2
7 changed files with 10 additions and 17 deletions

View File

@@ -193,7 +193,7 @@ class SubHeartflow:
related_info,grouped_results = await self.get_prompt_info(chat_observe_info + message_txt, 0.4) related_info,grouped_results = await self.get_prompt_info(chat_observe_info + message_txt, 0.4)
print(related_info) print(related_info)
for topic, results in grouped_results.items(): for _topic, results in grouped_results.items():
for result in results: for result in results:
print(result) print(result)
self.running_knowledges.append(result) self.running_knowledges.append(result)
@@ -457,8 +457,8 @@ class SubHeartflow:
# 按主题组织输出 # 按主题组织输出
for topic, results in grouped_results.items(): for topic, results in grouped_results.items():
related_info += f"【主题: {topic}\n" related_info += f"【主题: {topic}\n"
for i, result in enumerate(results, 1): for _i, result in enumerate(results, 1):
similarity = result["similarity"] _similarity = result["similarity"]
content = result["content"].strip() content = result["content"].strip()
# 调试:为内容添加序号和相似度信息 # 调试:为内容添加序号和相似度信息
# related_info += f"{i}. [{similarity:.2f}] {content}\n" # related_info += f"{i}. [{similarity:.2f}] {content}\n"

View File

@@ -1,7 +1,6 @@
from typing import Dict, List from typing import Dict, List
import json import json
import os import os
from pathlib import Path
from dotenv import load_dotenv from dotenv import load_dotenv
import sys import sys
import toml import toml
@@ -18,9 +17,9 @@ with open(config_path, "r", encoding="utf-8") as f:
config = toml.load(f) config = toml.load(f)
# 现在可以导入src模块 # 现在可以导入src模块
from src.individuality.scene import get_scene_by_factor, PERSONALITY_SCENES from src.individuality.scene import get_scene_by_factor, PERSONALITY_SCENES #noqa E402
from src.individuality.questionnaire import FACTOR_DESCRIPTIONS from src.individuality.questionnaire import FACTOR_DESCRIPTIONS #noqa E402
from src.individuality.offline_llm import LLM_request_off from src.individuality.offline_llm import LLM_request_off #noqa E402
# 加载环境变量 # 加载环境变量
env_path = os.path.join(root_path, ".env") env_path = os.path.join(root_path, ".env")
@@ -190,7 +189,7 @@ class PersonalityEvaluator_direct:
total_scenarios = len(self.scenarios) total_scenarios = len(self.scenarios)
progress_bar = tqdm(total=total_scenarios, desc="场景进度", ncols=100, bar_format='{l_bar}{bar}| {n_fmt}/{total_fmt} [{elapsed}<{remaining}]') progress_bar = tqdm(total=total_scenarios, desc="场景进度", ncols=100, bar_format='{l_bar}{bar}| {n_fmt}/{total_fmt} [{elapsed}<{remaining}]')
for i, scenario_data in enumerate(self.scenarios, 1): for _i, scenario_data in enumerate(self.scenarios, 1):
# print(f"\n{'-' * 20} 场景 {i}/{total_scenarios} - {scenario_data['场景编号']} {'-' * 20}") # print(f"\n{'-' * 20} 场景 {i}/{total_scenarios} - {scenario_data['场景编号']} {'-' * 20}")
# 改编场景,使其更适合当前角色 # 改编场景,使其更适合当前角色

View File

@@ -1,6 +1,5 @@
from dataclasses import dataclass from dataclasses import dataclass
from typing import Dict, List from typing import Dict, List
import os
import json import json
from pathlib import Path from pathlib import Path
import random import random

View File

@@ -18,7 +18,6 @@ from .chat_observer import ChatObserver
from .pfc_KnowledgeFetcher import KnowledgeFetcher from .pfc_KnowledgeFetcher import KnowledgeFetcher
from .reply_checker import ReplyChecker from .reply_checker import ReplyChecker
from .pfc_utils import get_items_from_json from .pfc_utils import get_items_from_json
import json
import time import time
logger = get_module_logger("pfc") logger = get_module_logger("pfc")

View File

@@ -1,6 +1,6 @@
import json import json
import re import re
from typing import Dict, Any, Optional, List, Tuple, Union from typing import Dict, Any, Optional, Tuple
from src.common.logger import get_module_logger from src.common.logger import get_module_logger
logger = get_module_logger("pfc_utils") logger = get_module_logger("pfc_utils")

View File

@@ -1,7 +1,6 @@
import random import random
import time import time
from typing import Optional, Union from typing import Optional, Union
import numpy as np
from ....common.database import db from ....common.database import db
from ...chat.utils import get_embedding, get_recent_group_detailed_plain_text, get_recent_group_speaker from ...chat.utils import get_embedding, get_recent_group_detailed_plain_text, get_recent_group_speaker
@@ -297,8 +296,8 @@ class PromptBuilder:
# 按主题组织输出 # 按主题组织输出
for topic, results in grouped_results.items(): for topic, results in grouped_results.items():
related_info += f"【主题: {topic}\n" related_info += f"【主题: {topic}\n"
for i, result in enumerate(results, 1): for _i, result in enumerate(results, 1):
similarity = result["similarity"] _similarity = result["similarity"]
content = result["content"].strip() content = result["content"].strip()
# 调试:为内容添加序号和相似度信息 # 调试:为内容添加序号和相似度信息
# related_info += f"{i}. [{similarity:.2f}] {content}\n" # related_info += f"{i}. [{similarity:.2f}] {content}\n"

View File

@@ -1,10 +1,7 @@
import random import random
import time
from typing import Optional from typing import Optional
from ...memory_system.Hippocampus import HippocampusManager
from ...moods.moods import MoodManager from ...moods.moods import MoodManager
from ...schedule.schedule_generator import bot_schedule
from ...config.config import global_config from ...config.config import global_config
from ...chat.utils import get_recent_group_detailed_plain_text, get_recent_group_speaker from ...chat.utils import get_recent_group_detailed_plain_text, get_recent_group_speaker
from ...chat.chat_stream import chat_manager from ...chat.chat_stream import chat_manager