middleware

package
v0.35.0 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Dec 29, 2025 License: MIT Imports: 24 Imported by: 0

Documentation

Overview

Package middleware provides the telemetry middleware for OpenTelemetry GenAI Semantic Conventions.

Index

Constants

View Source 
const (
	WarningTypeHelperOverflow       = "helper_overflow"
	WarningTypePrematureAbstraction = "premature_abstraction"
	WarningTypeUnusedFeature        = "unused_feature"
	WarningTypeBackwardsCompatHack  = "backwards_compat_hack"
	WarningTypeOverEngineering      = "over_engineering"
)

警告类型常量

View Source 
const (
	// AGENT_MEMORY_FILE_PATH Agent 记忆文件的默认路径
	AGENT_MEMORY_FILE_PATH = "/agent.md"
)
View Source 
const FILESYSTEM_SYSTEM_PROMPT = `` /* 531-byte string literal not displayed */

FILESYSTEM_SYSTEM_PROMPT 文件系统提示词

View Source 
const HITL_SYSTEM_PROMPT = `` /* 587-byte string literal not displayed */

HITL_SYSTEM_PROMPT HITL 系统提示词(可选)

View Source 
const (
	// MetadataKeyEventEmitter 事件发送器的 Metadata key
	// 值类型: EventEmitterFunc
	MetadataKeyEventEmitter = "event_emitter"
)

Metadata Keys

View Source 
const TODO_LIST_SYSTEM_PROMPT = `` /* 1993-byte string literal not displayed */

TODO_LIST_SYSTEM_PROMPT 任务列表系统提示词

View Source 
const TODO_TOOL_DESCRIPTION = `` /* 1720-byte string literal not displayed */

TODO_TOOL_DESCRIPTION write_todos 工具详细描述

Variables

View Source 
var DefaultRegistry = NewRegistry()

DefaultRegistry 全局默认注册表

Functions

func ExtractMessages 

func ExtractMessages(messages []types.Message, limit int) []types.Message

ExtractMessages 从父代理状态提取消息 用于子代理继承部分上下文

func IsolateContext 

func IsolateContext(parentState map[string]any, includeKeys []string) map[string]any

IsolateContext 隔离上下文 从父代理状态中提取必要信息,创建干净的子代理上下文

func NewExperimentalUpdateWorkingMemoryTool 

func NewExperimentalUpdateWorkingMemoryTool(manager *memory.WorkingMemoryManager) tools.Tool

NewExperimentalUpdateWorkingMemoryTool 创建实验性 Working Memory 更新工具

func NewMemorySearchTool 

func NewMemorySearchTool(manager *memory.Manager, baseNamespace string) tools.Tool

NewMemorySearchTool 创建搜索工具

func NewMemoryWriteTool 

func NewMemoryWriteTool(manager *memory.Manager, baseNamespace string) tools.Tool

NewMemoryWriteTool 创建写入工具

func NewUpdateWorkingMemoryTool 

func NewUpdateWorkingMemoryTool(manager *memory.WorkingMemoryManager) tools.Tool

NewUpdateWorkingMemoryTool 创建 Working Memory 更新工具

Types

type ActionRequest 

type ActionRequest struct {
	ToolCallID string         // 工具调用ID
	ToolName   string         // 工具名称
	Input      map[string]any // 工具输入参数
	Message    string         // 审核提示信息
}

ActionRequest 待审核的操作请求

type AgentMemoryMiddleware 

type AgentMemoryMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}

AgentMemoryMiddleware Agent 记忆中间件 功能: 1. 从 backend 加载 agent.md 文件内容 2. 将内容注入到 system prompt 开头 3. 提供长期记忆使用指南

func NewAgentMemoryMiddleware 

func NewAgentMemoryMiddleware(config *AgentMemoryMiddlewareConfig) (*AgentMemoryMiddleware, error)

NewAgentMemoryMiddleware 创建中间件

func (*AgentMemoryMiddleware) GetConfig 

func (m *AgentMemoryMiddleware) GetConfig() map[string]any

GetConfig 获取配置信息

func (*AgentMemoryMiddleware) GetMemoryContent 

func (m *AgentMemoryMiddleware) GetMemoryContent() string

GetMemoryContent 获取当前加载的记忆内容

func (*AgentMemoryMiddleware) IsMemoryLoaded 

func (m *AgentMemoryMiddleware) IsMemoryLoaded() bool

IsMemoryLoaded 检查记忆是否已加载

func (*AgentMemoryMiddleware) OnAgentStart 

func (m *AgentMemoryMiddleware) OnAgentStart(ctx context.Context, agentID string) error

OnAgentStart Agent 启动时加载记忆

func (*AgentMemoryMiddleware) ReloadMemory 

func (m *AgentMemoryMiddleware) ReloadMemory(ctx context.Context) error

ReloadMemory 重新加载记忆(用于动态更新)

func (*AgentMemoryMiddleware) Tools 

func (m *AgentMemoryMiddleware) Tools() []tools.Tool

Tools 返回由 AgentMemoryMiddleware 注入的长期记忆相关工具 这些工具专门针对 memoryPath 下的记忆文件,提供搜索与写入能力

func (*AgentMemoryMiddleware) WrapModelCall 

func (m *AgentMemoryMiddleware) WrapModelCall(ctx context.Context, req *ModelRequest, handler ModelCallHandler) (*ModelResponse, error)

WrapModelCall 包装模型调用,注入记忆到 system prompt

type AgentMemoryMiddlewareConfig 

type AgentMemoryMiddlewareConfig struct {
	Backend              backends.BackendProtocol // 存储后端
	MemoryPath           string                   // 长期记忆路径前缀,如 "/memories/"
	SystemPromptTemplate string                   // 可选,自定义模板
	// BaseNamespace 为所有记忆操作提供一个基础命名空间前缀,用于多租户隔离
	// 例如: "users/alice"、"tenants/acme"。memory_* 工具的 namespace 参数会在此基础上再追加。
	BaseNamespace string
}

AgentMemoryMiddlewareConfig 配置

type ApprovalHandler 

type ApprovalHandler func(ctx context.Context, request *ReviewRequest) ([]Decision, error)

ApprovalHandler 人工审核处理器 用于获取人工决策

type BaseMiddleware 

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

BaseMiddleware 基础中间件实现 提供默认的空实现,子类只需覆盖需要的方法

func NewBaseMiddleware 

func NewBaseMiddleware(name string, priority int) *BaseMiddleware

NewBaseMiddleware 创建基础中间件

func (*BaseMiddleware) Name 

func (m *BaseMiddleware) Name() string

func (*BaseMiddleware) OnAgentStart 

func (m *BaseMiddleware) OnAgentStart(ctx context.Context, agentID string) error

func (*BaseMiddleware) OnAgentStop 

func (m *BaseMiddleware) OnAgentStop(ctx context.Context, agentID string) error

func (*BaseMiddleware) Priority 

func (m *BaseMiddleware) Priority() int

func (*BaseMiddleware) Tools 

func (m *BaseMiddleware) Tools() []tools.Tool

func (*BaseMiddleware) WrapModelCall 

func (m *BaseMiddleware) WrapModelCall(ctx context.Context, req *ModelRequest, handler ModelCallHandler) (*ModelResponse, error)

func (*BaseMiddleware) WrapToolCall 

func (m *BaseMiddleware) WrapToolCall(ctx context.Context, req *ToolCallRequest, handler ToolCallHandler) (*ToolCallResponse, error)

type CompactionStrategy added in v0.33.0 

