rucora-core 0.2.0

Core abstractions and types for the rucora LLM agent framework
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

//! 共享测试工具——提供简单的 MockProvider 用于 Agent 单元测试

use async_trait::async_trait;
use futures_util::stream::BoxStream;
use crate::error::ProviderError;
use crate::provider::LlmProvider;
use crate::provider::types::{ChatMessage, ChatRequest, ChatResponse, ChatStreamChunk, Role};

/// 极简 MockProvider，返回固定 "Mock response" 文本，无工具调用，无流式数据。
pub struct MockProvider;

#[async_trait]
impl LlmProvider for MockProvider {
    async fn chat(&self, _request: ChatRequest) -> Result<ChatResponse, ProviderError> {
        Ok(ChatResponse {
            message: ChatMessage {
                role: Role::Assistant,
                content: "Mock response".to_string(),
                name: None,
            },
            tool_calls: vec![],
            usage: None,
            finish_reason: None,
        })
    }

    fn stream_chat(
        &self,
        _request: ChatRequest,
    ) -> Result<BoxStream<'static, Result<ChatStreamChunk, ProviderError>>, ProviderError>
    {
        Ok(Box::pin(futures_util::stream::empty()))
    }
}