react

type Agent interface {
	// Send a message to the agent, getting its response.
	// May pass parameters to stream back messages or the response.
	Send(msg string, opts ...SendMessageOpt) (string, error)
	// Fetch the entire history state of the agent.
	Messages() iter.Seq[Message]
}

type AgentMessage struct {
	Content string
}

type AgentMode uint8

type AgentModelBuilder interface {
	// Build a model for the agent.
	// A struct may be passed as the response type for a json schema, or it may be nil.
	// The stream callbacks may also be nil.
	BuildAgentModel(responseType any, onInitFinalStream func(), onDataFinalStream func(string)) jpf.Model
}

type AvailableToolDefinition struct {
	Name        string
	Description []string
}

type AvailableToolDefinitionsMessage struct {
	Tools []AvailableToolDefinition
}

type CraigOpt func(*craigKwargs)

type FragmentSelector interface {
	// Select any relevant fragments that should be added tp the conversation.
	SelectFragments([]PromptFragment, []Message) ([]PromptFragment, error)
}

type FragmentSelectorModelBuilder interface {
	// Build a model for a fragment selector.
	// A struct may be passed as the response type for a json schema, or it may be nil.
	BuildFragmentSelectorModel(responseType any) jpf.Model
}

type Message interface {
	// contains filtered or unexported methods
}

type MessageStreamer interface {
	// Try to send a message, ignoring errors.
	TrySendMessage(msg Message)
}

type ModeSwitchMessage struct {
	Mode AgentMode
}

type ModelBuilder interface {
	AgentModelBuilder
	FragmentSelectorModelBuilder
}

type NotificationMessage struct {
	Kind    string
	Content string
}

type PromptFragment struct {
	Key     string
	When    string
	Content string
}

type PromptFragmentMessage struct {
	Fragments []PromptFragment
}

type SendMessageOpt func(*streamers)

type SystemMessage struct {
	Content string
}

type TextStreamer interface {
	// Try to send a response text chunk back, ignoring errors.
	TrySendTextChunk(chunk string)
}

type Tool interface {
	// The name of the tool to be used by the agent. Should probably be snake_case.
	Name() string
	// Describe the tool in short bullet points to the agent.
	// Includes which parameters to provide.
	// Parameters can be described as any json type.
	Description() []string
	// Call the tool, providing a formatted response or an error if the tool call failed.
	// The values of the args will be the direct result of json decoding the tool call args.
	Call(map[string]any) (string, error)
}

type ToolCall struct {
	ToolName string
	ToolArgs []ToolCallArg
}

type ToolCallArg struct {
	ArgName  string
	ArgValue any
}

type ToolCallsMessage struct {
	Reasoning string
	ToolCalls []ToolCall
}

type ToolResponse struct {
	Response string
}

type ToolResponseMessage struct {
	Responses []ToolResponse
}

type UserMessage struct {
	Content string
}