mirror of
https://github.com/coaidev/coai.git
synced 2025-05-22 06:20:14 +09:00
40 lines
1.1 KiB
Go
40 lines
1.1 KiB
Go
package dashscope
|
|
|
|
import "chat/globals"
|
|
|
|
// ChatRequest is the request body for dashscope
|
|
type ChatRequest struct {
|
|
Model string `json:"model"`
|
|
Input ChatInput `json:"input"`
|
|
Parameters ChatParam `json:"parameters"`
|
|
}
|
|
|
|
type ChatInput struct {
|
|
Prompt string `json:"prompt"`
|
|
Messages []globals.Message `json:"messages"`
|
|
}
|
|
|
|
type ChatParam struct {
|
|
IncrementalOutput bool `json:"incremental_output"`
|
|
EnableSearch *bool `json:"enable_search,omitempty"`
|
|
MaxTokens *int `json:"max_tokens,omitempty"`
|
|
Temperature *float32 `json:"temperature,omitempty"`
|
|
TopP *float32 `json:"top_p,omitempty"`
|
|
TopK *int `json:"top_k,omitempty"`
|
|
RepetitionPenalty *float32 `json:"repetition_penalty,omitempty"`
|
|
}
|
|
|
|
// ChatResponse is the response body for dashscope
|
|
type ChatResponse struct {
|
|
Output struct {
|
|
FinishReason string `json:"finish_reason"`
|
|
Text string `json:"text"`
|
|
} `json:"output"`
|
|
RequestId string `json:"request_id"`
|
|
Usage struct {
|
|
InputTokens int `json:"input_tokens"`
|
|
OutputTokens int `json:"output_tokens"`
|
|
} `json:"usage"`
|
|
Message string `json:"message"`
|
|
}
|