Struct RichAgentContext 

pub struct RichAgentContext { /* private fields */ }

扩展的 Agent 上下文

提供业务特定的功能：

• 组件输出记录
• 执行指标跟踪
• 委托所有核心功能到 CoreAgentContext

示例

use mofa_foundation::agent::context::RichAgentContext;
use mofa_kernel::agent::context::CoreAgentContext;

let core_ctx = CoreAgentContext::new("execution-123");
let rich_ctx = RichAgentContext::from(core_ctx);

// 业务特定功能
rich_ctx.record_output("llm", serde_json::json!("response")).await;
rich_ctx.increment_component_calls("llm").await;

// 核心功能委托
rich_ctx.set("key", "value").await;

Implementations

impl RichAgentContext

pub fn new(inner: AgentContext) -> RichAgentContext

从 CoreAgentContext 创建 RichAgentContext

pub async fn record_output(&self, component: impl Into<String>, output: Value)

记录组件输出

pub async fn get_outputs(&self) -> Vec<ComponentOutput>

获取所有组件输出

pub async fn increment_component_calls(&self, component: &str)

增加组件调用计数

pub async fn add_tokens(&self, tokens: u64)

增加 Token 使用

pub async fn increment_tool_calls(&self)

增加工具调用计数

pub async fn get_metrics(&self) -> ExecutionMetrics

获取执行指标

pub async fn finish(&self)

结束执行 (记录结束时间)

pub async fn duration_ms(&self) -> u64

获取执行时长 (毫秒)

pub async fn get<T>(&self, key: &str) -> Option<T>

where T: DeserializeOwned,

获取值

pub async fn set<T>(&self, key: &str, value: T)

where T: Serialize,

设置值

pub async fn remove(&self, key: &str) -> Option<Value>

删除值

pub async fn contains(&self, key: &str) -> bool

检查是否存在值

pub async fn keys(&self) -> Vec<String>

获取所有键

pub async fn find<T>(&self, key: &str) -> Option<T>

where T: DeserializeOwned,

从父上下文查找值

pub fn execution_id(&self) -> &str

获取执行 ID

pub fn session_id(&self) -> Option<&str>

获取会话 ID

pub fn parent(&self) -> Option<&Arc<AgentContext>>

获取父上下文

pub fn is_interrupted(&self) -> bool

检查是否被中断

pub fn trigger_interrupt(&self)

触发中断

pub fn clear_interrupt(&self)

清除中断状态

pub fn config(&self) -> &ContextConfig

获取配置

pub async fn emit_event(&self, event: AgentEvent)

发送事件

pub async fn subscribe(&self, event_type: &str) -> Receiver<AgentEvent>

订阅事件

pub fn inner(&self) -> &AgentContext

获取内部核心上下文的引用

Trait Implementations

impl AsRef<AgentContext> for RichAgentContext

fn as_ref(&self) -> &AgentContext

Converts this type into a shared reference of the (usually inferred) input type.

impl Clone for RichAgentContext

fn clone(&self) -> RichAgentContext

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl From<AgentContext> for RichAgentContext

fn from(inner: AgentContext) -> RichAgentContext

Converts to this type from the input type.

impl From<RichAgentContext> for AgentContext

fn from(rich: RichAgentContext) -> AgentContext

Converts to this type from the input type.