logging

type Config struct {
	Severity      Severity
	Outputs       []Output
	SampleRate    uint32
	DefaultFields map[string]interface{}
}

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

type ConsoleOutputOption func(*ConsoleOutput)

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

type FileOutputOption func(*FileOutput)

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

type FlightRecorderOption func(*FlightRecorder)

type JSONFormatter struct{}

type LogEntry struct {
	// Standard fields
	Time     int64
	Severity Severity
	Message  string
	File     string
	Line     int
	Function string
	TraceID  string // Added trace ID field

	// LLM-specific fields
	ModelID   string          // The LLM model being used
	TokenInfo *core.TokenInfo // Token usage information
	Latency   int64           // Operation duration in milliseconds
	Cost      float64         // Operation cost in dollars

	// General structured data
	Fields map[string]interface{}
}

type LogFormatter interface {
	Format(entry LogEntry) string
}

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

type Output interface {
	Write(LogEntry) error
	Sync() error
	Close() error
}

type RLMCallEntry struct {
	Prompt           string  `json:"prompt"`
	Response         string  `json:"response"`
	PromptTokens     int     `json:"prompt_tokens"`
	CompletionTokens int     `json:"completion_tokens"`
	ExecutionTime    float64 `json:"execution_time"`
}

type RLMCodeBlock struct {
	Code   string        `json:"code"`
	Result RLMCodeResult `json:"result"`
}

type RLMCodeResult struct {
	Stdout        string         `json:"stdout"`
	Stderr        string         `json:"stderr"`
	Locals        map[string]any `json:"locals"`
	ExecutionTime float64        `json:"execution_time"`
	RLMCalls      []RLMCallEntry `json:"rlm_calls"`
}

type RLMIterationEntry struct {
	Type          string         `json:"type"`
	Iteration     int            `json:"iteration"`
	Timestamp     string         `json:"timestamp"`
	Prompt        []RLMMessage   `json:"prompt"`
	Response      string         `json:"response"`
	CodeBlocks    []RLMCodeBlock `json:"code_blocks"`
	FinalAnswer   any            `json:"final_answer"`
	IterationTime float64        `json:"iteration_time"`
}

type RLMMessage struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type RLMMetadataEntry struct {
	Type              string         `json:"type"`
	Timestamp         string         `json:"timestamp"`
	RootModel         string         `json:"root_model"`
	MaxDepth          int            `json:"max_depth"`
	MaxIterations     int            `json:"max_iterations"`
	Backend           string         `json:"backend"`
	BackendKwargs     map[string]any `json:"backend_kwargs"`
	EnvironmentType   string         `json:"environment_type"`
	EnvironmentKwargs map[string]any `json:"environment_kwargs"`
	OtherBackends     any            `json:"other_backends"`
	Context           string         `json:"context,omitempty"`
	Query             string         `json:"query,omitempty"`
}

type RLMSessionConfig struct {
	RootModel     string
	MaxIterations int
	Backend       string
	BackendKwargs map[string]any
	Context       string
	Query         string
}

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

type Severity int32

type TextFormatter struct {
	// Whether to include timestamps in the output
	IncludeTimestamp bool
	// Whether to include file and line information
	IncludeLocation bool
	// Whether to include stack traces for errors
	IncludeStackTrace bool
}

type TraceEvent struct {
	Type      TraceEventType         `json:"type"`
	Timestamp time.Time              `json:"timestamp"`
	TraceID   string                 `json:"trace_id"`
	SpanID    string                 `json:"span_id,omitempty"`
	ParentID  string                 `json:"parent_id,omitempty"`
	Data      map[string]interface{} `json:"data,omitempty"`
}

type TraceEventType string

type TraceFormat int

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

type TraceOutputOption func(*TraceOutput)

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