cf-mini-chat 0.1.31

Mini-chat module: multi-tenant AI chat
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
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100

use async_trait::async_trait;
use modkit_macros::domain_model;
use tracing::info;

use mini_chat_sdk::{
    MiniChatAuditPluginClientV1, MiniChatAuditPluginError, TurnAuditEvent, TurnDeleteAuditEvent,
    TurnEditAuditEvent, TurnRetryAuditEvent,
};

/// Service for the static audit plugin.
///
/// When `enabled` is `false`, all emit methods return `Ok(())` immediately
/// without writing to the log.
#[domain_model]
pub struct Service {
    pub enabled: bool,
}

#[async_trait]
impl MiniChatAuditPluginClientV1 for Service {
    async fn emit_turn_audit(&self, event: TurnAuditEvent) -> Result<(), MiniChatAuditPluginError> {
        if !self.enabled {
            return Ok(());
        }
        info!(
            event_type = %event.event_type,
            tenant_id = %event.tenant_id,
            user_id = %event.user_id,
            chat_id = %event.chat_id,
            turn_id = %event.turn_id,
            request_id = %event.request_id,
            selected_model = %event.selected_model,
            effective_model = %event.effective_model,
            input_tokens = event.usage.input_tokens,
            output_tokens = event.usage.output_tokens,
            cache_read_input_tokens = event.usage.cache_read_input_tokens,
            cache_write_input_tokens = event.usage.cache_write_input_tokens,
            reasoning_tokens = event.usage.reasoning_tokens,
            error_code = event.error_code.as_deref(),
            "audit: turn event"
        );
        Ok(())
    }

    async fn emit_turn_retry_audit(
        &self,
        event: TurnRetryAuditEvent,
    ) -> Result<(), MiniChatAuditPluginError> {
        if !self.enabled {
            return Ok(());
        }
        info!(
            event_type = %event.event_type,
            tenant_id = %event.tenant_id,
            actor_user_id = %event.actor_user_id,
            chat_id = %event.chat_id,
            original_request_id = %event.original_request_id,
            new_request_id = %event.new_request_id,
            "audit: turn retry event"
        );
        Ok(())
    }

    async fn emit_turn_edit_audit(
        &self,
        event: TurnEditAuditEvent,
    ) -> Result<(), MiniChatAuditPluginError> {
        if !self.enabled {
            return Ok(());
        }
        info!(
            event_type = %event.event_type,
            tenant_id = %event.tenant_id,
            actor_user_id = %event.actor_user_id,
            chat_id = %event.chat_id,
            original_request_id = %event.original_request_id,
            new_request_id = %event.new_request_id,
            "audit: turn edit event"
        );
        Ok(())
    }

    async fn emit_turn_delete_audit(
        &self,
        event: TurnDeleteAuditEvent,
    ) -> Result<(), MiniChatAuditPluginError> {
        if !self.enabled {
            return Ok(());
        }
        info!(
            event_type = %event.event_type,
            tenant_id = %event.tenant_id,
            actor_user_id = %event.actor_user_id,
            chat_id = %event.chat_id,
            request_id = %event.request_id,
            "audit: turn delete event"
        );
        Ok(())
    }
}