ollama

type ChatMessage struct {
	ToolCalls []ToolCall `json:"tool_calls,omitempty"`
	Role      string     `json:"role"`
	Content   string     `json:"content"`
}

type ChatRequest struct {
	Model       string        `json:"model"`
	Messages    []ChatMessage `json:"messages"`
	KeepAlive   int           `json:"keep_alive"`
	ContextSize int           `json:"context_size"`
}

type ChatResponse struct {
	Model      string      `json:"model"`
	CreatedAt  time.Time   `json:"created_at"`
	Message    ChatMessage `json:"message"`
	DoneReason string      `json:"done_reason,omitempty"`

	Done bool `json:"done"`
}

type ChatResponseHandler func(ChatResponse) error

type Client struct {
	Endpoint string `json:"endpoint"`
	Model    string `json:"model,omitempty"`
	Style    Style  `json:"style,omitempty"`
	// contains filtered or unexported fields
}

type EmbedRequest struct {
	Model string `json:"model,omitempty"`
	Input any    `json:"input"`
}

type EmbedResponse struct {
	Model      string      `json:"model"`
	Embeddings [][]float32 `json:"embeddings"`
}

type EmbedResponseHandler func(EmbedResponse) error

type RespondStyle struct {
	Style    Style  `json:"style"`
	Template string `json:"template"`
}

type Style string

type Tool struct {
	Type     string       `json:"type"`
	Function ToolFunction `json:"function"`
}

type ToolCall struct {
	Function ToolCallFunction `json:"function"`
}

type ToolCallFunction struct {
	Index     int            `json:"index,omitempty"`
	Name      string         `json:"name"`
	Arguments map[string]any `json:"arguments,omitempty"`
}

type ToolCallHandler func(ToolCall) (string, error)

type ToolFunction struct {
	Name        string                `json:"name"`
	Description string                `json:"description"`
	Parameters  ToolFunctionParameter `json:"parameters"`
}

type ToolFunctionParameter struct {
	Type       string                          `json:"type"`
	Required   []string                        `json:"required"`
	Properties map[string]ToolFunctionProperty `json:"properties,omitempty"`
}

type ToolFunctionProperty struct {
	Type        string   `json:"type"`
	Description string   `json:"description"`
	Enum        []string `json:"enum,omitempty"`
}