supervisoragents

type App struct {
	// App name
	Name string `json:"name"`
}

type Connection struct {
	Name string `json:"name"`
}

type CreateSupervisorAgentRequest struct {
	// The Supervisor Agent to create.
	SupervisorAgent SupervisorAgent `json:"supervisor_agent"`
}

type CreateToolRequest struct {
	// Parent resource where this tool will be created. Format:
	// supervisor-agents/{supervisor_agent_id}
	Parent string `json:"-" url:"-"`

	Tool Tool `json:"tool"`
	// The ID to use for the tool, which will become the final component of the
	// tool's resource name.
	ToolId string `json:"-" url:"tool_id"`
}

type DeleteSupervisorAgentRequest struct {
	// The resource name of the Supervisor Agent. Format:
	// supervisor-agents/{supervisor_agent_id}
	Name string `json:"-" url:"-"`
}

type DeleteToolRequest struct {
	// The resource name of the Tool. Format:
	// supervisor-agents/{supervisor_agent_id}/tools/{tool_id}
	Name string `json:"-" url:"-"`
}

type GenieSpace struct {
	// The ID of the genie space.
	Id string `json:"id"`
}

type GetSupervisorAgentRequest struct {
	// The resource name of the Supervisor Agent. Format:
	// supervisor-agents/{supervisor_agent_id}
	Name string `json:"-" url:"-"`
}

type GetToolRequest struct {
	// The resource name of the Tool. Format:
	// supervisor-agents/{supervisor_agent_id}/tools/{tool_id}
	Name string `json:"-" url:"-"`
}

