llm

type APIError struct {
	StatusCode int
	Body       string
}

type ChatRequest struct {
	Model           string           `json:"model"`
	Messages        []Message        `json:"messages"`
	Tools           []ToolDefinition `json:"tools,omitempty"`
	Temperature     float64          `json:"temperature,omitempty"`
	MaxTokens       int              `json:"max_tokens,omitempty"`
	ReasoningEffort string           `json:"reasoning_effort,omitempty"`
}

type ChatResponse struct {
	Content   string     `json:"content"`
	Reasoning string     `json:"reasoning,omitempty"`
	ToolCalls []ToolCall `json:"tool_calls,omitempty"`
	Usage     Usage      `json:"usage"`
}

type Config struct {
	Name         string         `json:"name"`
	ProviderType ProviderType   `json:"provider_type"`
	APIKey       string         `json:"api_key"`
	BaseURL      string         `json:"base_url"`
	Model        string         `json:"model"`
	ExtraConfig  map[string]any `json:"extra_config"`
}

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

type LMStudioProvider struct {
	*OpenAIProvider
}

type Message struct {
	Role       string     `json:"role"`
	Content    string     `json:"content,omitempty"`
	Reasoning  string     `json:"reasoning,omitempty"`
	ToolCalls  []ToolCall `json:"tool_calls,omitempty"`
	ToolCallID string     `json:"tool_call_id,omitempty"`
	Name       string     `json:"name,omitempty"`
}

type ModelInfo struct {
	ID            string `json:"id"`
	Name          string `json:"name,omitempty"`
	Description   string `json:"description,omitempty"`
	ContextLength int    `json:"context_length,omitempty"`
}

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

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

type OpenRouterProvider struct {
	*OpenAIProvider
}

type Provider interface {
	Chat(ctx context.Context, req ChatRequest) (*ChatResponse, error)
	Name() string
	Type() ProviderType
}

type ProviderType string

type StreamEvent struct {
	Type  StreamEventType `json:"type"`
	Delta string          `json:"delta,omitempty"`
	Usage Usage           `json:"usage,omitempty"`
}

type StreamEventType string

type StreamHandler func(StreamEvent) error

type StreamingProvider interface {
	ChatStream(ctx context.Context, req ChatRequest, handler StreamHandler) (*ChatResponse, error)
}

type ToolCall struct {
	ID       string       `json:"id"`
	Type     string       `json:"type"`
	Function ToolFunction `json:"function"`
}

type ToolChatEvent struct {
	Type       ToolChatEventType `json:"type"`
	Delta      string            `json:"delta,omitempty"`
	ToolCall   *ToolCall         `json:"tool_call,omitempty"`
	ToolResult *ToolResult       `json:"tool_result,omitempty"`
	Usage      *Usage            `json:"usage,omitempty"`
}

type ToolChatEventHandler func(ToolChatEvent) error

type ToolChatEventType string

type ToolChatOptions struct {
	ReasoningEffort string
	MaxRounds       int
	ContextWindow   int
}

type ToolClusterStore interface {
	List(ctx context.Context) ([]models.Cluster, error)
	GetByID(ctx context.Context, id string) (*models.Cluster, error)
}

type ToolDefinition struct {
	Type     string   `json:"type"`
	Function ToolSpec `json:"function"`
}

type ToolExecutionResult struct {
	Result  any                `json:"result,omitempty"`
	Display *ToolResultDisplay `json:"display,omitempty"`
}

type ToolExecutor interface {
	GetAllTools() []ToolDefinition
	Execute(ctx context.Context, name string, arguments json.RawMessage) (any, error)
}

type ToolFunction struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"`
}

type ToolHandler func(ctx context.Context, args json.RawMessage) (any, error)

type ToolKubernetesClusterStore interface {
	List(ctx context.Context) ([]models.KubernetesCluster, error)
	GetByID(ctx context.Context, id string) (*models.KubernetesCluster, error)
}

type ToolPipelineReloader interface {
	Reload(ctx context.Context) error
}

type ToolPipelineRunResult struct {
	ExecutionID  string `json:"execution_id"`
	PipelineID   string `json:"pipeline_id"`
	PipelineName string `json:"pipeline_name"`
	Status       string `json:"status"`
	NodesRun     int    `json:"nodes_run"`
	Returned     bool   `json:"returned"`
	ReturnValue  any    `json:"return_value,omitempty"`
}

type ToolPipelineRunner interface {
	Run(ctx context.Context, pipelineID string, input map[string]any) (*ToolPipelineRunResult, error)
}

type ToolPipelineStore interface {
	List(ctx context.Context) ([]models.Pipeline, error)
	GetByID(ctx context.Context, id string) (*models.Pipeline, error)
	Create(ctx context.Context, pipeline *models.Pipeline) error
	Update(ctx context.Context, pipeline *models.Pipeline) error
	Delete(ctx context.Context, id string) error
}

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

type ToolRegistryOptions struct {
	ProxmoxStore               ToolClusterStore
	KubernetesStore            ToolKubernetesClusterStore
	PipelineStore              ToolPipelineStore
	PipelineReloader           ToolPipelineReloader
	PipelineRunner             ToolPipelineRunner
	DefaultProxmoxClusterID    string
	DefaultKubernetesClusterID string
	EnableProxmox              bool
	EnableKubernetes           bool
	SkillStore                 skills.Reader
	ShellRunner                shellcmd.Runner
	WorkspaceRoot              string
	WebToolsConfig             *webtools.RuntimeConfig
}

type ToolResult struct {
	Tool      string             `json:"tool"`
	Arguments any                `json:"arguments,omitempty"`
	Result    any                `json:"result,omitempty"`
	Error     string             `json:"error,omitempty"`
	Display   *ToolResultDisplay `json:"display,omitempty"`
}

type ToolResultDisplay struct {
	Kind      ToolResultDisplayKind `json:"kind"`
	Title     string                `json:"title,omitempty"`
	Path      string                `json:"path,omitempty"`
	Summary   string                `json:"summary,omitempty"`
	Preview   string                `json:"preview,omitempty"`
	Diff      string                `json:"diff,omitempty"`
	Truncated bool                  `json:"truncated,omitempty"`
	Stats     map[string]any        `json:"stats,omitempty"`
}

type ToolResultDisplayKind string

type ToolSpec struct {
	Name        string                 `json:"name"`
	Description string                 `json:"description"`
	Parameters  map[string]interface{} `json:"parameters"`
}

type Usage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}