构建自己的Kimi Ai终端助手

Fri, 05 Jul 2024 13:58:18 +0800

获取Moonshot API key并配置到环境变量

登录Moonshot AI开放平台, 在API管理 -> 新建 -> 输入key名, 确定-> 复制密钥sk-HL...Ehvcn.

将以下环境变量写入系统环境变量

1
2
3

GPT_BASE_URL=https://api.moonshot.cn/v1
GPT_API_KEY=sk-HLdyzmmUgXQnM5RbhyOBJs0Y5nvPlEdSBa8lWJGZ0GcEhvcn
GPT_MODEL=moonshot-v1-8k

或者写入项目下的.env文件并用github.com/joho/godotenv之类的库进行读取

读取环境变量代码:

// 如果有.env文件, 则导入库 go get -u github.com/joho/godotenv
godotenv.Load()

apiKey := os.Getenv("GPT_API_KEY")
baseUrl := os.Getenv("GPT_BASE_URL")
model := os.Getenv("GPT_MODEL")

使用go-gpt3构建Client

获取go-gpt3库 go get -u github.com/PullRequestInc/go-gpt3, go-gpt3是一个兼容OPEN-AI GPT3 API的开源库. 支持stream和non stream的completion请求.

单轮对话示例

构建gpt3.Client实例

`1`	`client := gpt3.NewClient(apiKey, gpt3.WithBaseURL(baseUrl))`

初始化一个对话, 这里使用moonshot文档提供的prompt

var history []gpt3.ChatCompletionRequestMessage = []gpt3.ChatCompletionRequestMessage{
	{
		Role:    "system",
		Content: "你是 Kimi，由 Moonshot AI 提供的人工智能助手，你更擅长中文和英文的对话。你会为用户提供安全，有帮助，准确的回答。同时，你会拒绝一切涉及恐怖主义，种族歧视，黄色暴力等问题的回答。Moonshot AI 为专有名词，不可翻译成其他语言。",
	},
}

添加一个问题到对话中, moonshoot ai支持的对话Role为system, user, assistant, tool

GPT on 我的绕地飞行记录

构建自己的Kimi Ai终端助手

获取Moonshot API key并配置到环境变量

使用go-gpt3构建Client

单轮对话示例