Losita/mai-bot
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

remove：移除tool_use模型，修复Jargon提取问题，修改统计为tool统计

Browse Source
This commit is contained in:
SengokuCola
2026-03-29 16:26:34 +08:00
parent 868438e3c1
commit 82bbf0fd52
25 changed files with 906 additions and 311 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/learners/expression_auto_check_task.py
View File

@@ -76,7 +76,7 @@ def create_evaluation_prompt(situation: str, style: str) -> str:
return prompt
judge_llm = LLMServiceClient(task_name="tool_use", request_type="expression_check")
judge_llm = LLMServiceClient(task_name="utils", request_type="expression_check")
async def single_expression_check(situation: str, style: str) -> tuple[bool, str, str | None]:

4
src/learners/expression_learner.py
View File

@@ -30,8 +30,8 @@ logger = get_logger("expressor")
express_learn_model = LLMServiceClient(
task_name="utils", request_type="expression.learner"
)
summary_model = LLMServiceClient(task_name="tool_use", request_type="expression.summary")
check_model = LLMServiceClient(task_name="tool_use", request_type="expression.check")
summary_model = LLMServiceClient(task_name="utils", request_type="expression.summary")
check_model = LLMServiceClient(task_name="utils", request_type="expression.check")
class ExpressionLearner:

2
src/learners/expression_selector.py
View File

@@ -19,7 +19,7 @@ logger = get_logger("expression_selector")
class ExpressionSelector:
def __init__(self):
self.llm_model = LLMServiceClient(
task_name="tool_use", request_type="expression.selector"
task_name="utils", request_type="expression.selector"
)
def can_use_expression_for_chat(self, chat_id: str) -> bool:

2
src/learners/expression_utils.py
View File

@@ -12,7 +12,7 @@ from src.common.logger import get_logger
logger = get_logger("expression_utils")
judge_llm = LLMServiceClient(task_name="tool_use", request_type="expression_check")
judge_llm = LLMServiceClient(task_name="utils", request_type="expression_check")
def _normalize_repair_json_result(repaired_result: Any) -> str:

2
src/learners/jargon_explainer_old.py
View File

@@ -25,7 +25,7 @@ class JargonExplainer:
def __init__(self, chat_id: str) -> None:
self.chat_id = chat_id
self.llm = LLMServiceClient(
task_name="tool_use",
task_name="utils",
request_type="jargon.explain",
)