|
|
@@ -3,14 +3,18 @@ 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"`
|
|
|
+ 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"`
|
|
|
+ Tools []dto.ToolCall `json:"tools,omitempty"`
|
|
|
+ ResponseFormat *dto.ResponseFormat `json:"response_format,omitempty"`
|
|
|
+ FrequencyPenalty float64 `json:"frequency_penalty,omitempty"`
|
|
|
+ PresencePenalty float64 `json:"presence_penalty,omitempty"`
|
|
|
}
|
|
|
|
|
|
type OllamaEmbeddingRequest struct {
|
|
|
@@ -21,6 +25,3 @@ type OllamaEmbeddingRequest struct {
|
|
|
type OllamaEmbeddingResponse struct {
|
|
|
Embedding []float64 `json:"embedding,omitempty"`
|
|
|
}
|
|
|
-
|
|
|
-//type OllamaOptions struct {
|
|
|
-//}
|