local

type Backend string

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

type CompleteParams struct {
	Messages     []MessageParam `json:"messages"`
	Model        string         `json:"model,omitempty"`
	SystemPrompt string         `json:"system_prompt,omitempty"`
	MaxTokens    int            `json:"max_tokens,omitempty"`
	Temperature  float64        `json:"temperature,omitempty"`
	Stream       bool           `json:"stream,omitempty"`
	Options      map[string]any `json:"options,omitempty"`
}

type CompleteResult struct {
	Content      string      `json:"content"`
	Model        string      `json:"model,omitempty"`
	FinishReason string      `json:"finish_reason,omitempty"`
	Usage        UsageResult `json:"usage"`
}

type Config struct {
	// Backend specifies which local model backend to use.
	// Required. One of: "ollama", "llama.cpp", "vllm", "transformers"
	Backend Backend `json:"backend" yaml:"backend"`

	// SidecarPath is the path to the Python sidecar script.
	// Required.
	SidecarPath string `json:"sidecar_path" yaml:"sidecar_path"`

	// Model is the local model name to use.
	// Format depends on backend (e.g., "llama3.2:latest" for Ollama).
	Model string `json:"model" yaml:"model"`

	// Host is the API server address for applicable backends.
	// Default: "localhost:11434" for Ollama, "localhost:8000" for vLLM.
	Host string `json:"host" yaml:"host"`

	// PythonPath is the path to the Python interpreter.
	// Default: "python3"
	PythonPath string `json:"python_path" yaml:"python_path"`

	// StartupTimeout is how long to wait for sidecar to become ready.
	// Default: 30 seconds.
	StartupTimeout time.Duration `json:"startup_timeout" yaml:"startup_timeout"`

	// RequestTimeout is the default timeout for completion requests.
	// Default: 5 minutes.
	RequestTimeout time.Duration `json:"request_timeout" yaml:"request_timeout"`

	// WorkDir is the working directory for the sidecar process.
	WorkDir string `json:"work_dir" yaml:"work_dir"`

	// Env provides additional environment variables for the sidecar.
	Env map[string]string `json:"env" yaml:"env"`

	// MCPServers configures MCP servers to pass through to the sidecar.
	// The sidecar is responsible for connecting to MCP servers.
	MCPServers map[string]MCPServerConfig `json:"mcp_servers" yaml:"mcp_servers"`
}

type InitParams struct {
	Backend    string                     `json:"backend"`
	Model      string                     `json:"model"`
	Host       string                     `json:"host,omitempty"`
	MCPServers map[string]MCPServerConfig `json:"mcp_servers,omitempty"`
	Options    map[string]any             `json:"options,omitempty"`
}

type InitResult struct {
	Ready   bool   `json:"ready"`
	Version string `json:"version,omitempty"`
	Message string `json:"message,omitempty"`
}

type MCPServerConfig struct {
	Type    string            `json:"type"`              // "stdio", "http", "sse"
	Command string            `json:"command,omitempty"` // For stdio transport
	Args    []string          `json:"args,omitempty"`
	Env     map[string]string `json:"env,omitempty"`
	URL     string            `json:"url,omitempty"` // For http/sse transport
	Headers []string          `json:"headers,omitempty"`
}

type MessageParam struct {
	Role    string `json:"role"`
	Content string `json:"content"`
	Name    string `json:"name,omitempty"` // For tool results
}

type Notification struct {
	JSONRPC string `json:"jsonrpc"`
	Method  string `json:"method"`
	Params  any    `json:"params,omitempty"`
}

type Option func(*Client)

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

type RPCError struct {
	Code    int             `json:"code"`
	Message string          `json:"message"`
	Data    json.RawMessage `json:"data,omitempty"`
}

type Request struct {
	JSONRPC string `json:"jsonrpc"`
	Method  string `json:"method"`
	Params  any    `json:"params,omitempty"`
	ID      int64  `json:"id"`
}

type Response struct {
	JSONRPC string          `json:"jsonrpc"`
	Result  json.RawMessage `json:"result,omitempty"`
	Error   *RPCError       `json:"error,omitempty"`
	ID      int64           `json:"id"`
}

type ShutdownResult struct {
	Success bool   `json:"success"`
	Message string `json:"message,omitempty"`
}

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

type StreamChunkParams struct {
	Content string `json:"content"`
	Done    bool   `json:"done"`
}

type StreamDoneParams struct {
	Usage        UsageResult `json:"usage"`
	FinishReason string      `json:"finish_reason,omitempty"`
	Model        string      `json:"model,omitempty"`
}

type UsageResult struct {
	InputTokens  int `json:"input_tokens"`
	OutputTokens int `json:"output_tokens"`
	TotalTokens  int `json:"total_tokens"`
}