From 811dce4ead33fafc8afdde900c1bca9042c68d06 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=BB=84=20=E6=8C=AF?= <huangzhen@bangongyi.com>
Date: Tue, 8 Apr 2025 13:15:37 +0800
Subject: [PATCH] =?UTF-8?q?=E8=BE=93=E5=87=BA=E6=80=9D=E7=BB=B4=E9=93=BE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
doubao/doubaollm.go | 3 ++-
examples/doubao_example/doubao_test.go | 6 +++---
go.mod | 2 +-
go.sum | 2 ++
4 files changed, 8 insertions(+), 5 deletions(-)
diff --git a/doubao/doubaollm.go b/doubao/doubaollm.go
index 81be0f8..7ca864e 100644
--- a/doubao/doubaollm.go
+++ b/doubao/doubaollm.go
@@ -168,7 +168,8 @@ func (o *LLM) Chat(ctx context.Context, messages []*schema.ChatMessage, options
CompletionTokens: result.Usage.CompletionTokens,
TotalTokens: result.Usage.TotalTokens,
},
- Content: volcengine.StringValue(c.Message.Content.StringValue),
+ Content: volcengine.StringValue(c.Message.Content.StringValue),
+ ReasoningContent: volcengine.StringValue(c.Message.ReasoningContent),
}
choices = append(choices, cc)
diff --git a/examples/doubao_example/doubao_test.go b/examples/doubao_example/doubao_test.go
index 63291f8..a7a7fc9 100644
--- a/examples/doubao_example/doubao_test.go
+++ b/examples/doubao_example/doubao_test.go
@@ -25,7 +25,7 @@ var (
func TestChat(t *testing.T) {
ctx := context.Background()
- llm, err := doubao.New(doubao.WithVolcAccessKey(access), doubao.WithVolcSecretKey(secret))
+ llm, err := doubao.New(doubao.WithVolcAccessKey(access), doubao.WithVolcSecretKey(secret), doubao.WithModel(deepseekR1))
if err != nil {
fmt.Println(err)
return
@@ -40,8 +40,8 @@ func TestChat(t *testing.T) {
Content: "查询一下北京的天气？",
},
}
- resp, err := llm.Chat(ctx, messages, kpllms.WithTemperature(0.95),
- kpllms.WithModel(modelDoubaoPro32K))
+ resp, err := llm.Chat(ctx, messages, kpllms.WithTemperature(0.95))
+ // kpllms.WithModel(deepseekR1)
if err != nil {
fmt.Println(err.Error())
return
diff --git a/go.mod b/go.mod
index 43800bd..fda55d6 100644
--- a/go.mod
+++ b/go.mod
@@ -10,7 +10,7 @@ require (
github.com/pkoukk/tiktoken-go v0.1.7
github.com/tcolgate/mp3 v0.0.0-20170426193717-e79c5a46d300
github.com/volcengine/volc-sdk-golang v1.0.158
- github.com/volcengine/volcengine-go-sdk v1.0.183
+ github.com/volcengine/volcengine-go-sdk v1.1.1
k8s.io/kubernetes v1.13.0
)
diff --git a/go.sum b/go.sum
index 27463d4..eaeaf76 100644
--- a/go.sum
+++ b/go.sum
@@ -2662,6 +2662,8 @@ github.com/volcengine/volc-sdk-golang v1.0.158 h1:7VHA5SpMNFjVpskDl9uEzKNdeQXBzq
github.com/volcengine/volc-sdk-golang v1.0.158/go.mod h1:+epuCKzDYSqsciTxQdDOY5DCt8dNsmj2gRN6yI//fQo=
github.com/volcengine/volcengine-go-sdk v1.0.183 h1:2TuWnhuA6vb2sDYEb44ErGBVPLCPDW0s2JNgYb2RX+A=
github.com/volcengine/volcengine-go-sdk v1.0.183/go.mod h1:gfEDc1s7SYaGoY+WH2dRrS3qiuDJMkwqyfXWCa7+7oA=
+github.com/volcengine/volcengine-go-sdk v1.1.1 h1:ZtFi9iyCQ2SQzWHY55GGZFwK2eQ186LWyLUXxU2wX/s=
+github.com/volcengine/volcengine-go-sdk v1.1.1/go.mod h1:gfEDc1s7SYaGoY+WH2dRrS3qiuDJMkwqyfXWCa7+7oA=
github.com/willf/bitset v1.1.11-0.20200630133818-d5bec3311243/go.mod h1:RjeCKbqT1RxIR/KWY6phxZiaY1IyutSBfGjNPySAYV4=
github.com/willf/bitset v1.1.11/go.mod h1:83CECat5yLh5zVOf4P1ErAgKA5UDvKtgyUABdr3+MjI=
github.com/x448/float16 v0.8.4/go.mod h1:14CWIYCyZA/cWjXOioeEpHeN/83MdbZDRQHoFcYsOfg=
--
GitLab