ai

type AddMessageOptions struct {
	Role        Role
	Type        MessageType
	Tool        string
	ContentType MessageContentType
	Content     string
}

type AnalystAgent struct {
	Runtime *runtime.Runtime
}

type AnalystAgentArgs struct {
	Prompt     string                  `json:"prompt"`
	Explore    string                  `` /* 139-byte string literal not displayed */
	Dimensions []string                `` /* 151-byte string literal not displayed */
	Measures   []string                `` /* 143-byte string literal not displayed */
	Where      *metricsview.Expression `json:"where" yaml:"where" jsonschema:"Optional filter for queries. If provided, this filter will be applied to all queries."`
	TimeStart  time.Time               `` /* 134-byte string literal not displayed */
	TimeEnd    time.Time               `` /* 128-byte string literal not displayed */
}

type AnalystAgentResult struct {
	Response string `json:"response"`
}

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

type CallOptions struct {
	Role    Role
	Name    string
	Unwrap  bool
	Out     any
	Args    any
	Handler func(context.Context) (any, error)
}

type CallResult struct {
	Call   *Message
	Result *Message
}

type CallToolOptions struct {
	Role   Role
	Tool   string
	Unwrap bool
	Out    any
	Args   any
}

type CompiledTool struct {
	Name                  string
	Spec                  *mcp.Tool
	CheckAccess           func(context.Context) (bool, error)
	UnmarshalArgs         func(content string) (any, error)
	UnmarshalResult       func(content string) (any, error)
	JSONHandler           func(ctx context.Context, input json.RawMessage) (json.RawMessage, error)
	RegisterWithMCPServer func(srv *mcp.Server)
}

type CompleteOptions struct {
	Messages      []*aiv1.CompletionMessage
	Tools         []string
	MaxIterations int
	// The complete loop will add intermediate messages for LLM thinking and tool calls to session under the current call.
	// In some cases, it's desirable to capture these intermediate messages in the parent call's context, in other cases it's better to isolate them and only expose the final result to the parent context.
	// When UnwrapCall is true, we run the completion loop within the current call, otherwise we wrap the complete loop in a new call to isolate internal messages.
	UnwrapCall bool
}

type CreateChart struct {
	Runtime *runtime.Runtime
}

type CreateChartArgs map[string]any

type CreateChartResult struct {
	ChartType string         `json:"chart_type"`
	Spec      map[string]any `json:"spec"`
	Message   string         `json:"message"`
}

type DevelopMetricsView struct {
	Runtime *runtime.Runtime
}

type DevelopMetricsViewArgs struct {
	Path   string `json:"path" jsonschema:"The path of a .yaml file in which to create or update a Rill metrics view definition."`
	Prompt string `json:"prompt" jsonschema:"Optional description of changes to make if editing an existing metrics view."`
	Model  string `json:"model" jsonschema:"Optional Rill model to derive from if creating a new metrics view."`
}

type DevelopMetricsViewResult struct {
	MetricsViewName string `json:"metrics_view_name" jsonschema:"The name of the developed Rill metrics view."`
}

type DevelopModel struct {
	Runtime *runtime.Runtime
}

type DevelopModelArgs struct {
	Path   string `json:"path" jsonschema:"The path of a .yaml file in which to create or update a Rill model definition."`
	Prompt string `` /* 145-byte string literal not displayed */
}

type DevelopModelResult struct {
	ModelName string `json:"model_name" jsonschema:"The name of the developed Rill model."`
}

type DeveloperAgent struct {
	Runtime *runtime.Runtime
}

type DeveloperAgentArgs struct {
	Prompt          string `json:"prompt" jsonschema:"User prompt for the developer agent."`
	InitProject     bool   `` /* 243-byte string literal not displayed */
	CurrentFilePath string `json:"current_file_path,omitempty" jsonschema:"Optional path to the file that the user is currently viewing/editing."`
}

type DeveloperAgentResult struct {
	Response string `json:"response"`
}

type GetMetricsView struct {
	Runtime *runtime.Runtime
}

type GetMetricsViewArgs struct {
	MetricsView string `json:"metrics_view" jsonschema:"Name of the metrics view"`
}

type GetMetricsViewResult struct {
	Spec map[string]any `json:"spec"`
}

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

type LLMMarshaler interface {
	ToLLM() *aiv1.ContentBlock
}

type ListFiles struct {
	Runtime *runtime.Runtime
}

type ListFilesArgs struct{}

type ListFilesResult struct {
	Files []map[string]any `json:"files"`
}

type ListMetricsViews struct {
	Runtime *runtime.Runtime
}

type ListMetricsViewsArgs struct{}

type ListMetricsViewsResult struct {
	MetricsViews []map[string]any `json:"metrics_views"`
}

