mirror of
https://github.com/coaidev/coai.git
synced 2025-05-19 13:00:14 +09:00
130 lines
3.4 KiB
Go
130 lines
3.4 KiB
Go
package manager
|
|
|
|
import (
|
|
"chat/globals"
|
|
"chat/utils"
|
|
"fmt"
|
|
)
|
|
|
|
type Message struct {
|
|
Role string `json:"role"`
|
|
Content interface{} `json:"content"`
|
|
ToolCallId *string `json:"tool_call_id,omitempty"` // only `tool` role
|
|
ToolCalls *globals.ToolCalls `json:"tool_calls,omitempty"` // only `assistant` role
|
|
}
|
|
|
|
type ImageUrl struct {
|
|
Url string `json:"url"`
|
|
Detail *string `json:"detail,omitempty"`
|
|
}
|
|
|
|
type MessageContent struct {
|
|
Type string `json:"type"`
|
|
Text *string `json:"text,omitempty"`
|
|
ImageUrl *ImageUrl `json:"image_url,omitempty"`
|
|
}
|
|
|
|
type MessageContents []MessageContent
|
|
|
|
type TranshipmentForm struct {
|
|
Model string `json:"model" binding:"required"`
|
|
Messages []Message `json:"messages" binding:"required"`
|
|
Stream bool `json:"stream"`
|
|
MaxTokens int `json:"max_tokens"`
|
|
PresencePenalty *float32 `json:"presence_penalty"`
|
|
FrequencyPenalty *float32 `json:"frequency_penalty"`
|
|
RepetitionPenalty *float32 `json:"repetition_penalty"`
|
|
Temperature *float32 `json:"temperature"`
|
|
TopP *float32 `json:"top_p"`
|
|
TopK *int `json:"top_k"`
|
|
Tools *globals.FunctionTools
|
|
ToolChoice *interface{}
|
|
Official bool `json:"official"`
|
|
}
|
|
|
|
type Choice struct {
|
|
Index int `json:"index"`
|
|
Message globals.Message `json:"message"`
|
|
FinishReason string `json:"finish_reason"`
|
|
}
|
|
|
|
type Usage struct {
|
|
PromptTokens int `json:"prompt_tokens"`
|
|
CompletionTokens int `json:"completion_tokens"`
|
|
TotalTokens int `json:"total_tokens"`
|
|
}
|
|
|
|
type TranshipmentResponse struct {
|
|
Id string `json:"id"`
|
|
Object string `json:"object"`
|
|
Created int64 `json:"created"`
|
|
Model string `json:"model"`
|
|
Choices []Choice `json:"choices"`
|
|
Usage Usage `json:"usage"`
|
|
Quota *float32 `json:"quota,omitempty"`
|
|
}
|
|
|
|
type ChoiceDelta struct {
|
|
Index int `json:"index"`
|
|
Delta globals.Message `json:"delta"`
|
|
FinishReason interface{} `json:"finish_reason"`
|
|
}
|
|
|
|
type TranshipmentStreamResponse struct {
|
|
Id string `json:"id"`
|
|
Object string `json:"object"`
|
|
Created int64 `json:"created"`
|
|
Model string `json:"model"`
|
|
Choices []ChoiceDelta `json:"choices"`
|
|
Usage Usage `json:"usage"`
|
|
Quota *float32 `json:"quota,omitempty"`
|
|
Error error `json:"error,omitempty"`
|
|
}
|
|
|
|
type TranshipmentErrorResponse struct {
|
|
Error TranshipmentError `json:"error"`
|
|
}
|
|
|
|
type TranshipmentError struct {
|
|
Message string `json:"message"`
|
|
Type string `json:"type"`
|
|
}
|
|
|
|
func transformContent(content interface{}) string {
|
|
switch v := content.(type) {
|
|
case string:
|
|
return v
|
|
default:
|
|
var result string
|
|
data := utils.MapToStruct[MessageContents](v)
|
|
if data == nil || len(*data) == 0 {
|
|
return ""
|
|
}
|
|
|
|
for _, v := range *data {
|
|
if v.Text != nil {
|
|
result += *v.Text
|
|
}
|
|
|
|
if v.ImageUrl != nil {
|
|
result += fmt.Sprintf(" %s ", v.ImageUrl.Url)
|
|
}
|
|
}
|
|
return result
|
|
}
|
|
}
|
|
|
|
func transform(m []Message) []globals.Message {
|
|
var messages []globals.Message
|
|
for _, v := range m {
|
|
fmt.Println(transformContent(v.Content))
|
|
messages = append(messages, globals.Message{
|
|
Role: v.Role,
|
|
Content: transformContent(v.Content),
|
|
ToolCallId: v.ToolCallId,
|
|
ToolCalls: v.ToolCalls,
|
|
})
|
|
}
|
|
return messages
|
|
}
|