servertool

type AdvisorHook ¶

type AdvisorHook struct{}

AdvisorHook injects AdvisorContext before an advisor tool call. Exported for test compatibility in packages that alias this type.

func (h AdvisorHook) Match(toolName string) bool

func (h AdvisorHook) PrepareContext(deps HookDeps, ctx context.Context, messages []map[string]any) context.Context

type DefaultExecutor ¶

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

DefaultExecutor is the standard implementation of Executor.

func NewDefaultExecutor(rt RuntimeCaller, deps HookDeps) *DefaultExecutor

func (e *DefaultExecutor) Execute(ctx context.Context, call ToolCall) (context.Context, coretool.ToolResult, error)

type Executor ¶

type Executor interface {
	Execute(ctx context.Context, call ToolCall) (context.Context, coretool.ToolResult, error)
}

Executor executes a single MCP tool call through the full pipeline: name validation, legacy remapping, callable guard, hooks, and dispatch.

type Hook ¶

type Hook interface {
	Match(toolName string) bool
	PrepareContext(deps HookDeps, ctx context.Context, messages []map[string]any) context.Context
}

Hook prepares runtime context for a specific MCP servertool call. Hooks run after the worker response arrives and before local tool execution.

type HookDeps ¶

type HookDeps interface {
	// GetAdvisorMaxUses returns the configured maximum advisor calls per request.
	GetAdvisorMaxUses() int
	// GetScenarioSink returns an enabled recording sink for the given scenario, or nil.
	GetScenarioSink(ctx context.Context) *obs.Sink
}

HookDeps provides the server-level dependencies that hooks may need.

type Pipeline ¶

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

Pipeline owns all registered ToolProviders and builds executors with the correct hook set. Construct once at server startup; call Register for each tool, then RegisterInto to populate the VirtualToolRegistry.

func NewPipeline() *Pipeline

func (p *Pipeline) NewExecutor(rt RuntimeCaller, deps HookDeps) *DefaultExecutor

func (p *Pipeline) Register(provider ToolProvider)

func (p *Pipeline) RegisterInto(registry VirtualToolRegistry)

type RuntimeCaller ¶

type RuntimeCaller interface {
	CallTool(ctx context.Context, normalizedName, arguments string) (coretool.ToolResult, error)
	ListCallableServerToolNames(ctx context.Context) map[string]struct{}
	GetAdvisorMaxUses() int
}

RuntimeCaller is the subset of the tool runtime used by DefaultExecutor.

type ToolCall ¶

type ToolCall struct {
	NormalizedName string
	Arguments      string
	Messages       []map[string]any
}

ToolCall describes a single MCP tool invocation.

type ToolProvider ¶

type ToolProvider interface {
	// Descriptor returns the VirtualTool to register in the runtime registry.
	Descriptor() coretool.VirtualTool
	// Hook returns a pre-call context hook, or nil if not needed.
	Hook() Hook
}

ToolProvider is implemented by each virtual server tool. Register once; Pipeline derives hooks and registry entries from it.

type VirtualToolRegistry ¶

type VirtualToolRegistry interface {
	Register(coretool.VirtualTool)
}