extensions

type Config struct {
	// EnableLogging enables structured logging of tool calls.
	EnableLogging bool

	// EnableMetrics enables metrics collection.
	EnableMetrics bool

	// EnableMetadata enables metadata enrichment on tool results.
	EnableMetadata bool

	// EnableErrorHelp enables error hint enrichment.
	EnableErrorHelp bool

	// LogOutput is the writer for logging output. Defaults to os.Stderr.
	LogOutput io.Writer
}

type DataHubConfig struct {
	URL            string   `json:"url" yaml:"url"`
	Token          string   `json:"token" yaml:"token"`
	Timeout        Duration `json:"timeout" yaml:"timeout"`
	ConnectionName string   `json:"connection_name" yaml:"connection_name"`
	WriteEnabled   *bool    `json:"write_enabled" yaml:"write_enabled"`
}

type Duration struct {
	time.Duration
}

type ErrorHintMiddleware struct{}

type ExtFileConfig struct {
	Logging  *bool `json:"logging" yaml:"logging"`
	Metrics  *bool `json:"metrics" yaml:"metrics"`
	Metadata *bool `json:"metadata" yaml:"metadata"`
	Errors   *bool `json:"errors" yaml:"errors"`
}

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

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

type MetadataMiddleware struct{}

type MetricsCollector interface {
	// RecordCall records a tool call with its duration and success status.
	RecordCall(toolName string, duration time.Duration, success bool)
}

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

type ServerConfig struct {
	DataHub    DataHubConfig `json:"datahub" yaml:"datahub"`
	Toolkit    ToolkitConfig `json:"toolkit" yaml:"toolkit"`
	Extensions ExtFileConfig `json:"extensions" yaml:"extensions"`
}

type ToolMetrics struct {
	Calls      int64
	Errors     int64
	TotalNanos int64
}

type ToolkitConfig struct {
	DefaultLimit    int               `json:"default_limit" yaml:"default_limit"`
	MaxLimit        int               `json:"max_limit" yaml:"max_limit"`
	MaxLineageDepth int               `json:"max_lineage_depth" yaml:"max_lineage_depth"`
	Descriptions    map[string]string `json:"descriptions" yaml:"descriptions"`
}