type CompactionStrategy struct {
	// ToolResponseRemovalSteps 渐进式删除工具响应的比例步骤
	// 例如 [0.0, 0.1, 0.2, 0.5, 1.0] 表示先尝试不删除，然后删除10%，20%，50%，最后100%
	ToolResponseRemovalSteps []float64

	// PreserveSummary 是否保留摘要消息（而非替换原消息）
	// 如果为 true，原消息会被标记为 invisible，摘要被标记为 agent_only
	PreserveSummary bool

	// MaxSummaryTokens 摘要消息的最大 token 数
	MaxSummaryTokens int
}

CompactionStrategy 压缩策略 定义了在触发总结之前，如何渐进式地压缩工具响应

func DefaultCompactionStrategy added in v0.33.0 

func DefaultCompactionStrategy() *CompactionStrategy

DefaultCompactionStrategy 默认压缩策略

type ConversationPhase added in v0.16.0 

type ConversationPhase struct {
	Title  string
	Points []string
}

ConversationPhase 对话阶段

type Decision 

type Decision struct {
	Type        DecisionType   // 决策类型
	EditedInput map[string]any // 编辑后的参数(仅 type=edit 时有效)
	Reason      string         // 决策理由(可选)
}

Decision 人工决策

type DecisionType 

type DecisionType string

DecisionType 审核决策类型 

const (
	DecisionApprove DecisionType = "approve" // 批准执行
	DecisionReject  DecisionType = "reject"  // 拒绝执行
	DecisionEdit    DecisionType = "edit"    // 编辑参数后执行
)

type EditTool 

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

EditTool 文件编辑工具

func (*EditTool) Description 

func (t *EditTool) Description() string

func (*EditTool) Execute 

func (t *EditTool) Execute(ctx context.Context, input map[string]any, tc *tools.ToolContext) (any, error)

func (*EditTool) InputSchema 

func (t *EditTool) InputSchema() map[string]any

func (*EditTool) Name 

func (t *EditTool) Name() string

func (*EditTool) Prompt 

func (t *EditTool) Prompt() string

type EventEmitterFunc added in v0.25.0 

type EventEmitterFunc func(event types.EventType)

EventEmitterFunc 事件发送函数类型 中间件可以通过此函数发送事件到 EventBus

type ExperimentalUpdateWorkingMemoryTool 

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

ExperimentalUpdateWorkingMemoryTool Working Memory 更新工具（实验性，支持 find/replace） 对标 Mastra 的 __experimental_updateWorkingMemoryToolVNext

func (*ExperimentalUpdateWorkingMemoryTool) Description 

func (t *ExperimentalUpdateWorkingMemoryTool) Description() string

func (*ExperimentalUpdateWorkingMemoryTool) Execute 

func (t *ExperimentalUpdateWorkingMemoryTool) Execute(ctx context.Context, input map[string]any, tc *tools.ToolContext) (any, error)

func (*ExperimentalUpdateWorkingMemoryTool) InputSchema 

func (t *ExperimentalUpdateWorkingMemoryTool) InputSchema() map[string]any

func (*ExperimentalUpdateWorkingMemoryTool) Name 

func (t *ExperimentalUpdateWorkingMemoryTool) Name() string

func (*ExperimentalUpdateWorkingMemoryTool) Prompt 

func (t *ExperimentalUpdateWorkingMemoryTool) Prompt() string

type FailedToolCall 

type FailedToolCall struct {
	Timestamp  time.Time
	ToolName   string
	ToolCallID string
	Input      map[string]any
	Error      string
}

FailedToolCall 失败的工具调用记录

type FilesystemMiddleware 

type FilesystemMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}

FilesystemMiddleware 文件系统中间件 功能: 1. 注入文件系统工具 (ls, read, write, edit, glob, grep) 2. 自动驱逐大结果到文件 3. 增强系统提示词 4. 路径安全验证

func NewFilesystemMiddleware 

func NewFilesystemMiddleware(config *FilesystemMiddlewareConfig) *FilesystemMiddleware

NewFilesystemMiddleware 创建文件系统中间件

func (*FilesystemMiddleware) Tools 

func (m *FilesystemMiddleware) Tools() []tools.Tool

Tools 返回文件系统工具

func (*FilesystemMiddleware) WrapModelCall 

func (m *FilesystemMiddleware) WrapModelCall(ctx context.Context, req *ModelRequest, handler ModelCallHandler) (*ModelResponse, error)

WrapModelCall 包装模型调用

func (*FilesystemMiddleware) WrapToolCall 

func (m *FilesystemMiddleware) WrapToolCall(ctx context.Context, req *ToolCallRequest, handler ToolCallHandler) (*ToolCallResponse, error)

WrapToolCall 包装工具调用

type FilesystemMiddlewareConfig 

type FilesystemMiddlewareConfig struct {
	Backend                 backends.BackendProtocol // 后端存储
	TokenLimit              int                      // 大结果驱逐阈值(tokens)
	EnableEviction          bool                     // 是否启用自动驱逐
	AllowedPathPrefixes     []string                 // 允许的路径前缀列表(用于路径安全验证)
	EnablePathValidation    bool                     // 是否启用路径验证(默认: true)
	CustomToolDescriptions  map[string]string        // 自定义工具描述
	SystemPromptOverride    string                   // 覆盖默认系统提示词（设置为空字符串可禁用注入）
	HasSystemPromptOverride bool                     // 是否明确设置了 SystemPromptOverride
}

FilesystemMiddlewareConfig 文件系统中间件配置

type GlobTool 

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

GlobTool Glob 模式匹配工具

func (*GlobTool) Description 

func (t *GlobTool) Description() string

func (*GlobTool) Execute 

func (t *GlobTool) Execute(ctx context.Context, input map[string]any, tc *tools.ToolContext) (any, error)

func (*GlobTool) InputSchema 

func (t *GlobTool) InputSchema() map[string]any

func (*GlobTool) Name 

func (t *GlobTool) Name() string

func (*GlobTool) Prompt 

func (t *GlobTool) Prompt() string

type GrepTool 

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

GrepTool 正则搜索工具

func (*GrepTool) Description 

func (t *GrepTool) Description() string

func (*GrepTool) Execute 

func (t *GrepTool) Execute(ctx context.Context, input map[string]any, tc *tools.ToolContext) (any, error)

func (*GrepTool) InputSchema 

func (t *GrepTool) InputSchema() map[string]any

func (*GrepTool) Name 

func (t *GrepTool) Name() string

func (*GrepTool) Prompt 

func (t *GrepTool) Prompt() string

type HumanInTheLoopMiddleware 

type HumanInTheLoopMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}

HumanInTheLoopMiddleware 人工审核中间件 功能: 1. 拦截敏感工具调用 2. 请求人工批准/拒绝/编辑 3. 支持批量审核 4. 灵活的审核配置

func NewHumanInTheLoopMiddleware 

func NewHumanInTheLoopMiddleware(config *HumanInTheLoopMiddlewareConfig) (*HumanInTheLoopMiddleware, error)

NewHumanInTheLoopMiddleware 创建 HITL 中间件

func (*HumanInTheLoopMiddleware) GetInterruptConfig 

func (m *HumanInTheLoopMiddleware) GetInterruptConfig(toolName string) (*InterruptConfig, bool)

GetInterruptConfig 获取工具的审核配置

func (*HumanInTheLoopMiddleware) IsToolInterruptible 

func (m *HumanInTheLoopMiddleware) IsToolInterruptible(toolName string) bool

IsToolInterruptible 检查工具是否需要审核

func (*HumanInTheLoopMiddleware) ListInterruptibleTools 

func (m *HumanInTheLoopMiddleware) ListInterruptibleTools() []string

ListInterruptibleTools 列出所有需要审核的工具

func (*HumanInTheLoopMiddleware) SetApprovalHandler 

func (m *HumanInTheLoopMiddleware) SetApprovalHandler(handler ApprovalHandler)

SetApprovalHandler 设置审核处理器

func (*HumanInTheLoopMiddleware) WrapToolCall 

