Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

previous code post only last message, modify it to post all history messages

Browse Source
This commit is contained in:
fxsome 2024-06-07 21:07:31 +08:00
parent 9321427c6e
commit e3a41391bd
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
relay/adaptor/cloudflare/main.go
View File

@ -17,15 +17,21 @@ import (
) )
func ConvertRequest(textRequest model.GeneralOpenAIRequest) *Request { func ConvertRequest(textRequest model.GeneralOpenAIRequest) *Request {
lastMessage := textRequest.Messages[len(textRequest.Messages)-1] var promptBuilder strings.Builder
return &Request{ for _, message := range textRequest.Messages {
MaxTokens: textRequest.MaxTokens, promptBuilder.WriteString(message.StringContent())
Prompt: lastMessage.StringContent(), promptBuilder.WriteString("\n") // 添加换行符来分隔每个消息
Stream: textRequest.Stream, }
Temperature: textRequest.Temperature,
} return &Request{
MaxTokens: textRequest.MaxTokens,
Prompt: promptBuilder.String(),
Stream: textRequest.Stream,
Temperature: textRequest.Temperature,
}
} }
func ResponseCloudflare2OpenAI(cloudflareResponse *Response) *openai.TextResponse { func ResponseCloudflare2OpenAI(cloudflareResponse *Response) *openai.TextResponse {
choice := openai.TextResponseChoice{ choice := openai.TextResponseChoice{
Index: 0, Index: 0,