sugars_llm 0.2.0

LLM integration and AI agent builder utilities
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

use serde::{Deserialize, Serialize};
use crate::async_task::AsyncTask;
use crate::domain::{Message, MessageRole};
use crate::domain::completion::CompletionModel;
use crate::domain::agent::Agent;

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Prompt {
    pub content: String,
    #[serde(default = "default_role")]
    pub role: MessageRole,
}

fn default_role() -> MessageRole {
    MessageRole::User
}

impl Prompt {
    pub fn new(content: impl Into<String>) -> Self {
        Prompt {
            content: content.into(),
            role: MessageRole::User,
        }
    }
    
    pub fn content(&self) -> &str {
        &self.content
    }
}

pub struct PromptBuilder {
    content: String,
}

impl Prompt {
    // Semantic entry point
    pub fn ask(content: impl Into<String>) -> PromptBuilder {
        PromptBuilder {
            content: content.into(),
        }
    }
}

impl Into<String> for Prompt {
    fn into(self) -> String {
        self.content
    }
}

impl Into<Prompt> for PromptBuilder {
    fn into(self) -> Prompt {
        Prompt::new(self.content)
    }
}