mirror of
https://github.com/linux-do/new-api.git
synced 2025-09-17 07:56:38 +08:00
45 lines
1.2 KiB
Go
45 lines
1.2 KiB
Go
package cohere
|
|
|
|
type CohereRequest struct {
|
|
Model string `json:"model"`
|
|
ChatHistory []ChatHistory `json:"chat_history"`
|
|
Message string `json:"message"`
|
|
Stream bool `json:"stream"`
|
|
MaxTokens int64 `json:"max_tokens"`
|
|
}
|
|
|
|
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 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"`
|
|
}
|