selfware 0.6.1

Your personal AI workshop — software you own, software that lasts
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

use super::*;
use std::collections::VecDeque;
use std::sync::Mutex;

pub struct MockLlmClient {
    responses: Mutex<VecDeque<ChatResponse>>,
}

impl MockLlmClient {
    /// Create a new mock client with an empty response queue.
    pub fn new() -> Self {
        Self {
            responses: Mutex::new(VecDeque::new()),
        }
    }

    /// Create a mock client pre-loaded with the given responses.
    ///
    /// Responses are returned in FIFO order by `chat()`.
    pub fn with_responses(responses: Vec<ChatResponse>) -> Self {
        Self {
            responses: Mutex::new(VecDeque::from(responses)),
        }
    }
}

impl Default for MockLlmClient {
    fn default() -> Self {
        Self::new()
    }
}

#[async_trait]
impl LlmClient for MockLlmClient {
    async fn chat(
        &self,
        _messages: Vec<Message>,
        _tools: Option<Vec<ToolDefinition>>,
        _thinking: ThinkingMode,
    ) -> anyhow::Result<ChatResponse> {
        let mut queue = self
            .responses
            .lock()
            .map_err(|_| anyhow::anyhow!("Mock responses mutex poisoned"))?;
        queue
            .pop_front()
            .ok_or_else(|| anyhow::anyhow!("No more mock responses"))
    }

    async fn chat_stream(
        &self,
        _messages: Vec<Message>,
        _tools: Option<Vec<ToolDefinition>>,
        _thinking: ThinkingMode,
    ) -> anyhow::Result<StreamingResponse> {
        Err(anyhow::anyhow!("Streaming not supported in mock client"))
    }
}

#[cfg(test)]
#[path = "../../tests/unit/api/mock/mod_test.rs"]
mod tests;