Trait SecretaryBehavior

Source

pub trait SecretaryBehavior: Send + Sync {
    type Input: SecretaryInput;
    type Output: SecretaryOutput;
    type State: Send + Sync + 'static;

    // Required methods
    fn handle_input<'life0, 'life1, 'async_trait>(
        &'life0 self,
        input: Self::Input,
        ctx: &'life1 mut SecretaryContext<Self::State>,
    ) -> Pin<Box<dyn Future<Output = Result<Vec<Self::Output>, Error>> + Send + 'async_trait>>
       where 'life0: 'async_trait,
             'life1: 'async_trait,
             Self: 'async_trait;
    fn initial_state(&self) -> Self::State;

    // Provided methods
    fn welcome_message(&self) -> Option<Self::Output> { ... }
    fn periodic_check<'life0, 'life1, 'async_trait>(
        &'life0 self,
        _ctx: &'life1 mut SecretaryContext<Self::State>,
    ) -> Pin<Box<dyn Future<Output = Result<Vec<Self::Output>, Error>> + Send + 'async_trait>>
       where 'life0: 'async_trait,
             'life1: 'async_trait,
             Self: 'async_trait { ... }
    fn on_disconnect<'life0, 'life1, 'async_trait>(
        &'life0 self,
        _ctx: &'life1 mut SecretaryContext<Self::State>,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where 'life0: 'async_trait,
             'life1: 'async_trait,
             Self: 'async_trait { ... }
    fn handle_error(&self, _error: &Error) -> Option<Self::Output> { ... }
}

Expand description

秘书行为Trait - 核心抽象

这是框架最核心的trait，定义了秘书如何响应用户输入。开发者需要实现这个trait来创建自定义的秘书。

§类型参数

Input: 秘书接受的输入类型
Output: 秘书产生的输出类型
State: 秘书状态类型

§示例

struct MySecretary {
    name: String,
    llm: Arc<dyn LLMProvider>,
}

#[async_trait]
impl SecretaryBehavior for MySecretary {
    type Input = MyInput;
    type Output = MyOutput;
    type State = MyState;

    async fn handle_input(
        &self,
        input: Self::Input,
        ctx: &mut SecretaryContext<Self::State>,
    ) -> anyhow::Result<Vec<Self::Output>> {
        match input {
            MyInput::TextMessage(text) => {
                // 处理文本消息
                Ok(vec![MyOutput::TextReply(format!("收到: {}", text))])
            }
            _ => Ok(vec![]),
        }
    }

    fn welcome_message(&self) -> Option<Self::Output> {
        Some(MyOutput::TextReply(format!("你好，我是{}!", self.name)))
    }

    fn initial_state(&self) -> Self::State {
        MyState::new()
    }
}