Function generate_initial_solution

tests/leetcode_reflection_agent.py:424–453

生成初始解决方案

(
    task: Task,
    model: str,
)

async def generate_initial_solution(
    task: Task,
    model: str,
) -> CodeResponse:
    """生成初始解决方案"""
    question_text = task.input
    system_prompt_text = task.system_prompt
    
    # 检查模型是否支持视觉
    if model in NON_VISION_MODELS:
        question_content = question_text
    else:
        question_content = parse_markdown_with_images(question_text)
    
    messages = [
        SystemMessage(content=system_prompt_text),
        HumanMessage(content=question_content)
    ]
    
    response = await model_manager(
        model=model,
        messages=messages,
        response_format=CodeResponse,
        max_completion_tokens=65536
    )
    
    if response.success:
        return response.extra.parsed_model
    else:
        raise Exception(f"Model API Error: {response.message}")


def should_stop_optimization(eval_result: Dict[str, Any], optimization_threshold: float = 50.0) -> tuple[bool, str]: