文心千帆 GO SDK

本库为文心千帆GO语言SDK，非官方库,目前官方还没有GO语言的SDK 文心千帆

已支持自动刷新 accessToken, 需要通过 ernie.NewDefaultClient("API Key", "Secret Key") 方式初始化客户端

目前支持的模型：

ERNIE-Bot-4
ERNIE-Bot
ERNIE-Bot-turbo
ERNIE-Bot-8k
ERNIE-Bot-turbo-AI
BLOOMZ-7B
Llama-2
Embeddings
百度云所有模型

安装

go get github.com/anhao/go-ernie

需要 go 版本为 1.18+

使用示例

package main

import (
	"context"
	"fmt"
	ernie "github.com/anhao/go-ernie"
)

func main() {

	client := ernie.NewDefaultClient("API Key", "Secret Key")
	completion, err := client.CreateErnieBotChatCompletion(context.Background(), ernie.ErnieBotRequest{
		Messages: []ernie.ChatCompletionMessage{
			{
				Role:    ernie.MessageRoleUser,
				Content: "你好呀",
			},
		},
	})
	if err != nil {
		fmt.Printf("ernie bot error: %v\n", err)
		return
	}
	fmt.Println(completion)
}

获取文心千帆 APIKEY

在百度云官网进行申请：https://cloud.baidu.com/product/wenxinworkshop
申请通过后创建应用：https://console.bce.baidu.com/qianfan/ais/console/applicationConsole/application
获取 apikey 和 api secret

其他示例

ERNIE-Bot 4.0 对话

import (
	"context"
	"errors"
	"fmt"
	ernie "github.com/anhao/go-ernie"
	"io"
)

func main() {

	client := ernie.NewDefaultClient("API Key", "Secret Key")
	request := ernie.ErnieBot4Request{
		Messages: []ChatCompletionMessage{
			{
				Role:    "user",
				Content: "Hello",
			},
		},
		Stream: true,
	}

	stream, err := client.CreateErnieBot4ChatCompletionStream(context.Background(), request)
	if err != nil {
		fmt.Printf("ernie bot stream error: %v\n", err)
		return
	}
	defer stream.Close()
	for {
		response, err := stream.Recv()
		if errors.Is(err, io.EOF) {
			fmt.Println("ernie bot 4 Stream finished")
			return
		}
		if err != nil {
			fmt.Printf("ernie bot 4 stream error: %v\n", err)
			return
		}
		fmt.Println(response.Result)
	}
}

ERNIE-Bot stream流对话

import (
	"context"
	"errors"
	"fmt"
	ernie "github.com/anhao/go-ernie"
	"io"
)

func main() {

	client := ernie.NewDefaultClient("API Key", "Secret Key")
	request := ernie.ErnieBotRequest{
		Messages: []ChatCompletionMessage{
			{
				Role:    "user",
				Content: "Hello",
			},
		},
		Stream: true,
	}

	stream, err := client.CreateErnieBotChatCompletionStream(context.Background(), request)
	if err != nil {
		fmt.Printf("ernie bot stream error: %v\n", err)
		return
	}
	defer stream.Close()
	for {
		response, err := stream.Recv()
		if errors.Is(err, io.EOF) {
			fmt.Println("ernie bot Stream finished")
			return
		}
		if err != nil {
			fmt.Printf("ernie bot stream error: %v\n", err)
			return
		}
		fmt.Println(response.Result)
	}
}

ERNIE-Bot-Turbo 对话

package main

import (
	"context"
	"fmt"
	ernie "github.com/anhao/go-ernie"
)

func main() {

	client := ernie.NewDefaultClient("API Key", "Secret Key")
	completion, err := client.CreateErnieBotTurboChatCompletion(context.Background(), ernie.ErnieBotTurboRequest{
		Messages: []ernie.ChatCompletionMessage{
			{
				Role:    ernie.MessageRoleUser,
				Content: "你好呀",
			},
		},
	})
	if err != nil {
		fmt.Printf("ernie bot turbo error: %v\n", err)
		return
	}
	fmt.Println(completion)
}

ERNIE-Bot Turbo stream流对话

import (
	"context"
	"errors"
	"fmt"
	ernie "github.com/anhao/go-ernie"
	"io"
)

func main() {

	client := ernie.NewDefaultClient("API Key", "Secret Key")
	request := ernie.ErnieBotTurboRequest{
		Messages: []ChatCompletionMessage{
			{
				Role:    "user",
				Content: "Hello",
			},
		},
		Stream: true,
	}

	stream, err := client.CreateErnieBotTurboChatCompletionStream(context.Background(), request)
	if err != nil {
		fmt.Printf("ernie bot stream error: %v\n", err)
		return
	}
	defer stream.Close()
	for {
		response, err := stream.Recv()
		if errors.Is(err, io.EOF) {
			fmt.Println("ernie bot turbo Stream finished")
			return
		}
		if err != nil {
			fmt.Printf("ernie bot turbo stream error: %v\n", err)
			return
		}
		fmt.Println(response.Result)
	}
}

BLOOMZ-7B 对话

package main

import (
	"context"
	"fmt"
	ernie "github.com/anhao/go-ernie"
)

func main() {

	client := ernie.NewDefaultClient("API Key", "Secret Key")
	completion, err := client.CreateBloomz7b1ChatCompletion(context.Background(), ernie.Bloomz7b1Request{
		Messages: []ernie.ChatCompletionMessage{
			{
				Role:    ernie.MessageRoleUser,
				Content: "你好呀",
			},
		},
	})
	if err != nil {
		fmt.Printf("BLOOMZ-7B error: %v\n", err)
		return
	}
	fmt.Println(completion)
}

BLOOMZ-7B stream流对话

import (
	"context"
	"errors"
	"fmt"
	ernie "github.com/anhao/go-ernie"
	"io"
)

func main() {

	client := ernie.NewDefaultClient("API Key", "Secret Key")
	request := ernie.Bloomz7b1Request{
		Messages: []ChatCompletionMessage{
			{
				Role:    "user",
				Content: "Hello",
			},
		},
		Stream: true,
	}

	stream, err := client.CreateBloomz7b1ChatCompletionStream(context.Background(), request)
	if err != nil {
		fmt.Printf("BLOOMZ-7B error: %v\n", err)
		return
	}
	defer stream.Close()
	for {
		response, err := stream.Recv()
		if errors.Is(err, io.EOF) {
			fmt.Println("BLOOMZ-7B  Stream finished")
			return
		}
		if err != nil {
			fmt.Printf("BLOOMZ-7B stream error: %v\n", err)
			return
		}
		fmt.Println(response.Result)
	}
}

Llama2 对话

package main

import (
	"context"
	"fmt"
	ernie "github.com/anhao/go-ernie"
)

func main() {

	client := ernie.NewDefaultClient("API Key", "Secret Key")
	completion, err := client.CreateLlamaChatCompletion(context.Background(), ernie.LlamaChatRequest{
		Messages: []ernie.ChatCompletionMessage{
			{
				Role:    ernie.MessageRoleUser,
				Content: "你好呀",
			},
		},
		Model: "", //申请发布时填写的API名称
	})
	if err != nil {
		fmt.Printf("llama2 error: %v\n", err)
		return
	}
	fmt.Println(completion)
}

Llama2 stream流对话

import (
	"context"
	"errors"
	"fmt"
	ernie "github.com/anhao/go-ernie"
	"io"
)

func main() {

	client := ernie.NewDefaultClient("API Key", "Secret Key")
	request := ernie.LlamaChatRequest{
		Messages: []ChatCompletionMessage{
			{
				Role:    "user",
				Content: "Hello",
			},
		},
		Stream: true,
		Model: "", //申请发布时填写的API名称
	}

	stream, err := client.CreateLlamaChatCompletionStream(context.Background(), request)
	if err != nil {
		fmt.Printf("llama2 error: %v\n", err)
		return
	}
	defer stream.Close()
	for {
		response, err := stream.Recv()
		if errors.Is(err, io.EOF) {
			fmt.Println("llama2  Stream finished")
			return
		}
		if err != nil {
			fmt.Printf("llama2 stream error: %v\n", err)
			return
		}
		fmt.Println(response.Result)
	}
}

Embedding向量

package main

import (
	"context"
	"fmt"
	ernie "github.com/anhao/go-ernie"
)

func main() {

	client := ernie.NewDefaultClient("API Key", "Secret Key")
	request := ernie.EmbeddingRequest{
		Input: []string{
			"Hello",
		},
	}
	embeddings, err := client.CreateEmbeddings(context.Background(), request)
	if err != nil {
		fmt.Sprintf("embeddings err: %v", err)
		return
	}
	fmt.Println(embeddings)
}

自定义 accessToken

client :=ernie.NewClient("accessToken")

百度云其他Chat模型

百度云的其他模型需要自己部署，部署需要填写一个API地址

import (
	"context"
	"errors"
	"fmt"
	ernie "github.com/anhao/go-ernie"
	"io"
)

func main() {

	client := ernie.NewDefaultClient("API Key", "Secret Key")
	request := ernie.BaiduChatRequest{
		Messages: []ChatCompletionMessage{
			{
				Role:    "user",
				Content: "Hello",
			},
		},
		Stream: true,
		Model: "", //申请发布时填写的API名称
	}

	stream, err := client.CreateBaiduChatCompletionStream(context.Background(), request)
	if err != nil {
		fmt.Printf("baidu chat error: %v\n", err)
		return
	}
	defer stream.Close()
	for {
		response, err := stream.Recv()
		if errors.Is(err, io.EOF) {
			fmt.Println("baidu chat  Stream finished")
			return
		}
		if err != nil {
			fmt.Printf("baidu chat stream error: %v\n", err)
			return
		}
		fmt.Println(response.Result)
	}
}

文本生成图片

百度云的其他模型需要自己部署，部署需要填写一个API地址

import (
	"context"
	"errors"
	"fmt"
	ernie "github.com/anhao/go-ernie"
	"io"
)

func main() {

	client := ernie.NewDefaultClient("API Key", "Secret Key")
	request := ernie.BaiduTxt2ImgRequest{
		Model: "", //申请发布时填写的API名称
		Prompt: "",//提示词
	}

    response, err := client.CreateBaiduTxt2Img(context.Background(), request)
	if err != nil {
		fmt.Printf("baidu txt2img error: %v\n", err)
		return
	}
	fmt.Println(response)
}

插件应用

import (
	"context"
	"errors"
	"fmt"
	ernie "github.com/anhao/go-ernie"
	"io"
)

func main() {

	client := ernie.NewDefaultClient("API Key", "Secret Key")
	request := ernie.ErnieCustomPluginRequest{
		PluginName: "xxx", // 插件名称
		Query:      "你好",// 查询信息
		Stream:     false,
	}

    response, err := client.CreateErnieCustomPluginChatCompletion(context.Background(), request)
	if err != nil {
		fmt.Printf("ernie custom plugins error: %v\n", err)
		return
	}
	fmt.Println(response)
}

Name		Name	Last commit message	Last commit date
Latest commit History 22 Commits
internal		internal
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
baidu_chat.go		baidu_chat.go
baidu_chat_stream.go		baidu_chat_stream.go
baidu_txt2img.go		baidu_txt2img.go
bge_large_en.go		bge_large_en.go
bge_large_zh.go		bge_large_zh.go
bloomz_7b1.go		bloomz_7b1.go
bloomz_7b1_stream.go		bloomz_7b1_stream.go
bloomz_7b1_test.go		bloomz_7b1_test.go
cache.go		cache.go
cache_test.go		cache_test.go
client.go		client.go
completion.go		completion.go
completion_stream.go		completion_stream.go
config.go		config.go
embedding.go		embedding.go
embedding_test.go		embedding_test.go
ernie_bot.go		ernie_bot.go
ernie_bot_4.go		ernie_bot_4.go
ernie_bot_4_stream.go		ernie_bot_4_stream.go
ernie_bot_4_test.go		ernie_bot_4_test.go
ernie_bot_8k.go		ernie_bot_8k.go
ernie_bot_8k_stream.go		ernie_bot_8k_stream.go
ernie_bot_8k_test.go		ernie_bot_8k_test.go
ernie_bot_stream.go		ernie_bot_stream.go
ernie_bot_test.go		ernie_bot_test.go
ernie_bot_turbo.go		ernie_bot_turbo.go
ernie_bot_turbo_ai.go		ernie_bot_turbo_ai.go
ernie_bot_turbo_ai_stream.go		ernie_bot_turbo_ai_stream.go
ernie_bot_turbo_stream.go		ernie_bot_turbo_stream.go
ernie_bot_turbo_test.go		ernie_bot_turbo_test.go
ernie_custom_plugin.go		ernie_custom_plugin.go
ernie_custom_plugin_stream.go		ernie_custom_plugin_stream.go
ernie_custom_plugin_test.go		ernie_custom_plugin_test.go
error.go		error.go
go.mod		go.mod
image.go		image.go
llama_chat.go		llama_chat.go
llama_chat_stream.go		llama_chat_stream.go
stream.go		stream.go
stream_reader.go		stream_reader.go

License

anhao/go-ernie

Folders and files

Latest commit

History

Repository files navigation

文心千帆 GO SDK

安装

使用示例

获取文心千帆 APIKEY

其他示例

About

Topics

Resources

License

Stars

Watchers

Forks

Languages