mirror of
https://github.com/linux-do/new-api.git
synced 2025-09-17 16:06:38 +08:00
27 lines
739 B
Go
27 lines
739 B
Go
package ollama
|
|
|
|
import "one-api/dto"
|
|
|
|
type OllamaRequest struct {
|
|
Model string `json:"model,omitempty"`
|
|
Messages []dto.Message `json:"messages,omitempty"`
|
|
Stream bool `json:"stream,omitempty"`
|
|
Temperature float64 `json:"temperature,omitempty"`
|
|
Seed float64 `json:"seed,omitempty"`
|
|
Topp float64 `json:"top_p,omitempty"`
|
|
TopK int `json:"top_k,omitempty"`
|
|
Stop any `json:"stop,omitempty"`
|
|
}
|
|
|
|
type OllamaEmbeddingRequest struct {
|
|
Model string `json:"model,omitempty"`
|
|
Prompt any `json:"prompt,omitempty"`
|
|
}
|
|
|
|
type OllamaEmbeddingResponse struct {
|
|
Embedding []float64 `json:"embedding,omitempty"`
|
|
}
|
|
|
|
//type OllamaOptions struct {
|
|
//}
|