main
黄海 5 months ago
parent 8d5224a0bd
commit 7354d468cf

@ -1,2 +1,9 @@
# markDown文件的默认模板路径
mdWorkingPath = r'D:\dsWork\QingLong\AI\md-file\readme'
from pathlib import Path
# 正确路径拼接方式
mdWorkingPath = Path(__file__).parent / 'md-file' / 'readme'
DEFAULT_TEMPLATE = mdWorkingPath / 'default.md' # 使用 / 运算符
DEFAULT_OUTPUT_DIR = mdWorkingPath / 'output' # 使用 / 运算符
print(f"模板路径:{DEFAULT_TEMPLATE}")
print(f"输出目录:{DEFAULT_OUTPUT_DIR}")

@ -12,9 +12,8 @@ class MarkdownGenerator:
"""Markdown教学大纲生成器"""
# 固定配置项
DEFAULT_TEMPLATE = Path(mdWorkingPath + r'\default.md')
DEFAULT_OUTPUT_DIR = Path(mdWorkingPath)
MODEL_R1 = "deepseek-r1"
# MODEL_R1 = "deepseek-r1"
MODEL_V3 = "deepseek-v3"
API_KEY = "sk-01d13a39e09844038322108ecdbd1bbc"
def __init__(
@ -31,8 +30,8 @@ class MarkdownGenerator:
:param template_path: 模板文件路径可选
"""
self.course_name = course_name
self.template_path = template_path or self.DEFAULT_TEMPLATE
self.output_path = output_path or self.DEFAULT_OUTPUT_DIR / f"{course_name}.md"
self.template_path = template_path or DEFAULT_TEMPLATE
self.output_path = output_path or DEFAULT_OUTPUT_DIR / f"{course_name}.md"
self._validate_paths()
@ -61,7 +60,7 @@ class MarkdownGenerator:
)
return Generation.call(
model=self.MODEL_R1,
model=self.MODEL_V3,
api_key=self.API_KEY,
messages=[
{"role": "system", "content": system_prompt},
@ -135,5 +134,5 @@ if __name__ == "__main__":
dashscope.api_key = MarkdownGenerator.API_KEY
generate_document(
course_name="小学数学三角形面积",
output_path=mdWorkingPath + r"\5.md"
output_path=mdWorkingPath / '5.md"'
)

Loading…
Cancel
Save