codeinterpreter

type AuthenticationError struct {
	Message string
}

type BarChart struct {
	Chart2D
	Bars []BarData `json:"-"`
}

type BarData struct {
	Label string `json:"label"`
	Value string `json:"value"`
	Group string `json:"group"`
}

type BaseChart struct {
	Type     ChartType `json:"type"`
	Title    string    `json:"title"`
	Elements []any     `json:"elements"`
	// contains filtered or unexported fields
}

type BoxAndWhiskerChart struct {
	Chart2D
	Boxes []BoxAndWhiskerData `json:"-"`
}

type BoxAndWhiskerData struct {
	Label         string    `json:"label"`
	Min           float64   `json:"min"`
	FirstQuartile float64   `json:"first_quartile"`
	Median        float64   `json:"median"`
	ThirdQuartile float64   `json:"third_quartile"`
	Max           float64   `json:"max"`
	Outliers      []float64 `json:"outliers"`
}

type Chart interface {
	ChartType() ChartType
	ChartTitle() string
	// ToDict returns the raw JSON representation of the chart.
	ToJSON() map[string]any
}

type Chart2D struct {
	BaseChart
	XLabel string `json:"x_label,omitempty"`
	YLabel string `json:"y_label,omitempty"`
	XUnit  string `json:"x_unit,omitempty"`
	YUnit  string `json:"y_unit,omitempty"`
}

type ChartType string

type ConnectionConfig struct {
	// APIKey is the E2B API key. If empty the E2B_API_KEY environment variable
	// is used.
	APIKey string
	// AccessToken is an optional envd access token used to authenticate direct
	// requests to the sandbox.
	AccessToken string
	// TrafficAccessToken is an optional token used to bypass traffic controls.
	TrafficAccessToken string
	// Domain is the e2b domain (default: e2b.app). The E2B_DOMAIN env var is
	// used when empty.
	Domain string
	// Debug turns on debug-mode. When true, the SDK talks over http:// and
	// uses the sandbox host unchanged (useful for local development).
	Debug bool
	// RequestTimeout is the default HTTP request timeout.
	RequestTimeout time.Duration
	// HTTPClient is the underlying client used for all HTTP traffic. If nil a
	// sensible default is created.
	HTTPClient *http.Client
	// Headers lets callers inject additional headers on every request.
	Headers map[string]string
}

type Context struct {
	// ID of the context.
	ID string `json:"id"`
	// Language of the context.
	Language string `json:"language"`
	// Cwd is the working directory inside the sandbox.
	Cwd string `json:"cwd"`
}

type CreateCodeContextOpts struct {
	// Cwd is the working directory for the context (default /home/user).
	Cwd string
	// Language of the new context (default python).
	Language RunCodeLanguage
	// RequestTimeout overrides the default HTTP request timeout.
	RequestTimeout time.Duration
}

type Execution struct {
	// Results collects the cell's main result and any intermediate display
	// calls (e.g. matplotlib plots).
	Results []*Result `json:"results"`
	// Logs holds stdout/stderr lines printed during execution.
	Logs Logs `json:"logs"`
	// Error is set if the execution failed; nil otherwise.
	Error *ExecutionError `json:"error,omitempty"`
	// ExecutionCount is the execution count (cell index) reported by the kernel.
	ExecutionCount int `json:"execution_count,omitempty"`
}

type ExecutionError struct {
	// Name of the error (e.g. "NameError").
	Name string `json:"name"`
	// Value/message of the error.
	Value string `json:"value"`
	// Traceback is the raw traceback text.
	Traceback string `json:"traceback"`
}

type InvalidArgumentError struct {
	Message string
}

type LineChart struct {
	PointChart
}

type Logs struct {
	Stdout []string `json:"stdout"`
	Stderr []string `json:"stderr"`
}

type MIMEType = string

type NotFoundError struct {
	Message string
}

type OnErrorFunc func(err *ExecutionError)

type OnResultFunc func(result *Result)

type OnStderrFunc func(msg OutputMessage)

type OnStdoutFunc func(msg OutputMessage)

type OutputMessage struct {
	// Line is the raw output line.
	Line string
	// Timestamp is the unix epoch in nanoseconds.
	Timestamp int64
	// Error indicates whether this output originates from stderr.
	Error bool
}

type PieChart struct {
	BaseChart
	Slices []PieData `json:"-"`
}

