artifactize-core 0.22.0

The library behind artifactize: Artifacts, evals, reviews, the monitor and review models, and the shared operations every interface calls.
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
101
102
103
104
105
106
107
108
109
110
111
112

//! The MCP server artifactize hosts inside its own process for one Claude Code session.
//! Claude Code reaches it through its control protocol, so no socket or second process is
//! involved; the review runs every tool call itself, with its budgets and audit.

use serde_json::{Value, json};

use super::wire::{McpRequest, Rpc, RpcId};
use crate::tools::{Content, ToolDefinition, ToolResult};

/// The JSON-RPC error code of an unknown method.
const METHOD_NOT_FOUND: i64 = -32601;

/// The offered tools, in MCP's shape.
pub struct Server {
    tools: Vec<Value>,
}

/// What a message for the server needs: a reply now, or a tool run the review makes.
pub enum Answer {
    Reply(Value),
    Call {
        id: RpcId,
        name: String,
        arguments: Value,
    },
}

impl Server {
    pub fn new<'a>(tools: impl Iterator<Item = &'a ToolDefinition>) -> Self {
        Self {
            tools: tools
                .map(|tool| {
                    json!({
                        "name": tool.name,
                        "description": tool.description,
                        "inputSchema": tool.input_schema,
                    })
                })
                .collect(),
        }
    }

    pub fn answer(&self, message: Rpc) -> Answer {
        let request = match message {
            Rpc::Known(request) => request,
            // A notification needs no answer; Claude Code still expects the control reply.
            Rpc::Unknown { id: None, .. } => {
                return Answer::Reply(json!({"jsonrpc": "2.0", "result": {}}));
            }
            Rpc::Unknown {
                id: Some(id),
                method,
            } => {
                return Answer::Reply(json!({
                    "jsonrpc": "2.0",
                    "id": id,
                    "error": {"code": METHOD_NOT_FOUND, "message": format!("Unknown method {method}.")},
                }));
            }
        };
        Answer::Reply(match request {
            McpRequest::Initialize { id, params } => json!({
                "jsonrpc": "2.0",
                "id": id,
                "result": {
                    "protocolVersion": params.protocol_version,
                    "capabilities": {"tools": {}},
                    "serverInfo": {"name": super::command::SERVER, "version": env!("CARGO_PKG_VERSION")},
                },
            }),
            McpRequest::ToolsList { id } => {
                json!({"jsonrpc": "2.0", "id": id, "result": {"tools": self.tools}})
            }
            McpRequest::Ping { id } => json!({"jsonrpc": "2.0", "id": id, "result": {}}),
            McpRequest::ToolsCall { id, params } => {
                return Answer::Call {
                    id,
                    name: params.name,
                    arguments: params.arguments.unwrap_or_else(|| json!({})),
                };
            }
        })
    }
}

/// A tool's result as MCP content: text, JSON as text, and images.
pub fn call_reply(id: &RpcId, result: ToolResult) -> Value {
    let is_error = result.is_error();
    let content: Vec<Value> = result
        .into_content()
        .into_iter()
        .map(|block| match block {
            Content::Text { text } => json!({"type": "text", "text": text}),
            Content::Json { data } => json!({"type": "text", "text": data.to_string()}),
            Content::Image { data, mime_type } => {
                use artifactize_tools::image::Mime;
                let mime = match mime_type {
                    Mime::Png => "image/png",
                    Mime::Jpeg => "image/jpeg",
                    Mime::Webp => "image/webp",
                };
                json!({"type": "image", "data": data, "mimeType": mime})
            }
        })
        .collect();
    json!({"jsonrpc": "2.0", "id": id, "result": {"content": content, "isError": is_error}})
}

/// A tool call refused before it ran.
pub fn call_error(id: &RpcId, message: &str) -> Value {
    call_reply(id, ToolResult::error(message))
}