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