xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-17 13:43:42 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

🔖 chore: Rename relay/util to relay/relay_util package and add utils package

Browse Source
This commit is contained in:
MartialBE
2024-05-29 00:36:54 +08:00
parent 853f2681f4
commit 79524108a3
61 changed files with 309 additions and 265 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
providers/cloudflareAI/chat.go
View File

@@ -7,6 +7,7 @@ import (
"net/http"
"one-api/common"
"one-api/common/requester"
"one-api/common/utils"
"one-api/types"
"strings"
)
@@ -85,9 +86,9 @@ func (p *CloudflareAIProvider) convertToChatOpenai(response *ChatRespone, reques
}
openaiResponse = &types.ChatCompletionResponse{
ID: fmt.Sprintf("chatcmpl-%s", common.GetUUID()),
ID: fmt.Sprintf("chatcmpl-%s", utils.GetUUID()),
Object: "chat.completion",
Created: common.GetTimestamp(),
Created: utils.GetTimestamp(),
Model: request.Model,
Choices: []types.ChatCompletionChoice{{
Index: 0,
@@ -155,9 +156,9 @@ func (h *CloudflareAIStreamHandler) handlerStream(rawLine *[]byte, dataChan chan
func (h *CloudflareAIStreamHandler) convertToOpenaiStream(chatResponse *ChatResult, dataChan chan string, isStop bool) {
streamResponse := types.ChatCompletionStreamResponse{
ID: fmt.Sprintf("chatcmpl-%s", common.GetUUID()),
ID: fmt.Sprintf("chatcmpl-%s", utils.GetUUID()),
Object: "chat.completion.chunk",
Created: common.GetTimestamp(),
Created: utils.GetTimestamp(),
Model: h.Request.Model,
}

3
providers/cloudflareAI/image_generations.go
View File

@@ -6,6 +6,7 @@ import (
"net/http"
"one-api/common"
"one-api/common/storage"
"one-api/common/utils"
"one-api/types"
"time"
)
@@ -46,7 +47,7 @@ func (p *CloudflareAIProvider) CreateImageGenerations(request *types.ImageReques
url := ""
if request.ResponseFormat == "" || request.ResponseFormat == "url" {
url = storage.Upload(body, common.GetUUID()+".png")
url = storage.Upload(body, utils.GetUUID()+".png")
}
openaiResponse := &types.ImageResponse{