type PieData struct {
	Label  string  `json:"label"`
	Angle  float64 `json:"angle"`
	Radius float64 `json:"radius"`
}

type PointChart struct {
	Chart2D
	XTicks      []any       `json:"x_ticks"`
	XTickLabels []string    `json:"x_tick_labels"`
	XScale      ScaleType   `json:"x_scale"`
	YTicks      []any       `json:"y_ticks"`
	YTickLabels []string    `json:"y_tick_labels"`
	YScale      ScaleType   `json:"y_scale"`
	Points      []PointData `json:"-"`
}

type PointData struct {
	Label  string   `json:"label"`
	Points [][2]any `json:"points"`
}

type RateLimitError struct {
	Message string
}

type Result struct {
	Text       string         `json:"text,omitempty"`
	HTML       string         `json:"html,omitempty"`
	Markdown   string         `json:"markdown,omitempty"`
	SVG        string         `json:"svg,omitempty"`
	PNG        string         `json:"png,omitempty"`
	JPEG       string         `json:"jpeg,omitempty"`
	PDF        string         `json:"pdf,omitempty"`
	LaTeX      string         `json:"latex,omitempty"`
	JSON       map[string]any `json:"json,omitempty"`
	JavaScript string         `json:"javascript,omitempty"`
	Data       map[string]any `json:"data,omitempty"`
	// Chart is the structured chart data extracted by the server, if any.
	Chart Chart `json:"-"`
	// IsMainResult indicates whether this is the primary result of the cell
	// (as opposed to an intermediate display call).
	IsMainResult bool `json:"is_main_result,omitempty"`
	// Extra holds any additional representations not covered by the
	// standard fields above.
	Extra map[string]any `json:"extra,omitempty"`
	// Raw holds the full raw JSON payload as returned by the server.
	Raw map[string]any `json:"-"`
}

type RunCodeLanguage string

type RunCodeOpts struct {
	// Language to use. Must not be combined with Context.
	Language RunCodeLanguage
	// Context (pre-created kernel) to run the code in.
	Context *Context
	// OnStdout is called for every stdout chunk.
	OnStdout OnStdoutFunc
	// OnStderr is called for every stderr chunk.
	OnStderr OnStderrFunc
	// OnResult is called for every Result (display call or final result).
	OnResult OnResultFunc
	// OnError is called when the kernel reports an error for this cell.
	OnError OnErrorFunc
	// Envs are extra environment variables exposed to the running code.
	Envs map[string]string
	// Timeout is the maximum execution time for this cell (default: 300s).
	// Pass -1 to disable.
	Timeout time.Duration
	// RequestTimeout is the HTTP-level request timeout.
	RequestTimeout time.Duration
}

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

type SandboxError struct {
	Message    string
	StatusCode int
}

type SandboxInfo struct {
	SandboxID  string            `json:"sandboxID"`
	ClientID   string            `json:"clientID"`
	TemplateID string            `json:"templateID"`
	Alias      string            `json:"alias,omitempty"`
	Metadata   map[string]string `json:"metadata,omitempty"`
	StartedAt  string            `json:"startedAt,omitempty"`
	EndAt      string            `json:"endAt,omitempty"`
	State      string            `json:"state,omitempty"`
}

type SandboxOpts struct {
	// APIKey to use. Falls back to the E2B_API_KEY env variable.
	APIKey string
	// AccessToken to use.
	AccessToken string
	// Domain to use (defaults to e2b.app).
	Domain string
	// Debug, if true, uses plain http:// against the sandbox.
	Debug bool
	// RequestTimeout default HTTP request timeout.
	RequestTimeout time.Duration
	// Timeout is the sandbox lifetime in seconds (not the request timeout).
	Timeout time.Duration
	// Template id/alias to use when creating a sandbox. Defaults to the
	// code-interpreter template.
	Template string
	// Metadata attached to the sandbox.
	Metadata map[string]string
	// EnvVars passed to the sandbox at startup.
	EnvVars map[string]string
	// HTTPClient allows overriding the underlying http.Client.
	HTTPClient *http.Client
	// Headers are additional headers to send on every request.
	Headers map[string]string
}

type ScaleType string

type ScatterChart struct {
	PointChart
}

type SuperChart struct {
	BaseChart
	Charts []Chart `json:"-"`
}

type TimeoutError struct {
	Message string
}

type UnknownChart struct {
	BaseChart
}