ai_la/base6/test_new_question_generation.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
测试生成新题目，验证difficulty字段是否正确设置
"""

import requests
import json

def test_generate_questions():
    base_url = "http://localhost:8080"
    
    print("=== 测试生成新题目 ===")
    
    # 1. 先创建一个新目标
    create_goal_url = f"{base_url}/api/goals"
    goal_data = {
        "userId": 1,
        "subject": "数学",
        "goalType": "选择",
        "difficulty": "中等",
        "totalQuantity": 2,
        "goalContent": "测试difficulty字段的题目生成"
    }
    
    print(f"创建新目标: {goal_data}")
    create_response = requests.post(create_goal_url, json=goal_data)
    
    if create_response.status_code == 200:
        create_result = create_response.json()
        if create_result.get('code') == 200:
            goal_id = create_result['data']['goalId']
            print(f"✅ 目标创建成功，ID: {goal_id}")
            
            # 2. 等待一下让自动生成题目完成
            import time
            print("等待5秒让题目生成完成...")
            time.sleep(5)
            
            # 3. 获取生成的题目
            questions_url = f"{base_url}/api/ai/questions/{goal_id}"
            questions_response = requests.get(questions_url)
            
            if questions_response.status_code == 200:
                questions_result = questions_response.json()
                if questions_result.get('code') == 200:
                    questions = questions_result['data']
                    print(f"\n✅ 获取到 {len(questions)} 道题目")
                    
                    # 检查每道题目的difficulty字段
                    for i, question in enumerate(questions, 1):
                        difficulty = question.get('difficulty')
                        print(f"\n题目 {i}:")
                        print(f"  - detailId: {question.get('detailId')}")
                        print(f"  - difficulty: {difficulty}")
                        print(f"  - 题目内容: {question.get('question', '')[:50]}...")
                        
                        if difficulty:
                            print(f"  ✅ difficulty字段正常: {difficulty}")
                        else:
                            print(f"  ❌ difficulty字段为空或null")
                            
                else:
                    print(f"❌ 获取题目失败: {questions_result.get('message')}")
            else:
                print(f"❌ 获取题目请求失败: {questions_response.status_code}")
                print(f"响应内容: {questions_response.text}")
                
        else:
            print(f"❌ 创建目标失败: {create_result.get('message')}")
    else:
        print(f"❌ 创建目标请求失败: {create_response.status_code}")
        print(f"响应内容: {create_response.text}")

if __name__ == "__main__":
    test_generate_questions()