fix:ruff
This commit is contained in:
@@ -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"
|
||||||
|
|||||||
@@ -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}")
|
||||||
|
|
||||||
# 改编场景,使其更适合当前角色
|
# 改编场景,使其更适合当前角色
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
@@ -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")
|
||||||
|
|||||||
@@ -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")
|
||||||
|
|||||||
@@ -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"
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user