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"`
	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 {
	Model  string `json:"model,omitempty"`
	Prompt any    `json:"prompt,omitempty"`
}

type OllamaEmbeddingResponse struct {
	Embedding []float64 `json:"embedding,omitempty"`
}