fix:修复一些BUG
This commit is contained in:
@@ -4,11 +4,13 @@ from langchain_core.output_parsers import StrOutputParser
|
||||
from langchain_core.prompts import ChatPromptTemplate
|
||||
from langchain_openai import ChatOpenAI
|
||||
from loguru import logger
|
||||
|
||||
import traceback
|
||||
from config.config import load_config
|
||||
|
||||
config = load_config("config.toml")
|
||||
|
||||
class_type_config =[
|
||||
"本期开展了小袋鼠整合主题课程:(语言、社会、科学、健康、艺术)、生活数学;特色课程(英语、体能、美工、篮球)",
|
||||
"本学期开展了柏克莱主题课程(语言、社会、科学、艺术、健康);英语及特色课程(体能、舞蹈、美工、魔力猴、足球、国学)"
|
||||
]
|
||||
|
||||
def generate_comment(name, age_group, traits,sex):
|
||||
"""
|
||||
@@ -19,7 +21,14 @@ def generate_comment(name, age_group, traits,sex):
|
||||
:param sex: 性别
|
||||
:return: 评语
|
||||
"""
|
||||
|
||||
# 1. 加载配置文件
|
||||
try:
|
||||
config = load_config("config.toml")
|
||||
except Exception as e:
|
||||
logger.error(f"配置文件获取失败: {str(e)}")
|
||||
# 打印详细报错位置,方便调试
|
||||
logger.error(traceback.format_exc())
|
||||
return "配置文件加载失败,请检查文件路径和内容。"
|
||||
ai_config = config["ai"]
|
||||
llm = ChatOpenAI(
|
||||
base_url=ai_config["api_url"],
|
||||
@@ -42,7 +51,8 @@ def generate_comment(name, age_group, traits,sex):
|
||||
"name": name,
|
||||
"age_group": age_group,
|
||||
"traits": traits,
|
||||
"sex": sex
|
||||
"sex": sex,
|
||||
"class_type": class_type_config[(config.get("class_type", 0))],
|
||||
})
|
||||
cleaned_text = re.sub(r'\s+', '', comment)
|
||||
logger.success(f"学生:{name} =>生成评语成功: {cleaned_text}")
|
||||
|
||||
Reference in New Issue
Block a user