func (m *HumanInTheLoopMiddleware) WrapToolCall(ctx context.Context, req *ToolCallRequest, handler ToolCallHandler) (*ToolCallResponse, error)

WrapToolCall 拦截工具调用,请求人工审核

type HumanInTheLoopMiddlewareConfig 

type HumanInTheLoopMiddlewareConfig struct {
	// InterruptOn 配置哪些工具需要审核
	// key: 工具名称, value: 审核配置
	// 支持三种格式:
	// 1. true: 启用默认审核配置
	// 2. false: 禁用审核
	// 3. InterruptConfig: 自定义审核配置
	InterruptOn map[string]any

	// ApprovalHandler 人工审核处理器
	// 如果为 nil, 默认自动批准所有请求
	ApprovalHandler ApprovalHandler

	// DefaultAllowedDecisions 默认允许的决策类型
	DefaultAllowedDecisions []DecisionType
}

HumanInTheLoopMiddlewareConfig HITL 中间件配置

type InterruptConfig 

type InterruptConfig struct {
	Enabled          bool           // 是否启用审核
	AllowedDecisions []DecisionType // 允许的决策类型(默认: approve, edit, reject)
	Message          string         // 自定义审核提示信息
}

InterruptConfig 工具审核配置

type ListSubagentsTool added in v0.13.0 

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

ListSubagentsTool 列出所有子代理工具

func (*ListSubagentsTool) Description added in v0.13.0 

func (t *ListSubagentsTool) Description() string

func (*ListSubagentsTool) Execute added in v0.13.0 

func (t *ListSubagentsTool) Execute(ctx context.Context, input map[string]any, tc *tools.ToolContext) (any, error)

func (*ListSubagentsTool) InputSchema added in v0.13.0 

func (t *ListSubagentsTool) InputSchema() map[string]any

func (*ListSubagentsTool) Name added in v0.13.0 

func (t *ListSubagentsTool) Name() string

func (*ListSubagentsTool) Prompt added in v0.13.0 

func (t *ListSubagentsTool) Prompt() string

type LogicMemoryMiddleware added in v0.22.0 

type LogicMemoryMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}

LogicMemoryMiddleware Logic Memory 中间件 功能： 1. 自动捕获用户交互事件，识别并记录用户偏好和行为模式 2. 将相关 Memory 注入到 system prompt，实现个性化响应 3. 提供 Logic Memory 管理工具供 Agent 主动查询和更新

func NewLogicMemoryMiddleware added in v0.22.0 

func NewLogicMemoryMiddleware(config *LogicMemoryMiddlewareConfig) (*LogicMemoryMiddleware, error)

NewLogicMemoryMiddleware 创建 Logic Memory 中间件

func (*LogicMemoryMiddleware) CaptureEvent added in v0.22.0 

func (m *LogicMemoryMiddleware) CaptureEvent(event *logic.Event)

CaptureEvent 公开方法：允许外部代码手动捕获事件 这对于捕获 Middleware 无法自动感知的事件很有用

func (*LogicMemoryMiddleware) CaptureUserFeedback added in v0.22.0 

func (m *LogicMemoryMiddleware) CaptureUserFeedback(namespace string, feedback string, rating int, metadata map[string]any)

CaptureUserFeedback 捕获用户反馈事件

func (*LogicMemoryMiddleware) CaptureUserMessage added in v0.22.0 

func (m *LogicMemoryMiddleware) CaptureUserMessage(namespace, content string, metadata map[string]any)

CaptureUserMessage 捕获用户消息事件

func (*LogicMemoryMiddleware) CaptureUserRevision added in v0.22.0 

func (m *LogicMemoryMiddleware) CaptureUserRevision(namespace string, original, revised string, metadata map[string]any)

CaptureUserRevision 捕获用户修改事件（核心功能：学习用户偏好）

func (*LogicMemoryMiddleware) GetConfig added in v0.22.0 

func (m *LogicMemoryMiddleware) GetConfig() map[string]any

GetConfig 获取配置信息

func (*LogicMemoryMiddleware) GetManager added in v0.22.0 

func (m *LogicMemoryMiddleware) GetManager() *logic.Manager

GetManager 获取 Logic Memory 管理器

func (*LogicMemoryMiddleware) OnAgentStart added in v0.22.0 

func (m *LogicMemoryMiddleware) OnAgentStart(ctx context.Context, agentID string) error

OnAgentStart Agent 启动时启动事件处理器

func (*LogicMemoryMiddleware) OnAgentStop added in v0.22.0 

func (m *LogicMemoryMiddleware) OnAgentStop(ctx context.Context, agentID string) error

OnAgentStop Agent 停止时停止事件处理器

func (*LogicMemoryMiddleware) Tools added in v0.22.0 

func (m *LogicMemoryMiddleware) Tools() []tools.Tool

Tools 返回 Logic Memory 相关工具

func (*LogicMemoryMiddleware) WrapModelCall added in v0.22.0 

func (m *LogicMemoryMiddleware) WrapModelCall(ctx context.Context, req *ModelRequest, handler ModelCallHandler) (*ModelResponse, error)

WrapModelCall 包装模型调用，注入 Logic Memory 到 system prompt

func (*LogicMemoryMiddleware) WrapToolCall added in v0.22.0 

func (m *LogicMemoryMiddleware) WrapToolCall(ctx context.Context, req *ToolCallRequest, handler ToolCallHandler) (*ToolCallResponse, error)

WrapToolCall 包装工具调用，捕获工具执行事件

type LogicMemoryMiddlewareConfig added in v0.22.0 

type LogicMemoryMiddlewareConfig struct {
	// Manager Logic Memory 管理器（必需）
	Manager *logic.Manager

	// NamespaceExtractor 从请求中提取 namespace 的函数
	// 如果为空，使用默认提取器（从 metadata 中获取 user_id 或 namespace）
	NamespaceExtractor NamespaceExtractor

	// EnableCapture 是否启用自动捕获（默认 true）
	EnableCapture bool

	// CaptureChannels 要捕获的事件通道（默认 Control + Monitor）
	CaptureChannels []types.AgentChannel

	// EnableInjection 是否启用 Memory 注入到 Prompt（默认 true）
	EnableInjection bool

	// InjectionPoint 注入位置："system_prompt_start", "system_prompt_end", "both"
	InjectionPoint string

	// MaxMemories 注入的最大 Memory 数量（TopK，默认 5）
	MaxMemories int

	// MinConfidence 最低置信度阈值（默认 0.6）
	MinConfidence float64

	// AsyncCapture 是否异步捕获事件（默认 true）
	// 异步模式不阻塞主流程，但可能丢失部分事件
	AsyncCapture bool

	// EventBufferSize 事件缓冲区大小（默认 100）
	EventBufferSize int

	// CacheTTL Memory 缓存时间（默认 5 分钟）
	CacheTTL time.Duration

	// SystemPromptTemplate Memory 注入的模板
	// 占位符：%s = Memory 列表的 Markdown 格式
	SystemPromptTemplate string

	// Priority 中间件优先级（默认 7，在 working_memory 之后）
	Priority int
}

LogicMemoryMiddlewareConfig Logic Memory 中间件配置

type LogicMemoryQueryTool added in v0.22.0 

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

LogicMemoryQueryTool 查询 Logic Memory 工具 允许 Agent 主动查询用户偏好和行为模式

func NewLogicMemoryQueryTool added in v0.22.0 

func NewLogicMemoryQueryTool(manager *logic.Manager) *LogicMemoryQueryTool

NewLogicMemoryQueryTool 创建 Logic Memory 查询工具

func (*LogicMemoryQueryTool) Description added in v0.22.0 

func (t *LogicMemoryQueryTool) Description() string

Description 返回工具描述

func (*LogicMemoryQueryTool) Execute added in v0.22.0 

