Skip to content

Commit de3fa6f

Browse files
committed
fix(rmcp): import base64-only prompt content types under the feature
1 parent 79437f2 commit de3fa6f

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

‎crates/rmcp/src/model/prompt.rs‎

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ use serde::{Deserialize, Serialize};
22

33
use super::{
44
Annotations, ContentBlock, Icon, MetaObject, Role,
5-
content::{AudioContent, EmbeddedResource, ImageContent, TextContent},
5+
content::{EmbeddedResource, TextContent},
66
resource::ResourceContents,
77
};
88

@@ -149,6 +149,8 @@ impl PromptMessage {
149149
) -> Self {
150150
use base64::{Engine, prelude::BASE64_STANDARD};
151151

152+
use super::content::ImageContent;
153+
152154
let base64 = BASE64_STANDARD.encode(data);
153155
Self {
154156
role,
@@ -171,6 +173,8 @@ impl PromptMessage {
171173
) -> Self {
172174
use base64::{Engine, prelude::BASE64_STANDARD};
173175

176+
use super::content::AudioContent;
177+
174178
let base64 = BASE64_STANDARD.encode(data);
175179
Self {
176180
role,
@@ -244,6 +248,7 @@ mod tests {
244248
use serde_json;
245249

246250
use super::*;
251+
use crate::model::content::{AudioContent, ImageContent};
247252

248253
#[test]
249254
fn test_prompt_message_image_serialization() {

0 commit comments

Comments
 (0)