Moonshot AI API使用(2)-环境变量配置和简单对话demo

Sophie'sCookingLab

已于 2024-06-07 17:02:20 修改

阅读量995

点赞数 9

分类专栏：大模型文章标签：语言模型 nlp

于 2024-06-07 15:57:15 首次发布

本文链接：https://blog.csdn.net/weixin_40566713/article/details/139528937

版权

1、添加环境变量

api key有多种使用方式，添加环境变量这种是最安全的，强烈推荐！

打开文件

vi ~/.env

在打开的文件中添加下面两行

export MOONSHOT_API_KEY="你的MOONSHOT API KEY"
export MOONSHOT_API_URL="https://api.moonshot.cn/v1"

保存并退出

:wq

2、安装需要的包

pip install -upgrade openai
pip install python-dotenv  # 安装dotenv，用于加载环境变量

3、查看可用的模型

不同级别的模型使用时的价格是不一样的，具体参考官网。

import os
from openai import OpenAI
from dotenv import load_dotenv, find_dotenv
_ = load_dotenv(find_dotenv())  # 读取本地 .env 文件，里面定义了 MOONSHOT_API_KEY和 MOONSHOT_API_URL

client = OpenAI(
    api_key = os.getenv("MOONSHOT_API_KEY"),
    base_url = os.getenv("MOONSHOT_API_URL")
)
 
model_list = client.models.list()
model_data = model_list.data
 
for i, model in enumerate(model_data):
    print(f"model[