单测修复

baidubce · Dec 18, 2024 · 2473ce3 · 2473ce3
1 parent 3db1f96
commit 2473ce3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/go/qianfan/chat_completion.go b/go/qianfan/chat_completion.go
@@ -330,7 +330,7 @@ func (c *ChatCompletion) Do(ctx context.Context, request *ChatCompletionRequest)
 		return nil, runErr
 	}
 
-	if resp.ModelAPIError.ErrorMsg != "" {
+	if resp != nil && resp.ModelAPIError.ErrorMsg != "" {
 		return nil, fmt.Errorf(
 			"code: %d, message: %s",
 			resp.ModelAPIError.ErrorCode,

diff --git a/go/qianfan/completion.go b/go/qianfan/completion.go
@@ -135,7 +135,7 @@ func (c *Completion) Do(ctx context.Context, request *CompletionRequest) (*Model
 		return nil, runErr
 	}
 
-	if resp.ModelAPIError.ErrorMsg != "" {
+	if resp != nil && resp.ModelAPIError.ErrorMsg != "" {
 		return nil, fmt.Errorf(
 			"code: %d, message: %s",
 			resp.ModelAPIError.ErrorCode,