tools

type Callback interface {
	OnToolStart(context.Context, ITool, string)
	OnToolEnd(context.Context, ITool, string, string)
	OnToolError(context.Context, ITool, string, error)
	OnToolLLMCall(context.Context, ITool, []llms.MessageContent)
}

type IMCPTool interface {
	ITool
	RegisterMCP(registrator McpServerRegistrator) error
}

type ITool interface {
	// Name returns the name of the Tool.
	Name() string
	// Description returns the description of the tool, to be used in the prompt.
	// Should not exceed LLM model limit.
	Description() string
	// Parameters returns the parameters definition of the function, to be used in the prompt.
	Parameters() any

	// Call executes the tool with the given input and returns the result.
	Call(context.Context, string) (string, error)
}

type MCPTool[I any] interface {
	IMCPTool
	RunMCP(context.Context, *I) (*mcp.ToolResponse, error)
}

type McpServerRegistrator interface {
	RegisterTool(name string, description string, handler any) error
}

type Tool[I any, O any] interface {
	ITool
	Run(context.Context, *I) (*O, error)
}