ollama

type AnalysisPrompt struct {
	Type        string
	Template    string
	MaxTokens   int
	Temperature float32
}

type OllamaAgent struct {
	// contains filtered or unexported fields
}

type OllamaAgentOptions struct {
	Endpoint    string        // Ollama server endpoint (default: http://localhost:11434)
	Model       string        // Model name (e.g., "codellama:13b", "llama2:7b")
	Timeout     time.Duration // Request timeout
	MaxTokens   int           // Maximum tokens in response
	Temperature float32       // Response creativity (0.0 to 1.0)
}

type OllamaModelInfo struct {
	Name       string    `json:"name"`
	Size       int64     `json:"size"`
	Digest     string    `json:"digest"`
	ModifiedAt time.Time `json:"modified_at"`
}

type OllamaModelsResponse struct {
	Models []OllamaModelInfo `json:"models"`
}

type OllamaRequest struct {
	Model   string                 `json:"model"`
	Prompt  string                 `json:"prompt"`
	Stream  bool                   `json:"stream"`
	Options map[string]interface{} `json:"options,omitempty"`
	Context []int                  `json:"context,omitempty"`
}

type OllamaResponse struct {
	Model              string `json:"model"`
	CreatedAt          string `json:"created_at"`
	Response           string `json:"response"`
	Done               bool   `json:"done"`
	Context            []int  `json:"context,omitempty"`
	TotalDuration      int64  `json:"total_duration,omitempty"`
	LoadDuration       int64  `json:"load_duration,omitempty"`
	PromptEvalCount    int    `json:"prompt_eval_count,omitempty"`
	PromptEvalDuration int64  `json:"prompt_eval_duration,omitempty"`
	EvalCount          int    `json:"eval_count,omitempty"`
	EvalDuration       int64  `json:"eval_duration,omitempty"`
}