Upstream/waveterm
1
0
Fork 0
mirror of https://github.com/wavetermdev/waveterm.git synced 2025-01-02 18:39:05 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: typecast fix for new ai library version

Browse Source
This commit is contained in:
Sylvia Crowe 2024-09-17 14:12:24 -07:00
parent 9014660dc3
commit a2a5e5afdb
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
wavesrv/pkg/remote/openai/openai.go
View File

@ -190,7 +190,7 @@ func RunCompletionStream(ctx context.Context, opts *sstore.OpenAIOptsType, promp
pk := packet.MakeOpenAIPacket() pk := packet.MakeOpenAIPacket()
pk.Index = choice.Index pk.Index = choice.Index
pk.Text = choice.Delta.Content pk.Text = choice.Delta.Content
pk.FinishReason = choice.FinishReason pk.FinishReason = string(choice.FinishReason)
rtn <- pk rtn <- pk
} }
} }
@ -209,7 +209,7 @@ func marshalResponse(resp openaiapi.ChatCompletionResponse) []*packet.OpenAIPack
choicePk := packet.MakeOpenAIPacket() choicePk := packet.MakeOpenAIPacket()
choicePk.Index = choice.Index choicePk.Index = choice.Index
choicePk.Text = choice.Message.Content choicePk.Text = choice.Message.Content
choicePk.FinishReason = choice.FinishReason choicePk.FinishReason = string(choice.FinishReason)
rtn = append(rtn, choicePk) rtn = append(rtn, choicePk)
} }
return rtn return rtn