func (t *LogicMemoryQueryTool) Execute(ctx context.Context, input map[string]any, tctx *tools.ToolContext) (any, error)

Execute 执行工具

func (*LogicMemoryQueryTool) InputSchema added in v0.22.0 

func (t *LogicMemoryQueryTool) InputSchema() map[string]any

InputSchema 返回输入 JSON Schema

func (*LogicMemoryQueryTool) Name added in v0.22.0 

func (t *LogicMemoryQueryTool) Name() string

Name 返回工具名称

func (*LogicMemoryQueryTool) Prompt added in v0.22.0 

func (t *LogicMemoryQueryTool) Prompt() string

Prompt 返回工具使用说明

type LogicMemoryUpdateTool added in v0.22.0 

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

LogicMemoryUpdateTool 更新 Logic Memory 工具 允许 Agent 主动记录用户偏好和行为模式

func NewLogicMemoryUpdateTool added in v0.22.0 

func NewLogicMemoryUpdateTool(manager *logic.Manager) *LogicMemoryUpdateTool

NewLogicMemoryUpdateTool 创建 Logic Memory 更新工具

func (*LogicMemoryUpdateTool) Description added in v0.22.0 

func (t *LogicMemoryUpdateTool) Description() string

Description 返回工具描述

func (*LogicMemoryUpdateTool) Execute added in v0.22.0 

func (t *LogicMemoryUpdateTool) Execute(ctx context.Context, input map[string]any, tctx *tools.ToolContext) (any, error)

Execute 执行工具

func (*LogicMemoryUpdateTool) InputSchema added in v0.22.0 

func (t *LogicMemoryUpdateTool) InputSchema() map[string]any

InputSchema 返回输入 JSON Schema

func (*LogicMemoryUpdateTool) Name added in v0.22.0 

func (t *LogicMemoryUpdateTool) Name() string

Name 返回工具名称

func (*LogicMemoryUpdateTool) Prompt added in v0.22.0 

func (t *LogicMemoryUpdateTool) Prompt() string

Prompt 返回工具使用说明

type LsTool 

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

LsTool 目录列表工具

func (*LsTool) Description 

func (t *LsTool) Description() string

func (*LsTool) Execute 

func (t *LsTool) Execute(ctx context.Context, input map[string]any, tc *tools.ToolContext) (any, error)

func (*LsTool) InputSchema 

func (t *LsTool) InputSchema() map[string]any

func (*LsTool) Name 

func (t *LsTool) Name() string

func (*LsTool) Prompt 

func (t *LsTool) Prompt() string

type MemorySearchTool 

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

MemorySearchTool 基于 Manager 的记忆搜索工具 特点: - 仅在 memoryPath 下搜索,避免扫描整个文件系统 - 默认使用大小写不敏感的字面量匹配

func (*MemorySearchTool) Description 

func (t *MemorySearchTool) Description() string

func (*MemorySearchTool) Execute 

func (t *MemorySearchTool) Execute(ctx context.Context, input map[string]any, tc *tools.ToolContext) (any, error)

func (*MemorySearchTool) InputSchema 

func (t *MemorySearchTool) InputSchema() map[string]any

func (*MemorySearchTool) Name 

func (t *MemorySearchTool) Name() string

func (*MemorySearchTool) Prompt 

func (t *MemorySearchTool) Prompt() string

type MemoryWriteTool 

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

MemoryWriteTool 记忆写入/追加工具 通过普通文本写入而不是向量数据库,保持可读可编辑

func (*MemoryWriteTool) Description 

func (t *MemoryWriteTool) Description() string

func (*MemoryWriteTool) Execute 

func (t *MemoryWriteTool) Execute(ctx context.Context, input map[string]any, tc *tools.ToolContext) (any, error)

func (*MemoryWriteTool) InputSchema 

func (t *MemoryWriteTool) InputSchema() map[string]any

func (*MemoryWriteTool) Name 

func (t *MemoryWriteTool) Name() string

func (*MemoryWriteTool) Prompt 

func (t *MemoryWriteTool) Prompt() string

type Middleware 

type Middleware interface {
	// Name 返回中间件名称
	Name() string

	// Priority 返回优先级 (数值越小优先级越高,越早执行)
	// 建议范围: 0-1000
	// 0-100: 系统核心中间件
	// 100-500: 功能中间件
	// 500-1000: 用户自定义中间件
	Priority() int

	// Tools 注入工具列表
	// 返回中间件提供的工具,会被合并到 Agent 的工具集中
	Tools() []tools.Tool

	// WrapModelCall 包装模型调用
	// 在模型调用前后执行自定义逻辑
	// handler: 下一层处理器(可能是下一个中间件或最终的模型调用)
	WrapModelCall(ctx context.Context, req *ModelRequest, handler ModelCallHandler) (*ModelResponse, error)

	// WrapToolCall 包装工具调用
	// 在工具调用前后执行自定义逻辑
	// handler: 下一层处理器(可能是下一个中间件或最终的工具执行)
	WrapToolCall(ctx context.Context, req *ToolCallRequest, handler ToolCallHandler) (*ToolCallResponse, error)

	// OnAgentStart Agent 启动时回调
	OnAgentStart(ctx context.Context, agentID string) error

	// OnAgentStop Agent 停止时回调
	OnAgentStop(ctx context.Context, agentID string) error
}

Middleware 中间件接口 中间件采用洋葱模型,支持请求和响应的拦截处理

func BuildSubAgentMiddlewareStack 

func BuildSubAgentMiddlewareStack(spec SubAgentSpec, parentMiddlewares []Middleware) []Middleware

BuildSubAgentMiddlewareStack 构建子代理的中间件栈 根据子代理规格决定是否继承父代理中间件

type MiddlewareFactory 

type MiddlewareFactory func(config *MiddlewareFactoryConfig) (Middleware, error)

MiddlewareFactory 中间件工厂函数 config参数可用于传递Provider等依赖

type MiddlewareFactoryConfig 

type MiddlewareFactoryConfig struct {
	Provider     provider.Provider
	AgentID      string
	Metadata     map[string]any
	CustomConfig map[string]any  // 自定义配置
	Sandbox      sandbox.Sandbox // 可选: 需要访问沙箱文件系统的中间件
}

MiddlewareFactoryConfig 工厂配置

type ModelCallHandler 

type ModelCallHandler func(ctx context.Context, req *ModelRequest) (*ModelResponse, error)

ModelCallHandler 模型调用处理器

type ModelRequest 

type ModelRequest struct {
	Messages     []types.Message
	SystemPrompt string
	Tools        []tools.Tool
	Metadata     map[string]any
}

ModelRequest 模型请求

func (*ModelRequest) EmitEvent added in v0.25.0 

func (r *ModelRequest) EmitEvent(event types.EventType)

EmitEvent 发送事件的便捷方法 如果 Metadata 中包含 EventEmitter，则调用它发送事件

type ModelResponse 

type ModelResponse struct {
	Message  types.Message
	Metadata map[string]any
}

ModelResponse 模型响应

type NamespaceExtractor added in v0.22.0 

type NamespaceExtractor func(req *ModelRequest) string

NamespaceExtractor 从请求中提取 namespace 的函数 应用层可以自定义此函数来实现不同的租户隔离策略

type ObservationCompressionConfig added in v0.17.0 

type ObservationCompressionConfig struct {
	// Compressor 自定义压缩器（可选，默认使用 DefaultObservationCompressor）
	Compressor memory.ObservationCompressor

	// ReferenceRegistry 引用注册表（可选）
	ReferenceRegistry memory.ReferenceRegistry

	// Enabled 是否启用压缩
	Enabled bool

	// MinContentLength 触发压缩的最小内容长度
	MinContentLength int
}

ObservationCompressionConfig 配置

type ObservationCompressionMiddleware added in v0.17.0 

type ObservationCompressionMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}