type KnowledgeAssistant struct {
	// The ID of the knowledge assistant.
	KnowledgeAssistantId string `json:"knowledge_assistant_id"`
	// Deprecated: use knowledge_assistant_id instead.
	ServingEndpointName string `json:"serving_endpoint_name,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type ListSupervisorAgentsRequest struct {
	// The maximum number of supervisor agents to return. If unspecified, at
	// most 100 supervisor agents will be returned. The maximum value is 100;
	// values above 100 will be coerced to 100.
	PageSize int `json:"-" url:"page_size,omitempty"`
	// A page token, received from a previous `ListSupervisorAgents` call.
	// Provide this to retrieve the subsequent page. If unspecified, the first
	// page will be returned.
	PageToken string `json:"-" url:"page_token,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type ListSupervisorAgentsResponse struct {
	// A token that can be sent as `page_token` to retrieve the next page. If
	// this field is omitted, there are no subsequent pages.
	NextPageToken string `json:"next_page_token,omitempty"`

	SupervisorAgents []SupervisorAgent `json:"supervisor_agents,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type ListToolsRequest struct {
	PageSize int `json:"-" url:"page_size,omitempty"`

	PageToken string `json:"-" url:"page_token,omitempty"`
	// Parent resource to list from. Format:
	// supervisor-agents/{supervisor_agent_id}
	Parent string `json:"-" url:"-"`

	ForceSendFields []string `json:"-" url:"-"`
}

type ListToolsResponse struct {
	NextPageToken string `json:"next_page_token,omitempty"`

	Tools []Tool `json:"tools,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type SupervisorAgent struct {
	// Creation timestamp.
	CreateTime *time.Time `json:"create_time,omitempty"`
	// The creator of the Supervisor Agent.
	Creator string `json:"creator,omitempty"`
	// Description of what this agent can do (user-facing).
	Description string `json:"description"`
	// The display name of the Supervisor Agent, unique at workspace level.
	DisplayName string `json:"display_name"`
	// The name of the supervisor agent's serving endpoint.
	EndpointName string `json:"endpoint_name,omitempty"`
	// The MLflow experiment ID.
	ExperimentId string `json:"experiment_id,omitempty"`
	// Deprecated: Use supervisor_agent_id instead.
	Id string `json:"id,omitempty"`
	// Optional natural-language instructions for the supervisor agent.
	Instructions string `json:"instructions,omitempty"`
	// The resource name of the SupervisorAgent. Format:
	// supervisor-agents/{supervisor_agent_id}
	Name string `json:"name,omitempty"`
	// The universally unique identifier (UUID) of the Supervisor Agent.
	SupervisorAgentId string `json:"supervisor_agent_id,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

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

type SupervisorAgentsInterface interface {

	// Creates a new Supervisor Agent.
	CreateSupervisorAgent(ctx context.Context, request CreateSupervisorAgentRequest) (*SupervisorAgent, error)

	// Creates a Tool under a Supervisor Agent. Specify one of "genie_space",
	// "knowledge_assistant", "uc_function", "connection", "app", "volume",
	// "lakeview_dashboard" in the request body.
	CreateTool(ctx context.Context, request CreateToolRequest) (*Tool, error)

	// Deletes a Supervisor Agent.
	DeleteSupervisorAgent(ctx context.Context, request DeleteSupervisorAgentRequest) error

	// Deletes a Tool.
	DeleteTool(ctx context.Context, request DeleteToolRequest) error

	// Gets a Supervisor Agent.
	GetSupervisorAgent(ctx context.Context, request GetSupervisorAgentRequest) (*SupervisorAgent, error)

	// Gets a Tool.
	GetTool(ctx context.Context, request GetToolRequest) (*Tool, error)

	// Lists Supervisor Agents.
	//
	// This method is generated by Databricks SDK Code Generator.
	ListSupervisorAgents(ctx context.Context, request ListSupervisorAgentsRequest) listing.Iterator[SupervisorAgent]

	// Lists Supervisor Agents.
	//
	// This method is generated by Databricks SDK Code Generator.
	ListSupervisorAgentsAll(ctx context.Context, request ListSupervisorAgentsRequest) ([]SupervisorAgent, error)

	// Lists Tools under a Supervisor Agent.
	//
	// This method is generated by Databricks SDK Code Generator.
	ListTools(ctx context.Context, request ListToolsRequest) listing.Iterator[Tool]

	// Lists Tools under a Supervisor Agent.
	//
	// This method is generated by Databricks SDK Code Generator.
	ListToolsAll(ctx context.Context, request ListToolsRequest) ([]Tool, error)

	// Updates a Supervisor Agent. The fields that are required depend on the paths
	// specified in `update_mask`. Only fields included in the mask will be updated.
	UpdateSupervisorAgent(ctx context.Context, request UpdateSupervisorAgentRequest) (*SupervisorAgent, error)

	// Updates a Tool. Only the `description` field can be updated. To change
	// immutable fields such as tool type, spec, or tool ID, delete the tool and
	// recreate it.
	UpdateTool(ctx context.Context, request UpdateToolRequest) (*Tool, error)
}

type Tool struct {
	App *App `json:"app,omitempty"`

	Connection *Connection `json:"connection,omitempty"`
	// Description of what this tool does (user-facing).
	Description string `json:"description"`

	GenieSpace *GenieSpace `json:"genie_space,omitempty"`
	// Deprecated: Use tool_id instead.
	Id string `json:"id,omitempty"`

	KnowledgeAssistant *KnowledgeAssistant `json:"knowledge_assistant,omitempty"`
	// Full resource name:
	// supervisor-agents/{supervisor_agent_id}/tools/{tool_id}
	Name string `json:"name,omitempty"`
	// User specified id of the Tool.
	ToolId string `json:"tool_id,omitempty"`
	// Tool type. Must be one of: "genie_space", "knowledge_assistant",
	// "uc_function", "connection", "app", "volume", "lakeview_dashboard",
	// "serving_endpoint".
	ToolType string `json:"tool_type"`

	UcFunction *UcFunction `json:"uc_function,omitempty"`

	Volume *Volume `json:"volume,omitempty"`

	ForceSendFields []string `json:"-" url:"-"`
}

type UcFunction struct {
	// Full uc function name
	Name string `json:"name"`
}

type UpdateSupervisorAgentRequest struct {
	// The resource name of the SupervisorAgent. Format:
	// supervisor-agents/{supervisor_agent_id}
	Name string `json:"-" url:"-"`
	// The SupervisorAgent to update.
	SupervisorAgent SupervisorAgent `json:"supervisor_agent"`
	// Field mask for fields to be updated.
	UpdateMask fieldmask.FieldMask `json:"-" url:"update_mask"`
}

type UpdateToolRequest struct {
	// Full resource name:
	// supervisor-agents/{supervisor_agent_id}/tools/{tool_id}
	Name string `json:"-" url:"-"`
	// The Tool to update.
	Tool Tool `json:"tool"`
	// Field mask for fields to be updated.
	UpdateMask fieldmask.FieldMask `json:"-" url:"update_mask"`
}

type Volume struct {
	// Full uc volume name
	Name string `json:"name"`
}