plugininternal

Documentation

Index

• func ToContext(ctx context.Context, cfg *PluginManager) context.Context
• type PluginConfig
• type PluginManager
  • func NewPluginManager(cfg PluginConfig) (*PluginManager, error)
  • func (pm *PluginManager) Close() error
  • func (pm *PluginManager) RunAfterAgentCallback(cctx agent.CallbackContext) (*genai.Content, error)
  • func (pm *PluginManager) RunAfterModelCallback(cctx agent.CallbackContext, llmResponse *model.LLMResponse, ...) (*model.LLMResponse, error)
  • func (pm *PluginManager) RunAfterRunCallback(cctx agent.InvocationContext)
  • func (pm *PluginManager) RunAfterToolCallback(ctx tool.Context, tool tool.Tool, args, result map[string]any, err error) (map[string]any, error)
  • func (pm *PluginManager) RunBeforeAgentCallback(cctx agent.CallbackContext) (*genai.Content, error)
  • func (pm *PluginManager) RunBeforeModelCallback(cctx agent.CallbackContext, llmRequest *model.LLMRequest) (*model.LLMResponse, error)
  • func (pm *PluginManager) RunBeforeRunCallback(cctx agent.InvocationContext) (*genai.Content, error)
  • func (pm *PluginManager) RunBeforeToolCallback(ctx tool.Context, tool tool.Tool, args map[string]any) (map[string]any, error)
  • func (pm *PluginManager) RunOnEventCallback(cctx agent.InvocationContext, event *session.Event) (*session.Event, error)
  • func (pm *PluginManager) RunOnModelErrorCallback(cctx agent.CallbackContext, llmRequest *model.LLMRequest, ...) (*model.LLMResponse, error)
  • func (pm *PluginManager) RunOnToolErrorCallback(ctx tool.Context, tool tool.Tool, args map[string]any, err error) (map[string]any, error)
  • func (pm *PluginManager) RunOnUserMessageCallback(cctx agent.InvocationContext, userMessage *genai.Content) (*genai.Content, error)

Functions

func ToContext

func ToContext(ctx context.Context, cfg *PluginManager) context.Context

Types

type PluginConfig

type PluginConfig struct {
	Plugins      []*plugin.Plugin
	CloseTimeout time.Duration
}

type PluginManager

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

PluginManager manages the registration and execution of plugins.

func NewPluginManager

func NewPluginManager(cfg PluginConfig) (*PluginManager, error)

NewPluginManager creates a new PluginManager.

func (*PluginManager) Close

func (pm *PluginManager) Close() error

Close calls the CloseFunc on all registered plugins.

func (*PluginManager) RunAfterAgentCallback

func (pm *PluginManager) RunAfterAgentCallback(cctx agent.CallbackContext) (*genai.Content, error)

RunAfterAgentCallback runs the AfterAgentCallback for all plugins.

func (*PluginManager) RunAfterModelCallback

func (pm *PluginManager) RunAfterModelCallback(cctx agent.CallbackContext, llmResponse *model.LLMResponse, llmResponseError error) (*model.LLMResponse, error)

RunAfterModelCallback runs the AfterModelCallback for all plugins.

func (*PluginManager) RunAfterRunCallback

func (pm *PluginManager) RunAfterRunCallback(cctx agent.InvocationContext)

RunAfterRunCallback runs the AfterRunCallback for all plugins.

func (*PluginManager) RunAfterToolCallback

func (pm *PluginManager) RunAfterToolCallback(ctx tool.Context, tool tool.Tool, args, result map[string]any, err error) (map[string]any, error)

RunAfterToolCallback runs the AfterToolCallback for all plugins.

func (*PluginManager) RunBeforeAgentCallback

func (pm *PluginManager) RunBeforeAgentCallback(cctx agent.CallbackContext) (*genai.Content, error)

RunBeforeAgentCallback runs the BeforeAgentCallback for all plugins.

func (*PluginManager) RunBeforeModelCallback

func (pm *PluginManager) RunBeforeModelCallback(cctx agent.CallbackContext, llmRequest *model.LLMRequest) (*model.LLMResponse, error)

RunBeforeModelCallback runs the BeforeModelCallback for all plugins.

func (*PluginManager) RunBeforeRunCallback

func (pm *PluginManager) RunBeforeRunCallback(cctx agent.InvocationContext) (*genai.Content, error)

RunBeforeRunCallback runs the BeforeRunCallback for all plugins.

func (*PluginManager) RunBeforeToolCallback

func (pm *PluginManager) RunBeforeToolCallback(ctx tool.Context, tool tool.Tool, args map[string]any) (map[string]any, error)

RunBeforeToolCallback runs the BeforeToolCallback for all plugins.

func (*PluginManager) RunOnEventCallback

func (pm *PluginManager) RunOnEventCallback(cctx agent.InvocationContext, event *session.Event) (*session.Event, error)

RunOnEventCallback runs the OnEventCallback for all plugins.

func (*PluginManager) RunOnModelErrorCallback

func (pm *PluginManager) RunOnModelErrorCallback(cctx agent.CallbackContext, llmRequest *model.LLMRequest, llmResponseError error) (*model.LLMResponse, error)

RunOnModelErrorCallback runs the OnModelErrorCallback for all plugins.

func (*PluginManager) RunOnToolErrorCallback

func (pm *PluginManager) RunOnToolErrorCallback(ctx tool.Context, tool tool.Tool, args map[string]any, err error) (map[string]any, error)

RunOnToolErrorCallback runs the OnToolErrorCallback for all plugins.

func (*PluginManager) RunOnUserMessageCallback

func (pm *PluginManager) RunOnUserMessageCallback(cctx agent.InvocationContext, userMessage *genai.Content) (*genai.Content, error)

RunOnUserMessageCallback runs the OnUserMessageCallback for all plugins.