diff --git a/src/types/common.rs b/src/types/common.rs index d4e70d6..90a220d 100644 --- a/src/types/common.rs +++ b/src/types/common.rs @@ -24,6 +24,10 @@ impl Content { }) } + pub fn system_prompt>(system_prompt: S) -> Self { + Self::builder().add_text_part(system_prompt).build() + } + pub fn builder() -> ContentBuilder { ContentBuilder::default() } @@ -127,13 +131,13 @@ pub enum PartData { } impl Part { - pub fn from_text(text: String) -> Self { + pub fn from_text>(text: S) -> Self { Self { thought: None, thought_signature: None, part_metadata: None, media_resolution: None, - data: PartData::Text(text), + data: PartData::Text(text.into()), } } }