#!/usr/bin/env python3
"""
使用月之暗面 Moonshot AI 进行代码生成
月之暗面提供强大的中文代码生成能力
"""

import os
from openai import OpenAI

def generate_code(client, description):
    """使用月之暗面 API 生成代码"""
    try:
        response = client.chat.completions.create(
            model="moonshot-v1-8k",  # 月之暗面模型
            messages=[
                {
                    "role": "system", 
                    "content": "你是一个专业的程序员助手，专门帮助生成高质量的代码。"
                },
                {
                    "role": "user", 
                    "content": f"请生成以下功能的 Python 代码：{description}"
                }
            ],
            max_tokens=500,
            temperature=0.1
        )
        
        return response.choices[0].message.content
        
    except Exception as e:
        return f"错误: {e}"

def main():
    # 使用月之暗面的配置
    api_key = "sk-lEk0pAIZ1EDgUkflq2is5uT2VbhuoKGpO5sNOSnuuccsD68r"
    base_url = "https://api.moonshot.cn/v1"
    
    # 初始化客户端
    client = OpenAI(
        api_key=api_key,
        base_url=base_url
    )
    
    # 示例：生成不同类型的代码
    examples = [
        "一个计算两个数字最大公约数的函数",
        "一个简单的 HTTP 服务器类",
        "一个数据验证装饰器"
    ]
    
    for i, description in enumerate(examples, 1):
        print(f"\n=== 示例 {i}: {description} ===")
        code = generate_code(client, description)
        print(code)
        print("-" * 50)

if __name__ == "__main__":
    main()