修了点pyright错误喵~

This commit is contained in:
ikun-11451
2025-11-29 20:40:21 +08:00
parent 311a161941
commit f0b6a969d5
17 changed files with 194 additions and 37 deletions

View File

@@ -78,6 +78,8 @@ def weighted_sample(population: list[dict], weights: list[float], k: int) -> lis
class ExpressionSelector:
def __init__(self, chat_id: str = ""):
self.chat_id = chat_id
if model_config is None:
raise RuntimeError("Model config is not initialized")
self.llm_model = LLMRequest(
model_set=model_config.model_task_config.utils_small, request_type="expression.selector"
)
@@ -94,6 +96,8 @@ class ExpressionSelector:
bool: 是否允许使用表达
"""
try:
if global_config is None:
return False
use_expression, _, _ = global_config.expression.get_expression_config_for_chat(chat_id)
return use_expression
except Exception as e:
@@ -122,6 +126,8 @@ class ExpressionSelector:
def get_related_chat_ids(self, chat_id: str) -> list[str]:
"""根据expression.rules配置获取与当前chat_id相关的所有chat_id包括自身"""
if global_config is None:
return [chat_id]
rules = global_config.expression.rules
current_group = None
@@ -280,6 +286,9 @@ class ExpressionSelector:
else:
chat_info = chat_history
if global_config is None:
raise RuntimeError("Global config is not initialized")
# 根据配置选择模式
mode = global_config.expression.mode
logger.debug(f"使用表达选择模式: {mode}")
@@ -582,6 +591,9 @@ class ExpressionSelector:
target_message_str = ""
target_message_extra_block = ""
if global_config is None:
raise RuntimeError("Global config is not initialized")
# 3. 构建prompt只包含情境不包含完整的表达方式
prompt = (await global_prompt_manager.get_prompt_async("expression_evaluation_prompt")).format(
bot_name=global_config.bot.nickname,