mirror of
https://github.com/linux-do/new-api.git
synced 2025-09-17 16:06:38 +08:00
61 lines
1.7 KiB
Go
61 lines
1.7 KiB
Go
package cohere
|
|
|
|
import "one-api/dto"
|
|
|
|
type CohereRequest struct {
|
|
Model string `json:"model"`
|
|
ChatHistory []ChatHistory `json:"chat_history"`
|
|
Message string `json:"message"`
|
|
Stream bool `json:"stream"`
|
|
MaxTokens int `json:"max_tokens"`
|
|
SafetyMode string `json:"safety_mode,omitempty"`
|
|
}
|
|
|
|
type ChatHistory struct {
|
|
Role string `json:"role"`
|
|
Message string `json:"message"`
|
|
}
|
|
|
|
type CohereResponse struct {
|
|
IsFinished bool `json:"is_finished"`
|
|
EventType string `json:"event_type"`
|
|
Text string `json:"text,omitempty"`
|
|
FinishReason string `json:"finish_reason,omitempty"`
|
|
Response *CohereResponseResult `json:"response"`
|
|
}
|
|
|
|
type CohereResponseResult struct {
|
|
ResponseId string `json:"response_id"`
|
|
FinishReason string `json:"finish_reason,omitempty"`
|
|
Text string `json:"text"`
|
|
Meta CohereMeta `json:"meta"`
|
|
}
|
|
|
|
type CohereRerankRequest struct {
|
|
Documents []any `json:"documents"`
|
|
Query string `json:"query"`
|
|
Model string `json:"model"`
|
|
TopN int `json:"top_n"`
|
|
ReturnDocuments bool `json:"return_documents"`
|
|
}
|
|
|
|
type CohereRerankResponseResult struct {
|
|
Results []dto.RerankResponseDocument `json:"results"`
|
|
Meta CohereMeta `json:"meta"`
|
|
}
|
|
|
|
type CohereMeta struct {
|
|
//Tokens CohereTokens `json:"tokens"`
|
|
BilledUnits CohereBilledUnits `json:"billed_units"`
|
|
}
|
|
|
|
type CohereBilledUnits struct {
|
|
InputTokens int `json:"input_tokens"`
|
|
OutputTokens int `json:"output_tokens"`
|
|
}
|
|
|
|
type CohereTokens struct {
|
|
InputTokens int `json:"input_tokens"`
|
|
OutputTokens int `json:"output_tokens"`
|
|
}
|