elicitation 0.10.0

Conversational elicitation of strongly-typed Rust values via MCP
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

//! `reqwest::Response` elicitation (Survey pattern).
//!
//! Elicits a status code, body, and URL, then constructs a mock `reqwest::Response`
//! via `http::Response::builder().into()`. Invaluable for agents constructing
//! mock responses in tests without a live HTTP server.

use crate::{
    ElicitCommunicator, ElicitIntrospect, ElicitResult, Elicitation, ElicitationPattern,
    PatternDetails, Prompt, TypeMetadata,
};
use bytes::Bytes;
use reqwest::Response;
use reqwest::ResponseBuilderExt;
use url::Url;

crate::default_style!(Response => ResponseStyle);

impl Prompt for Response {
    fn prompt() -> Option<&'static str> {
        Some(
            "Construct a mock HTTP response — provide a status code, optional body, \
             and optional URL. Useful for testing and agent simulations.",
        )
    }
}

impl Elicitation for Response {
    type Style = ResponseStyle;

    #[tracing::instrument(skip(communicator))]
    async fn elicit<C: ElicitCommunicator>(communicator: &C) -> ElicitResult<Self> {
        use crate::{ElicitError, ElicitErrorKind};

        tracing::debug!("Eliciting reqwest::Response (mock)");

        let status = reqwest::StatusCode::elicit(communicator).await?;
        tracing::debug!(%status, "Status code selected");

        let body_text = String::elicit(communicator).await?;
        tracing::debug!(body_len = body_text.len(), "Body elicited");

        let url = Url::elicit(communicator).await?;
        tracing::debug!(%url, "URL selected");

        let response = http::Response::builder()
            .status(status)
            .url(url)
            .body(Bytes::from(body_text))
            .map_err(|e| {
                ElicitError::new(ElicitErrorKind::ParseError(format!(
                    "Failed to build mock response: {e}"
                )))
            })?;

        Ok(Response::from(response))
    }

    fn kani_proof() -> proc_macro2::TokenStream {
        crate::verification::proof_helpers::kani_trusted_opaque("response")
    }

    fn verus_proof() -> proc_macro2::TokenStream {
        crate::verification::proof_helpers::verus_trusted_opaque("response")
    }

    fn creusot_proof() -> proc_macro2::TokenStream {
        crate::verification::proof_helpers::creusot_trusted_opaque("response")
    }
}

impl ElicitIntrospect for Response {
    fn pattern() -> ElicitationPattern {
        ElicitationPattern::Survey
    }

    fn metadata() -> TypeMetadata {
        TypeMetadata {
            type_name: "reqwest::Response",
            description: Self::prompt(),
            details: PatternDetails::Survey {
                fields: vec![
                    crate::FieldInfo {
                        name: "status",
                        type_name: "reqwest::StatusCode",
                        prompt: Some("HTTP status code (e.g. 200, 404, 500)"),
                    },
                    crate::FieldInfo {
                        name: "body",
                        type_name: "String",
                        prompt: Some("Response body text (empty for no body)"),
                    },
                    crate::FieldInfo {
                        name: "url",
                        type_name: "url::Url",
                        prompt: Some("Response URL"),
                    },
                ],
            },
        }
    }
}