chat-prompts 0.1.1

Chat prompt template
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

pub mod belle;
pub mod llama;
pub mod mistral;

use crate::error::Result;
use belle::*;
use endpoints::chat::ChatCompletionRequestMessage;
use llama::*;
use mistral::*;

#[enum_dispatch::enum_dispatch]
pub trait BuildChatPrompt: Send {
    fn build(&self, messages: &mut Vec<ChatCompletionRequestMessage>) -> Result<String>;
}

#[enum_dispatch::enum_dispatch(BuildChatPrompt)]
pub enum ChatPrompt {
    Llama2ChatPrompt,
    MistralInstructPrompt,
    CodeLlamaInstructPrompt,
    BelleLlama2ChatPrompt,
}