pkg_ai

各平台大模型api请求封装

安装

go get github.com/juxiaoming/pkg_ai

支持第三方登录

Ai大模型	参考文档	应用申请
	参考文档	应用申请
	参考文档	应用申请
	参考文档	应用申请

使用

// 初始化配置
pkg_ai.Init(pkg_ai.NewMoonshotConf("request_url" , "sk-your_key"))

// 初始化多服务配置
pkg_ai.Init(&pkg_ai.Config{...})

// 初始化服务
server, err := pkg_ai.NewServer(pkg_ai.ImplementMoonshot)
if err != nil {
    fmt.Println("服务初始化失败", err)
    return
}

// 请求数据
requestData := pkg_ai.RequestData{
    Model:     "moonshot-v1-8k",
    UserQuery: "帮我写出岳飞的满江红",
}

// 阻塞式请求
fmt.Println(server.Chat(data))

// 流式请求
msgChan, errChan, stopChan := make(chan string, 10000), make(chan error), make(chan struct{})
fmt.Println(server.ChatStream(data, msgChan, errChan, stopChan))

// 自定义请求参数
fmt.Println(server.CustomizeChat([]byte("{....}")))
fmt.Println(server.CustomizeChatStream([]byte("{....}"), msgChan, errChan, stopChan))

建议

建议初始化配置文件之后单次调用pkg_login.Init()方法注册服务配置

更多

等我！