ObservationCompressionMiddleware 观察结果压缩中间件 用于压缩工具执行结果，同时保留可恢复的引用信息 这是 Manus 团队"文件系统作为上下文"理念的实现

func NewObservationCompressionMiddleware added in v0.17.0 

func NewObservationCompressionMiddleware(config *ObservationCompressionConfig) *ObservationCompressionMiddleware

NewObservationCompressionMiddleware 创建观察结果压缩中间件

func (*ObservationCompressionMiddleware) CompressString added in v0.17.0 

func (m *ObservationCompressionMiddleware) CompressString(ctx context.Context, toolName, content string) (string, []memory.Reference, error)

CompressString 便捷方法：直接压缩字符串内容

func (*ObservationCompressionMiddleware) GetReferenceRegistry added in v0.17.0 

func (m *ObservationCompressionMiddleware) GetReferenceRegistry() memory.ReferenceRegistry

GetReferenceRegistry 获取引用注册表

func (*ObservationCompressionMiddleware) GetStats added in v0.17.0 

func (m *ObservationCompressionMiddleware) GetStats() map[string]any

GetStats 获取压缩统计

func (*ObservationCompressionMiddleware) WrapModelCall added in v0.17.0 

func (m *ObservationCompressionMiddleware) WrapModelCall(
	ctx context.Context,
	req *ModelRequest,
	handler ModelCallHandler,
) (*ModelResponse, error)

WrapModelCall 包装模型调用，在发送前压缩历史工具结果

type PatchToolCallsMiddleware 

type PatchToolCallsMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}

PatchToolCallsMiddleware 工具调用错误恢复中间件 功能: 1. 捕获工具调用异常,防止崩溃 2. 提供友好的错误响应 3. 记录失败的工具调用供调试

func NewPatchToolCallsMiddleware 

func NewPatchToolCallsMiddleware(config *PatchToolCallsMiddlewareConfig) *PatchToolCallsMiddleware

NewPatchToolCallsMiddleware 创建中间件

func (*PatchToolCallsMiddleware) ClearFailedCalls 

func (m *PatchToolCallsMiddleware) ClearFailedCalls()

ClearFailedCalls 清空失败记录

func (*PatchToolCallsMiddleware) GetFailedCallCount 

func (m *PatchToolCallsMiddleware) GetFailedCallCount() int

GetFailedCallCount 获取失败次数

func (*PatchToolCallsMiddleware) GetFailedCalls 

func (m *PatchToolCallsMiddleware) GetFailedCalls() []FailedToolCall

GetFailedCalls 获取失败的工具调用记录

func (*PatchToolCallsMiddleware) GetFailedCallsByTool 

func (m *PatchToolCallsMiddleware) GetFailedCallsByTool(toolName string) []FailedToolCall

GetFailedCallsByTool 获取特定工具的失败记录

func (*PatchToolCallsMiddleware) WrapToolCall 

func (m *PatchToolCallsMiddleware) WrapToolCall(ctx context.Context, req *ToolCallRequest, handler ToolCallHandler) (*ToolCallResponse, error)

WrapToolCall 包装工具调用,捕获异常

type PatchToolCallsMiddlewareConfig 

type PatchToolCallsMiddlewareConfig struct {
	EnableLogging  bool // 是否记录失败调用
	MaxFailedCalls int  // 最多保留多少条失败记录
	ProvideHints   bool // 是否在错误响应中提供修复提示
}

PatchToolCallsMiddlewareConfig 配置

type QuerySubagentTool added in v0.13.0 

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

QuerySubagentTool 查询子代理状态工具

func (*QuerySubagentTool) Description added in v0.13.0 

func (t *QuerySubagentTool) Description() string

func (*QuerySubagentTool) Execute added in v0.13.0 

func (t *QuerySubagentTool) Execute(ctx context.Context, input map[string]any, tc *tools.ToolContext) (any, error)

func (*QuerySubagentTool) InputSchema added in v0.13.0 

func (t *QuerySubagentTool) InputSchema() map[string]any

func (*QuerySubagentTool) Name added in v0.13.0 

func (t *QuerySubagentTool) Name() string

func (*QuerySubagentTool) Prompt added in v0.13.0 

func (t *QuerySubagentTool) Prompt() string

type ReasoningMiddleware added in v0.13.0 

type ReasoningMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}

ReasoningMiddleware 推理中间件

func NewReasoningMiddleware added in v0.13.0 

func NewReasoningMiddleware(config *ReasoningMiddlewareConfig) *ReasoningMiddleware

NewReasoningMiddleware 创建推理中间件

func (*ReasoningMiddleware) Tools added in v0.13.0 

func (rm *ReasoningMiddleware) Tools() []tools.Tool

Tools 返回中间件提供的工具

func (*ReasoningMiddleware) WrapModelCall added in v0.13.0 

func (rm *ReasoningMiddleware) WrapModelCall(ctx context.Context, req *ModelRequest, handler ModelCallHandler) (*ModelResponse, error)

WrapModelCall 包装模型调用（实现 Middleware 接口）

type ReasoningMiddlewareConfig added in v0.13.0 

type ReasoningMiddlewareConfig struct {
	Provider      provider.Provider
	MinSteps      int
	MaxSteps      int
	MinConfidence float64
	UseJSON       bool
	Temperature   float64
	Enabled       bool
	Priority      int
}

ReasoningMiddlewareConfig 推理中间件配置

type ReasoningTool added in v0.13.0 

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

ReasoningTool 推理工具

func NewReasoningTool added in v0.13.0 

func NewReasoningTool(engine *reasoning.Engine) *ReasoningTool

NewReasoningTool 创建推理工具

func (*ReasoningTool) Description added in v0.13.0 

func (rt *ReasoningTool) Description() string

Description 返回工具描述

func (*ReasoningTool) Execute added in v0.13.0 

func (rt *ReasoningTool) Execute(ctx context.Context, input map[string]any, tc *tools.ToolContext) (any, error)

Execute 执行工具

func (*ReasoningTool) InputSchema added in v0.13.0 

func (rt *ReasoningTool) InputSchema() map[string]any

InputSchema 返回输入 Schema

func (*ReasoningTool) Name added in v0.13.0 

func (rt *ReasoningTool) Name() string

Name 返回工具名称

func (*ReasoningTool) Prompt added in v0.13.0 

func (rt *ReasoningTool) Prompt() string

Prompt 返回工具提示

type Registry 

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

Registry 中间件注册表

func NewRegistry 

func NewRegistry() *Registry

NewRegistry 创建注册表

func (*Registry) Create 

func (r *Registry) Create(name string, config *MiddlewareFactoryConfig) (Middleware, error)

Create 创建中间件实例

func (*Registry) List 

func (r *Registry) List() []string

List 列出所有已注册的中间件名称

func (*Registry) Register 

func (r *Registry) Register(name string, factory MiddlewareFactory)

Register 注册中间件工厂

type ResumeSubagentTool added in v0.13.0 

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

ResumeSubagentTool 恢复子代理工具

func (*ResumeSubagentTool) Description added in v0.13.0 

func (t *ResumeSubagentTool) Description() string

func (*ResumeSubagentTool) Execute added in v0.13.0 

func (t *ResumeSubagentTool) Execute(ctx context.Context, input map[string]any, tc *tools.ToolContext) (any, error)

func (*ResumeSubagentTool) InputSchema added in v0.13.0 

func (t *ResumeSubagentTool) InputSchema() map[string]any

func (*ResumeSubagentTool) Name added in v0.13.0 

func (t *ResumeSubagentTool) Name() string

func (*ResumeSubagentTool) Prompt added in v0.13.0 

func (t *ResumeSubagentTool) Prompt() string

type ReviewRequest 

