llmworker

type ActivationArgs struct {
	workers.BaseActivationArgs
	// Messages are appended to the conversation when the worker activates.
	Messages []map[string]any
	// RunLLM says whether the model answers them. Nil answers, which is what a
	// caller sending messages almost always means.
	RunLLM *bool
}

type Config struct {
	// Name is what other workers address this one by.
	Name string
	// LLM is the language model service the worker runs. The tools are
	// registered on it.
	LLM ToolRegistrar
	// Pipeline is the pipeline to run. Nil runs the model on its own, which is
	// what a worker doing nothing but answering needs.
	Pipeline processor.Processor
	// Tools are the tools the model may call. Each is registered on the service
	// with the call options it carries, wrapped so that what its handler queues
	// is held until the call finishes.
	Tools []frames.Tool
	// Active reports whether the worker starts active. Nil leaves it inactive,
	// because a worker with a model behind it is almost always started by
	// something else deciding it is time.
	Active *bool
	// Bridged names the bridges this worker's pipeline exchanges frames over.
	// Non-nil wraps the pipeline in bus edges, and turns off the RTVI processor
	// a standalone worker gets, since a bridged worker has no client of its own.
	Bridged []string
	// DeferToolFrames says whether what a tool handler queues is held until
	// every call in flight has finished. Nil holds it, which is what keeps a
	// tool's own frames behind the result the model is waiting for.
	DeferToolFrames *bool
	// WorkerConfig is the rest of the pipeline worker's configuration. Name,
	// Active, Bridged and the RTVI setting are taken from the fields above.
	WorkerConfig pipeline.WorkerConfig
}

type ContextConfig struct {
	// Name is what other workers address this one by.
	Name string
	// LLM is the language model service the worker runs.
	LLM ToolRegistrar
	// Tools are the tools the model may call.
	Tools []frames.Tool
	// Active reports whether the worker starts active; nil leaves it inactive.
	Active *bool
	// Bridged names the bridges this worker's pipeline exchanges frames over.
	Bridged []string
	// DeferToolFrames says whether what a tool handler queues is held until the
	// calls in flight finish; nil holds it.
	DeferToolFrames *bool
	// Context is the conversation to run. Nil starts an empty one.
	Context *frames.LLMContext
	// AggregatorOptions configure the aggregator pair the worker builds.
	AggregatorOptions []aggregators.Option
	// WorkerConfig is the rest of the pipeline worker's configuration.
	WorkerConfig pipeline.WorkerConfig
}

type ContextWorker struct {
	*Worker
	// contains filtered or unexported fields
}

type ToolRegistrar interface {
	processor.Processor
	RegisterFunction(name string, h llm.FunctionCallHandler, opts ...llm.RegisterOption)
}

type Worker struct {
	*pipeline.Worker
	// contains filtered or unexported fields
}