'commit'

dsLightRag/TeacherHelper/T4_ZuoYe.py

@@ -0,0 +1,96 @@
+import asyncio
+import logging
+import os
+import sys
+
+# 导入现有工具类（不创建新类）
+from TeacherHelper.Kit.TeacherHelper import (
+    LLMClient,
+    save_lesson_plan
+)
+
+# 配置日志
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+
+async def generate_gravitation_homework():
+    """生成万有引力课后作业的异步函数
+
+    Returns:
+        异步生成器: 流式返回作业内容
+    """
+    # 系统指令 - 符合布鲁姆认知分类和双减政策
+    system_prompt = "你是命题专家，熟悉布鲁姆认知分类和'双减'政策。客观题采用'四选一'单选，难度比例易:中:难=6:3:1；主观题设置2小问，第1问'解释现象'对应'理解'，第2问'方案设计'对应'创新'。题量控制为20分钟完成。"
+
+    # 创建LLM客户端实例（使用现有类）
+    llm_client = LLMClient(system_prompt=system_prompt)  # 修改为与T3_KeJian.py一致
+
+    # 作业生成提示词
+    prompt = """请输出'万有引力'课后作业，满分100分，含：
+A. 客观题（8题×5分=40分）
+   - 前3题考'史实&概念'识记
+   - 中间3题考'公式变形&比例'理解
+   - 后2题考'情境估算'应用
+B. 主观题（2题，30+30分）
+   - 题1：结合'天问一号'发射新闻，解释地球与火星之间的引力如何变化
+   - 题2：设计一个实验，用智能手机+免费APP估算地球质量，写出步骤与所需测量量
+C. 评分标准（主观题分点给分，每点10分）
+D. 参考答案与解析（客观题给出选项+一句话解析；主观题给出关键公式与评分关键词）
+
+【格式要求】
+### A. 客观题
+<ul>
+<li>1. ……</li>
+……
+</ul>
+
+### B. 主观题
+#### 1. ……
+#### 2. ……
+
+### C. 评分标准
+……
+
+### D. 参考答案与解析
+……"""
+
+    # 流式响应生成器
+    async def get_homework(): 
+        async for chunk in llm_client.get_response(prompt, stream=True):
+            yield chunk
+
+    return get_homework()
+
+
+async def test_generate_homework():
+    """测试生成万有引力课后作业并保存"""
+    print("===== 开始生成万有引力课后作业 =====")
+
+    try:
+        # 修复：添加await关键字获取生成器
+        homework_generator = await generate_gravitation_homework()
+        output_path = "./markdown/万有引力作业.md"
+
+        # 修复：正确调用save_lesson_plan函数，使用位置参数而非关键字参数
+        full_content, success = await save_lesson_plan(
+            homework_generator,
+            output_path,  # 修改为位置参数
+            stream_print=True
+        )
+
+        if success:
+            print(f"\n作业生成成功！文件路径: {os.path.abspath(output_path)}")
+            return full_content
+        else:
+            print("\n作业生成失败")
+            return None
+
+    except Exception as e:
+        print(f"生成过程中发生错误: {str(e)}", file=sys.stderr)
+        return None
+
+
+if __name__ == "__main__":
+    # 运行异步测试函数
+    asyncio.run(test_generate_homework())