type ReviewRequest struct {
	ActionRequests []ActionRequest   // 待审核的操作列表
	ReviewConfigs  []InterruptConfig // 每个操作的审核配置
}

ReviewRequest 审核请求

type SimpleSubAgent 

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

SimpleSubAgent 简单子代理实现 用于测试和演示

func NewSimpleSubAgent 

func NewSimpleSubAgent(name, prompt string, execFn func(context.Context, string, map[string]any) (string, error)) *SimpleSubAgent

func (*SimpleSubAgent) Close 

func (a *SimpleSubAgent) Close() error

func (*SimpleSubAgent) Execute 

func (a *SimpleSubAgent) Execute(ctx context.Context, description string, parentContext map[string]any) (string, error)

func (*SimpleSubAgent) Name 

func (a *SimpleSubAgent) Name() string

type SimplicityCheckerConfig added in v0.18.0 

type SimplicityCheckerConfig struct {
	// Enabled 是否启用检测
	Enabled bool

	// MaxHelperFunctions 最大辅助函数创建数（单次会话）
	MaxHelperFunctions int

	// WarnOnPrematureAbstraction 是否警告过早抽象
	WarnOnPrematureAbstraction bool

	// WarnOnUnusedParams 是否警告未使用的参数重命名
	WarnOnUnusedParams bool

	// OnWarning 警告回调（可选）
	OnWarning func(warning SimplicityWarning)
}

SimplicityCheckerConfig 简洁性检查配置

type SimplicityCheckerMiddleware added in v0.18.0 

type SimplicityCheckerMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}

SimplicityCheckerMiddleware 简洁性检查中间件 检测过度工程和未请求的功能添加，发出警告但不阻断执行

func NewSimplicityCheckerMiddleware added in v0.18.0 

func NewSimplicityCheckerMiddleware(config *SimplicityCheckerConfig) *SimplicityCheckerMiddleware

NewSimplicityCheckerMiddleware 创建简洁性检查中间件

func (*SimplicityCheckerMiddleware) GetWarnings added in v0.18.0 

func (m *SimplicityCheckerMiddleware) GetWarnings() []SimplicityWarning

GetWarnings 获取所有已发出的警告

func (*SimplicityCheckerMiddleware) OnAgentStart added in v0.18.0 

func (m *SimplicityCheckerMiddleware) OnAgentStart(ctx context.Context, agentID string) error

OnAgentStart Agent 启动时重置统计

func (*SimplicityCheckerMiddleware) Reset added in v0.18.0 

func (m *SimplicityCheckerMiddleware) Reset()

Reset 重置会话统计

func (*SimplicityCheckerMiddleware) WrapToolCall added in v0.18.0 

func (m *SimplicityCheckerMiddleware) WrapToolCall(ctx context.Context, req *ToolCallRequest, handler ToolCallHandler) (*ToolCallResponse, error)

WrapToolCall 包装工具调用，检测简洁性问题

type SimplicityWarning added in v0.18.0 

type SimplicityWarning struct {
	Type    string         `json:"type"`    // 警告类型
	Message string         `json:"message"` // 警告消息
	File    string         `json:"file"`    // 相关文件
	Details map[string]any `json:"details"` // 详细信息
}

SimplicityWarning 简洁性警告

type Stack 

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

Stack 中间件栈 管理多个中间件,构建洋葱模型的调用链

func NewStack 

func NewStack(middlewares []Middleware) *Stack

NewStack 创建中间件栈

func (*Stack) ExecuteModelCall 

func (s *Stack) ExecuteModelCall(
	ctx context.Context,
	req *ModelRequest,
	finalHandler ModelCallHandler,
) (*ModelResponse, error)

ExecuteModelCall 执行模型调用(通过中间件栈)

func (*Stack) ExecuteToolCall 

func (s *Stack) ExecuteToolCall(
	ctx context.Context,
	req *ToolCallRequest,
	finalHandler ToolCallHandler,
) (*ToolCallResponse, error)

ExecuteToolCall 执行工具调用(通过中间件栈)

func (*Stack) Middlewares 

func (s *Stack) Middlewares() []Middleware

Middlewares 返回中间件列表

func (*Stack) OnAgentStart 

func (s *Stack) OnAgentStart(ctx context.Context, agentID string) error

OnAgentStart 通知所有中间件 Agent 启动

func (*Stack) OnAgentStop 

func (s *Stack) OnAgentStop(ctx context.Context, agentID string) error

OnAgentStop 通知所有中间件 Agent 停止

func (*Stack) Tools 

func (s *Stack) Tools() []tools.Tool

Tools 收集所有中间件提供的工具

type StopSubagentTool added in v0.13.0 

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

StopSubagentTool 停止子代理工具

func (*StopSubagentTool) Description added in v0.13.0 

func (t *StopSubagentTool) Description() string

func (*StopSubagentTool) Execute added in v0.13.0 

func (t *StopSubagentTool) Execute(ctx context.Context, input map[string]any, tc *tools.ToolContext) (any, error)

func (*StopSubagentTool) InputSchema added in v0.13.0 

func (t *StopSubagentTool) InputSchema() map[string]any

func (*StopSubagentTool) Name added in v0.13.0 

func (t *StopSubagentTool) Name() string

func (*StopSubagentTool) Prompt added in v0.13.0 

func (t *StopSubagentTool) Prompt() string

type StructuredOutputMiddleware added in v0.13.0 

type StructuredOutputMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}

StructuredOutputMiddleware 在模型响应后尝试解析结构化输出，并将结果写入 Metadata。 - 若解析成功: Metadata["structured_data"] = 解析后的对象，Metadata["structured_raw_json"] = 原始 JSON 文本 - 若解析失败: 根据配置决定是否回退；错误记录在 Metadata["structured_error"]

func NewStructuredOutputMiddleware added in v0.13.0 

func NewStructuredOutputMiddleware(cfg *StructuredOutputMiddlewareConfig) (*StructuredOutputMiddleware, error)

NewStructuredOutputMiddleware 创建中间件实例

func (*StructuredOutputMiddleware) WrapModelCall added in v0.13.0 

func (m *StructuredOutputMiddleware) WrapModelCall(ctx context.Context, req *ModelRequest, handler ModelCallHandler) (*ModelResponse, error)

WrapModelCall 尝试解析结构化输出

type StructuredOutputMiddlewareConfig added in v0.13.0 

type StructuredOutputMiddlewareConfig struct {
	Spec       structured.OutputSpec
	Parser     structured.Parser // 可选，默认 JSONParser
	AllowError bool              // 解析失败时是否忽略错误并回退到原始文本
	Priority   int               // 可选，默认 60
}

StructuredOutputMiddlewareConfig 配置

type SubAgent 

type SubAgent interface {
	// Name 返回子代理名称
	Name() string

	// Execute 执行任务
	// description: 任务描述
	// context: 父代理上下文(可选)
	Execute(ctx context.Context, description string, parentContext map[string]any) (string, error)

	// Close 关闭子代理
	Close() error
}

SubAgent 子代理接口

type SubAgentFactory 

type SubAgentFactory func(ctx context.Context, spec SubAgentSpec) (SubAgent, error)

SubAgentFactory 子代理工厂函数 用于创建子代理实例

type SubAgentMiddleware 

type SubAgentMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}

SubAgentMiddleware 子代理中间件 功能: 1. 管理多个子代理实例 2. 提供 task 工具启动子代理 3. 支持任务上下文隔离 4. 支持并发执行 5. 支持异步执行和状态查询 6. 支持 Resume 机制 7. 支持资源监控和限制

func NewSubAgentMiddleware 

func NewSubAgentMiddleware(config *SubAgentMiddlewareConfig) (*SubAgentMiddleware, error)

NewSubAgentMiddleware 创建子代理中间件

func (*SubAgentMiddleware) GetMiddlewareForSubAgent 

