#!/usr/bin/env python3
"""
月之暗面 Moonshot AI 配置文件
"""

from openai import OpenAI

# 月之暗面配置
MOONSHOT_API_KEY = "sk-lEk0pAIZ1EDgUkflq2is5uT2VbhuoKGpO5sNOSnuuccsD68r"
MOONSHOT_BASE_URL = "https://api.moonshot.cn/v1"

# 可用的模型
MOONSHOT_MODELS = [
    "moonshot-v1-8k",    # 8K 上下文
    "moonshot-v1-32k",   # 32K 上下文  
    "moonshot-v1-128k"   # 128K 上下文
]

def get_moonshot_client():
    """获取月之暗面客户端"""
    return OpenAI(
        api_key=MOONSHOT_API_KEY,
        base_url=MOONSHOT_BASE_URL
    )

def test_connection():
    """测试连接"""
    client = get_moonshot_client()
    
    try:
        response = client.chat.completions.create(
            model="moonshot-v1-8k",
            messages=[
                {"role": "user", "content": "你好，请简单介绍一下你自己"}
            ],
            max_tokens=100
        )
        
        print("连接成功！")
        print("回复:", response.choices[0].message.content)
        return True
        
    except Exception as e:
        print(f"连接失败: {e}")
        return False

if __name__ == "__main__":
    test_connection()