llm

type Capabilities struct {
	MaxTokens         int  `json:"max_tokens"`
	SupportsStreaming bool `json:"supports_streaming"`
	SupportsJSON      bool `json:"supports_json"`
	SupportsFunctions bool `json:"supports_functions"`
}

type Client interface {
	// Complete generates a completion for the given prompt
	Complete(ctx context.Context, prompt string, options *CompletionOptions) (string, error)

	// Stream generates a streaming completion
	Stream(ctx context.Context, prompt string, options *CompletionOptions, callback StreamCallback) error

	// GetCapabilities returns the client's capabilities
	GetCapabilities() Capabilities
}

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

type CompletionOptions struct {
	MaxTokens   int      `json:"max_tokens,omitempty"`
	Temperature float32  `json:"temperature,omitempty"`
	TopP        float32  `json:"top_p,omitempty"`
	Stop        []string `json:"stop,omitempty"`
}

type Config struct {
	Enabled     bool
	Provider    string
	Endpoint    string
	Model       string
	APIKey      string
	MaxTokens   int
	Temperature float32
	Timeout     time.Duration
}

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

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

type ProviderConstructor func(config Config) (detection.LLMValidator, error)

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

type StreamCallback func(chunk string) error

type ValidationConfig struct {
	// AllowedHosts is a list of allowed hostnames (default: localhost only)
	AllowedHosts []string

	// AllowedNetworks is a list of allowed IP networks in CIDR notation
	AllowedNetworks []string

	// AllowPrivateIPs allows RFC1918 private IP addresses
	AllowPrivateIPs bool

	// RequireHTTPS enforces HTTPS for non-localhost endpoints
	RequireHTTPS bool

	// MaxResponseTime is the maximum allowed response time for health checks
	MaxResponseTime time.Duration

	// EnableStrictMode enforces localhost-only connections
	EnableStrictMode bool
}

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