func (m *SubAgentMiddleware) GetMiddlewareForSubAgent(spec SubAgentSpec, parentMiddlewares []Middleware) []Middleware

GetMiddlewareForSubAgent 获取子代理应该使用的中间件栈 这是一个辅助函数,供 SubAgentFactory 实现使用

func (*SubAgentMiddleware) GetSubAgent 

func (m *SubAgentMiddleware) GetSubAgent(name string) (SubAgent, error)

GetSubAgent 获取子代理

func (*SubAgentMiddleware) ListSubAgents 

func (m *SubAgentMiddleware) ListSubAgents() []string

ListSubAgents 列出所有子代理

func (*SubAgentMiddleware) OnAgentStop 

func (m *SubAgentMiddleware) OnAgentStop(ctx context.Context, agentID string) error

OnAgentStop 清理子代理

func (*SubAgentMiddleware) Tools 

func (m *SubAgentMiddleware) Tools() []tools.Tool

Tools 返回 task 工具和管理工具

type SubAgentMiddlewareConfig 

type SubAgentMiddlewareConfig struct {
	Specs                  []SubAgentSpec          // 子代理规格列表
	Factory                SubAgentFactory         // 子代理工厂
	Manager                builtin.SubagentManager // 子代理管理器（可选，默认使用 FileSubagentManager）
	EnableParallel         bool                    // 是否支持并行执行
	EnableGeneralPurpose   bool                    // 是否启用通用子代理(默认 true)
	EnableAsync            bool                    // 是否启用异步执行（默认 false）
	EnableProcessIsolation bool                    // 是否启用进程级隔离（默认 false）
	DefaultTimeout         time.Duration           // 默认超时时间（默认 1 小时）
	ParentMiddlewareGetter func() []Middleware
}

SubAgentMiddlewareConfig 子代理中间件配置

type SubAgentResult 

type SubAgentResult struct {
	Success      bool           `json:"success"`
	SubAgentType string         `json:"subagent_type"`
	Result       string         `json:"result"`
	Error        string         `json:"error,omitempty"`
	Metadata     map[string]any `json:"metadata,omitempty"`
}

SubAgentResult 子代理执行结果

type SubAgentSpec 

type SubAgentSpec struct {
	Name                string         // 子代理名称
	Description         string         // 子代理描述
	Prompt              string         // 子代理专用提示词
	Tools               []string       // 工具名称列表(可选,默认继承父代理)
	Config              map[string]any // 自定义配置
	InheritMiddlewares  bool           // 是否继承父代理的中间件栈(默认 false)
	MiddlewareOverrides []Middleware   // 子代理专用中间件(覆盖或追加)
}

SubAgentSpec 子代理规格

type SummarizationMiddleware 

type SummarizationMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}

SummarizationMiddleware 自动总结对话历史以管理上下文窗口 功能: 1. 监控消息历史的 token 数量 2. 当超过阈值时,自动总结旧消息 3. 保留最近的 N 条消息 4. 用总结消息替换旧的历史记录 5. (增强) 支持渐进式压缩和消息元数据可见性控制

func NewSummarizationMiddleware 

func NewSummarizationMiddleware(config *SummarizationMiddlewareConfig) (*SummarizationMiddleware, error)

NewSummarizationMiddleware 创建中间件

func (*SummarizationMiddleware) GetConfig 

func (m *SummarizationMiddleware) GetConfig() map[string]any

GetConfig 获取当前配置

func (*SummarizationMiddleware) GetSummarizationCount 

func (m *SummarizationMiddleware) GetSummarizationCount() int

GetSummarizationCount 获取总结触发次数

func (*SummarizationMiddleware) ResetSummarizationCount 

func (m *SummarizationMiddleware) ResetSummarizationCount()

ResetSummarizationCount 重置计数器

func (*SummarizationMiddleware) UpdateConfig 

func (m *SummarizationMiddleware) UpdateConfig(maxTokens, messagesToKeep int)

UpdateConfig 动态更新配置

func (*SummarizationMiddleware) WrapModelCall 

func (m *SummarizationMiddleware) WrapModelCall(ctx context.Context, req *ModelRequest, handler ModelCallHandler) (*ModelResponse, error)

WrapModelCall 包装模型调用,在调用前检查是否需要总结

type SummarizationMiddlewareConfig 

type SummarizationMiddlewareConfig struct {
	Summarizer             SummarizerFunc   // 用于生成总结的函数
	MaxTokensBeforeSummary int              // 触发总结的 token 阈值
	MessagesToKeep         int              // 总结后保留的最近消息数量
	SummaryPrefix          string           // 总结消息的前缀
	TokenCounter           TokenCounterFunc // 自定义 token 计数器

	// 新增配置项
	CompactionStrategy       *CompactionStrategy // 渐进式压缩策略
	UseMetadataVisibility    bool                // 使用消息元数据控制可见性（而非删除）
	EnableProgressiveCompact bool                // 是否启用渐进式压缩
}

SummarizationMiddlewareConfig 配置

type SummarizerFunc 

type SummarizerFunc func(ctx context.Context, messages []types.Message) (string, error)

SummarizerFunc 总结生成函数类型 接收要总结的消息列表,返回总结内容字符串

type TaskTool 

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

TaskTool task 工具实现

func (*TaskTool) Description 

func (t *TaskTool) Description() string

func (*TaskTool) Execute 

func (t *TaskTool) Execute(ctx context.Context, input map[string]any, tc *tools.ToolContext) (any, error)

func (*TaskTool) InputSchema 

func (t *TaskTool) InputSchema() map[string]any

func (*TaskTool) Name 

func (t *TaskTool) Name() string

func (*TaskTool) Prompt 

func (t *TaskTool) Prompt() string

type TelemetryMiddleware added in v0.32.0 

type TelemetryMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}

TelemetryMiddleware 遥测中间件 实现 OpenTelemetry GenAI Semantic Conventions 用于追踪 Agent 的 LLM 调用和工具执行

func NewTelemetryMiddleware added in v0.32.0 

func NewTelemetryMiddleware(config *TelemetryMiddlewareConfig) *TelemetryMiddleware

NewTelemetryMiddleware 创建遥测中间件

func (*TelemetryMiddleware) GetTracer added in v0.32.0 

func (m *TelemetryMiddleware) GetTracer() telemetry.Tracer

GetTracer 获取 tracer 实例

func (*TelemetryMiddleware) OnAgentStart added in v0.32.0 

func (m *TelemetryMiddleware) OnAgentStart(ctx context.Context, agentID string) error

OnAgentStart Agent 启动时创建根 span

func (*TelemetryMiddleware) OnAgentStop added in v0.32.0 

func (m *TelemetryMiddleware) OnAgentStop(ctx context.Context, agentID string) error

OnAgentStop Agent 停止时的回调

func (*TelemetryMiddleware) SetConversationID added in v0.32.0 

func (m *TelemetryMiddleware) SetConversationID(conversationID string)

SetConversationID 动态设置会话 ID

func (*TelemetryMiddleware) SetModel added in v0.32.0 

func (m *TelemetryMiddleware) SetModel(model string)

SetModel 动态设置模型名称

func (*TelemetryMiddleware) SetProvider added in v0.32.0 

func (m *TelemetryMiddleware) SetProvider(provider string)

SetProvider 动态设置提供商

func (*TelemetryMiddleware) Tools added in v0.32.0 

func (m *TelemetryMiddleware) Tools() []tools.Tool

Tools 返回中间件提供的工具 (无)

func (*TelemetryMiddleware) WrapModelCall added in v0.32.0 

func (m *TelemetryMiddleware) WrapModelCall(ctx context.Context, req *ModelRequest, handler ModelCallHandler) (*ModelResponse, error)

WrapModelCall 包装模型调用，添加 GenAI 追踪