type Message struct {
	// ID is unique for each message.
	ID string `json:"id" yaml:"id"`
	// ParentID is the ID of the parent message, usually the current tool call.
	ParentID string `json:"parent_id" yaml:"parent_id"`
	// SessionID is the ID of the session this message belongs to.
	SessionID string `json:"session_id" yaml:"session_id"`
	// Time the message was created.
	Time time.Time `json:"time" yaml:"time"`
	// Index of the message in the session. Used to order messages returned at the same time.
	Index int `json:"index" yaml:"index"`
	// Role is the actor that created the message.
	Role Role `json:"role" yaml:"role"`
	// Type is the type of the message.
	// For any given call, there will be only one "result" or "error" message.
	Type MessageType `json:"type" yaml:"type"`
	// Tool is the name of the tool that emitted the message, if any.
	Tool string `json:"tool"`
	// ContentType is the type of the Content string.
	ContentType MessageContentType `json:"content_type" yaml:"content_type"`
	// Content is the content of the message.
	Content string `json:"content" yaml:"content"`
	// contains filtered or unexported fields
}

type MessageContentType string

type MessageType string

type Predicate func(*Message) bool

type QueryMetricsView struct {
	Runtime *runtime.Runtime
}

type QueryMetricsViewArgs map[string]any

type QueryMetricsViewResult struct {
	Data    []map[string]any `json:"data"`
	OpenURL string           `json:"open_url,omitempty"`
}

type QueryMetricsViewSummary struct {
	Runtime *runtime.Runtime
}

type QueryMetricsViewSummaryArgs struct {
	MetricsView string `json:"metrics_view" jsonschema:"Name of the metrics view"`
}

type QueryMetricsViewSummaryResult struct {
	Data map[string]any `json:"data"`
}

type ReadFile struct {
	Runtime *runtime.Runtime
}

type ReadFileArgs struct {
	Path string `json:"path" jsonschema:"The path of the file to read"`
}

type ReadFileResult struct {
	Contents string
}

type Role string

type RouterAgent struct {
	Runtime *runtime.Runtime
}

type RouterAgentArgs struct {
	Prompt             string              `json:"prompt" jsonschema:"The user's prompt to be routed."`
	Agent              string              `` /* 131-byte string literal not displayed */
	AnalystAgentArgs   *AnalystAgentArgs   `` /* 136-byte string literal not displayed */
	DeveloperAgentArgs *DeveloperAgentArgs `` /* 142-byte string literal not displayed */
	SkipHandoff        bool                `` /* 158-byte string literal not displayed */
}

type RouterAgentResult struct {
	Response string `json:"response"`
	Agent    string `json:"agent"`
}

type Runner struct {
	Runtime  *runtime.Runtime
	Activity *activity.Client
	Tools    map[string]*CompiledTool
}

type SearchFiles struct {
	Runtime *runtime.Runtime
}

type SearchFilesArgs struct {
	Pattern       string `json:"pattern" jsonschema:"The pattern to search for. Supports regular expressions."`
	CaseSensitive bool   `json:"case_sensitive,omitempty" jsonschema:"Whether the search should be case-sensitive. Defaults to false."`
	GlobPattern   string `` /* 152-byte string literal not displayed */
}

type SearchFilesResult struct {
	Matches []SearchMatch `json:"matches"`
}

type SearchMatch struct {
	Path     string   `json:"path"`
	Lines    []int    `json:"lines"`
	Snippets []string `json:"snippets"`
}

type Session struct {
	*BaseSession
	ParentID string
}

type SessionOptions struct {
	InstanceID        string
	SessionID         string
	CreateIfNotExists bool
	Claims            *runtime.SecurityClaims
	UserAgent         string
}

type Tool[In, Out any] interface {
	Spec() *mcp.Tool
	CheckAccess(context.Context) (bool, error)
	Handler(ctx context.Context, args In) (Out, error)
}

type WriteFile struct {
	Runtime *runtime.Runtime
}

type WriteFileArgs struct {
	Path     string `json:"path" jsonschema:"The path of the file to write"`
	Contents string `` /* 134-byte string literal not displayed */
	Remove   bool   `json:"remove,omitempty" jsonschema:"Optional flag to remove the file instead of writing to it. Defaults to false."`
}

type WriteFileResult struct {
	Diff       string           `json:"diff,omitempty" jsonschema:"Diff of the file contents."`
	IsNewFile  bool             `json:"is_new_file,omitempty" jsonschema:"Indicates if the tool created a new file."`
	Resources  []map[string]any `json:"resources,omitempty" jsonschema:"The Rill resources declared in the file, if any."`
	ParseError string           `json:"parse_error,omitempty" jsonschema:"Parse error encountered when parsing the file, if any."`
}