customllms

type CancelCustomLlmOptimizationRunRequest struct {
	Id *string
}

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

type CreateCustomLlmRequest struct {
	// Name of the custom LLM. Only alphanumeric characters and dashes allowed.
	Name *string
	// Instructions for the custom LLM to follow
	Instructions *string
	// Datasets used for training and evaluating the model, not for inference.
	// Currently, only 1 dataset is accepted.
	Datasets []Dataset
	// Guidelines for the custom LLM to adhere to
	Guidelines []string
	// This will soon be deprecated!! Optional: UC path for agent artifacts. If you
	// are using a dataset that you only have read permissions, please provide a
	// destination path where you have write permissions. Please provide this in
	// catalog.schema format.
	AgentArtifactPath *string
}

type CustomLlm struct {
	Id *string `fieldmask:"id"`
	// Name of the custom LLM
	Name *string `fieldmask:"name"`
	// Name of the endpoint that will be used to serve the custom LLM
	EndpointName *string `fieldmask:"endpoint_name"`
	// Instructions for the custom LLM to follow
	Instructions *string `fieldmask:"instructions"`
	// Datasets used for training and evaluating the model, not for inference
	Datasets []Dataset `fieldmask:"datasets"`
	// Guidelines for the custom LLM to adhere to
	Guidelines []string `fieldmask:"guidelines"`
	// If optimization is kicked off, tracks the state of the custom LLM
	OptimizationState State `fieldmask:"optimization_state"`
	// Creator of the custom LLM
	Creator *string `fieldmask:"creator"`
	// Creation timestamp of the custom LLM
	CreationTime      *types.Time `fieldmask:"creation_time"`
	AgentArtifactPath *string     `fieldmask:"agent_artifact_path"`
}

type Dataset struct {
	Table *Table
}

type DeleteCustomLlmRequest struct {
	// The id of the custom llm
	Id *string
}

type GetCustomLlmRequest struct {
	// The id of the custom llm
	Id *string
}

type StartCustomLlmOptimizationRunRequest struct {
	// The Id of the tile.
	Id *string
}

type State string

type Table struct {
	// Full UC table path in catalog.schema.table_name format
	TablePath *string
	// Name of the request column
	RequestCol *string
	// Optional: Name of the response column if the data is labeled
	ResponseCol *string
}

type UpdateCustomLlmRequest struct {
	// The id of the custom llm
	Id *string
	// The CustomLlm containing the fields which should be updated.
	CustomLlm *CustomLlm
	// The list of the CustomLlm fields to update. These should correspond to the
	// values (or lack thereof) present in `custom_llm`.
	UpdateMask *types.FieldMask[CustomLlm]
}