createbatch

package
v0.25.0 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Jan 8, 2026 License: Apache-2.0 Imports: 14 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type BatchBuilder

type BatchBuilder interface {
	Parameters() parameters.Parameters
	BuildBatch(params parameters.ParamValues) (*dataprocpb.Batch, error)
}

type Config

type Config struct {
	Name              string                        `yaml:"name" validate:"required"`
	Kind              string                        `yaml:"kind" validate:"required"`
	Source            string                        `yaml:"source" validate:"required"`
	Description       string                        `yaml:"description"`
	RuntimeConfig     *dataprocpb.RuntimeConfig     `yaml:"runtimeConfig"`
	EnvironmentConfig *dataprocpb.EnvironmentConfig `yaml:"environmentConfig"`
	AuthRequired      []string                      `yaml:"authRequired"`
}

Config is a common config that can be used with any type of create batch tool. However, each tool will still need its own config type, embedding this Config, so it can provide a type-specific Initialize implementation.

func NewConfig

func NewConfig(ctx context.Context, name string, decoder *yaml.Decoder) (Config, error)

type Tool

type Tool struct {
	Config

	Builder BatchBuilder

	Parameters parameters.Parameters
	// contains filtered or unexported fields
}

func NewTool

func NewTool(cfg Config, originalCfg tools.ToolConfig, srcs map[string]sources.Source, builder BatchBuilder) (*Tool, error)

func (*Tool) Authorized

func (t *Tool) Authorized(services []string) bool

func (Tool) EmbedParams added in v0.25.0

func (t Tool) EmbedParams(ctx context.Context, paramValues parameters.ParamValues, embeddingModelsMap map[string]embeddingmodels.EmbeddingModel) (parameters.ParamValues, error)

func (*Tool) GetAuthTokenHeaderName

func (t *Tool) GetAuthTokenHeaderName(resourceMgr tools.SourceProvider) (string, error)

func (*Tool) Invoke

func (t *Tool) Invoke(ctx context.Context, resourceMgr tools.SourceProvider, params parameters.ParamValues, accessToken tools.AccessToken) (any, error)

func (*Tool) Manifest

func (t *Tool) Manifest() tools.Manifest

func (*Tool) McpManifest

func (t *Tool) McpManifest() tools.McpManifest

func (*Tool) ParseParams

func (t *Tool) ParseParams(data map[string]any, claims map[string]map[string]any) (parameters.ParamValues, error)

func (*Tool) RequiresClientAuthorization

func (t *Tool) RequiresClientAuthorization(resourceMgr tools.SourceProvider) (bool, error)

func (*Tool) ToConfig

func (t *Tool) ToConfig() tools.ToolConfig

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL