48 lines
1.1 KiB
Python
48 lines
1.1 KiB
Python
#!/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() |