func (*TelemetryMiddleware) WrapToolCall added in v0.32.0 

func (m *TelemetryMiddleware) WrapToolCall(ctx context.Context, req *ToolCallRequest, handler ToolCallHandler) (*ToolCallResponse, error)

WrapToolCall 包装工具调用，添加 GenAI 追踪

type TelemetryMiddlewareConfig added in v0.32.0 

type TelemetryMiddlewareConfig struct {
	// Tracer 自定义 tracer (可选，默认使用全局 tracer)
	Tracer telemetry.Tracer

	// AgentID Agent 唯一标识
	AgentID string

	// AgentName Agent 名称
	AgentName string

	// Provider LLM 提供商名称
	Provider string

	// Model 模型名称
	Model string

	// ConversationID 会话 ID (可选)
	ConversationID string

	// RecordPrompts 是否记录提示词内容 (默认 false，出于安全考虑)
	RecordPrompts bool

	// RecordCompletions 是否记录完成内容 (默认 false，出于安全考虑)
	RecordCompletions bool
}

TelemetryMiddlewareConfig 遥测中间件配置

type TodoItem 

type TodoItem struct {
	Content    string     `json:"content"`    // 任务内容描述
	Status     TodoStatus `json:"status"`     // 任务状态
	ActiveForm string     `json:"activeForm"` // 进行时描述
}

TodoItem 任务项

type TodoListMiddleware 

type TodoListMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}

TodoListMiddleware 任务列表中间件 功能: 1. 提供 write_todos 工具,允许 Agent 进行任务规划 2. 管理任务列表的状态 3. 引导 Agent 使用任务分解策略

func NewTodoListMiddleware 

func NewTodoListMiddleware(config *TodoListMiddlewareConfig) *TodoListMiddleware

NewTodoListMiddleware 创建任务列表中间件

func (*TodoListMiddleware) GetTodos 

func (m *TodoListMiddleware) GetTodos() []TodoItem

GetTodos 获取当前任务列表

func (*TodoListMiddleware) SetTodos 

func (m *TodoListMiddleware) SetTodos(todos []TodoItem)

SetTodos 设置任务列表

func (*TodoListMiddleware) Tools 

func (m *TodoListMiddleware) Tools() []tools.Tool

Tools 返回 write_todos 工具

func (*TodoListMiddleware) WrapModelCall 

func (m *TodoListMiddleware) WrapModelCall(ctx context.Context, req *ModelRequest, handler ModelCallHandler) (*ModelResponse, error)

WrapModelCall 包装模型调用,注入系统提示词

type TodoListMiddlewareConfig 

type TodoListMiddlewareConfig struct {
	// StoreGetter 从外部状态获取任务列表
	// 如果为 nil, 使用内部存储
	StoreGetter func() any

	// StoreSetter 保存任务列表到外部状态
	// 如果为 nil, 使用内部存储
	StoreSetter func([]TodoItem)

	// EnableSystemPrompt 是否注入系统提示词
	EnableSystemPrompt bool
}

TodoListMiddlewareConfig 任务列表中间件配置

type TodoStatus 

type TodoStatus string

TodoStatus 任务状态 

const (
	TodoStatusPending    TodoStatus = "pending"
	TodoStatusInProgress TodoStatus = "in_progress"
	TodoStatusCompleted  TodoStatus = "completed"
)

type TokenCounterFunc 

type TokenCounterFunc func(messages []types.Message) int

TokenCounterFunc 自定义 token 计数函数类型

type ToolCallHandler 

type ToolCallHandler func(ctx context.Context, req *ToolCallRequest) (*ToolCallResponse, error)

ToolCallHandler 工具调用处理器

type ToolCallRequest 

type ToolCallRequest struct {
	ToolCallID string
	ToolName   string
	ToolInput  map[string]any
	Tool       tools.Tool
	Context    *tools.ToolContext
	Metadata   map[string]any
}

ToolCallRequest 工具调用请求

type ToolCallResponse 

type ToolCallResponse struct {
	Result   any
	Metadata map[string]any
}

ToolCallResponse 工具调用响应

type UpdateWorkingMemoryTool 

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

UpdateWorkingMemoryTool Working Memory 更新工具 特点： - 自动从 ToolContext 获取 threadID 和 resourceID - 支持 JSON Schema 验证（如果配置） - 支持 Markdown 或 JSON 格式的内容

func (*UpdateWorkingMemoryTool) Description 

func (t *UpdateWorkingMemoryTool) Description() string

func (*UpdateWorkingMemoryTool) Execute 

func (t *UpdateWorkingMemoryTool) Execute(ctx context.Context, input map[string]any, tc *tools.ToolContext) (any, error)

func (*UpdateWorkingMemoryTool) InputSchema 

func (t *UpdateWorkingMemoryTool) InputSchema() map[string]any

func (*UpdateWorkingMemoryTool) Name 

func (t *UpdateWorkingMemoryTool) Name() string

func (*UpdateWorkingMemoryTool) Prompt 

func (t *UpdateWorkingMemoryTool) Prompt() string

type WorkingMemoryMiddleware 

type WorkingMemoryMiddleware struct {
	*BaseMiddleware
	// contains filtered or unexported fields
}

WorkingMemoryMiddleware Working Memory 中间件 功能： 1. 根据 threadID/resourceID 加载 Working Memory 2. 将 Working Memory 内容注入到 system prompt 3. 提供 update_working_memory 工具

func NewWorkingMemoryMiddleware 

func NewWorkingMemoryMiddleware(config *WorkingMemoryMiddlewareConfig) (*WorkingMemoryMiddleware, error)

NewWorkingMemoryMiddleware 创建 Working Memory 中间件

func (*WorkingMemoryMiddleware) GetConfig 

func (m *WorkingMemoryMiddleware) GetConfig() map[string]any

GetConfig 获取配置信息

func (*WorkingMemoryMiddleware) GetManager 

func (m *WorkingMemoryMiddleware) GetManager() *memory.WorkingMemoryManager

GetManager 获取 Working Memory 管理器（用于测试或高级用法）

func (*WorkingMemoryMiddleware) Tools 

func (m *WorkingMemoryMiddleware) Tools() []tools.Tool

Tools 返回 Working Memory 相关工具

func (*WorkingMemoryMiddleware) WrapModelCall 

func (m *WorkingMemoryMiddleware) WrapModelCall(ctx context.Context, req *ModelRequest, handler ModelCallHandler) (*ModelResponse, error)

WrapModelCall 包装模型调用，注入 Working Memory 到 system prompt

type WorkingMemoryMiddlewareConfig 

type WorkingMemoryMiddlewareConfig struct {
	Backend              backends.BackendProtocol  // 存储后端
	BasePath             string                    // 存储根路径，默认 "/working_memory/"
	Scope                memory.WorkingMemoryScope // "thread" | "resource"
	Schema               *memory.JSONSchema        // 可选的 JSON Schema
	Template             string                    // 可选的 Markdown 模板
	TTL                  time.Duration             // 可选的过期时间（0 表示不过期）
	SystemPromptTemplate string                    // 可选，自定义 system prompt 模板
	Experimental         bool                      // 是否启用实验性功能
}

WorkingMemoryMiddlewareConfig 配置

type WriteTodosTool 

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

WriteTodosTool write_todos 工具实现

func (*WriteTodosTool) Description 

func (t *WriteTodosTool) Description() string

func (*WriteTodosTool) Execute 

func (t *WriteTodosTool) Execute(ctx context.Context, input map[string]any, tc *tools.ToolContext) (any, error)

func (*WriteTodosTool) InputSchema 

func (t *WriteTodosTool) InputSchema() map[string]any

func (*WriteTodosTool) Name 

func (t *WriteTodosTool) Name() string

func (*WriteTodosTool) Prompt 

func (t *WriteTodosTool) Prompt() string

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.