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 ColumnInfo struct {
	Name string `json:"name"`
	Type string `json:"type"`
}

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 ListBucketObjects struct {
	Runtime *runtime.Runtime
}

type ListBucketObjectsArgs struct {
	Connector string `json:"connector" jsonschema:"The name of an object store connector (e.g., s3, gcs, azure)."`
	Bucket    string `json:"bucket" jsonschema:"The bucket name to list objects from."`
	Path      string `json:"path,omitempty" jsonschema:"Optional path prefix to list objects under. Defaults to root."`
	PageSize  int    `json:"page_size,omitempty" jsonschema:"Maximum number of objects to return. Defaults to 10."`
	PageToken string `json:"page_token,omitempty" jsonschema:"Token for pagination. Use next_page_token from previous response to get next page."`
}

type ListBucketObjectsResult struct {
	Objects       []ObjectInfo `json:"objects"`
	NextPageToken string       `json:"next_page_token,omitempty"`
}

type ListBuckets struct {
	Runtime *runtime.Runtime
}

type ListBucketsArgs struct {
	Connector string `json:"connector" jsonschema:"The name of an object store connector (e.g., s3, gcs, azure)."`
}

type ListBucketsResult struct {
	Buckets []string `json:"buckets"`
}

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 ListTables struct {
	Runtime *runtime.Runtime
}

type ListTablesArgs struct {
	Connector     string `json:"connector,omitempty" jsonschema:"Optional OLAP connector name. Defaults to the instance's default OLAP connector."`
	SearchPattern string `json:"search_pattern,omitempty" jsonschema:"Optional pattern to filter table names (uses SQL LIKE syntax)."`
	PageSize      int    `json:"page_size,omitempty" jsonschema:"Maximum number of tables to return. Defaults to 100."`
	PageToken     string `json:"page_token,omitempty" jsonschema:"Token for pagination. Use next_page_token from previous response to get next page."`
}

type ListTablesResult struct {
	Tables        []TableInfo `json:"tables"`
	NextPageToken string      `json:"next_page_token,omitempty"`
}

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 ObjectInfo struct {
	Path       string     `json:"path"`
	IsDir      bool       `json:"is_dir"`
	Size       int64      `json:"size,omitempty"`
	ModifiedOn *time.Time `json:"modified_on,omitempty"`
}

type Predicate func(*Message) bool

type ProjectStatus struct {
	Runtime *runtime.Runtime
}

type ProjectStatusArgs struct {
	WhereError bool   `json:"where_error,omitempty" jsonschema:"Optional flag to only return resources that have reconcile errors."`
	Kind       string `json:"kind,omitempty" jsonschema:"Optional filter to only return resources of the specified kind."`
	Name       string `json:"name,omitempty" jsonschema:"Optional filter to only return the resource with the specified name."`
	Path       string `json:"path,omitempty" jsonschema:"Optional filter to only return resources declared in the specified file path."`
}

type ProjectStatusResult struct {
	DefaultOLAPConnector string           `` /* 160-byte string literal not displayed */
	VariablesNames       []string         `` /* 136-byte string literal not displayed */
	Resources            []map[string]any `json:"resources" jsonschema:"List of resources and their status."`
	ParseErrors          []map[string]any `json:"parse_errors" jsonschema:"List of parse errors encountered when parsing project files."`
}

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 QuerySQL struct {
	Runtime *runtime.Runtime
}

type QuerySQLArgs struct {
	Connector      string `json:"connector,omitempty" jsonschema:"Optional OLAP connector name. Defaults to the instance's default OLAP connector."`
	SQL            string `json:"sql" jsonschema:"The SQL query to execute."`
	TimeoutSeconds int    `json:"timeout_seconds,omitempty" jsonschema:"Query timeout in seconds. Defaults to 30."`
}

type QuerySQLResult 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 ShowTable struct {
	Runtime *runtime.Runtime
}

type ShowTableArgs struct {
	Connector      string `json:"connector,omitempty" jsonschema:"Optional OLAP connector name. Defaults to the instance's default OLAP connector."`
	Table          string `json:"table" jsonschema:"The name of the table to describe."`
	Database       string `json:"database,omitempty" jsonschema:"Optional database name for connectors that support multiple databases."`
	DatabaseSchema string `json:"database_schema,omitempty" jsonschema:"Optional database schema name."`
}

type ShowTableResult struct {
	Name              string       `json:"name"`
	IsView            bool         `json:"is_view"`
	Columns           []ColumnInfo `json:"columns"`
	PhysicalSizeBytes int64        `` /* 142-byte string literal not displayed */
}

type TableInfo struct {
	Database                string `json:"database,omitempty"`
	DatabaseSchema          string `json:"database_schema,omitempty"`
	IsDefaultDatabaseSchema bool   `json:"is_default_database_schema,omitempty"`
	Name                    string `json:"name"`
	IsView                  bool   `json:"is_view"`
}

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."`
}