56 lines
2.0 KiB
Rust
56 lines
2.0 KiB
Rust
|
use async_openai::types::{
|
||
|
|
ChatCompletionRequestAssistantMessageArgs, ChatCompletionRequestMessage,
|
||
|
|
ChatCompletionRequestSystemMessageArgs, ChatCompletionRequestUserMessageArgs,
|
||
|
|
};
|
||
|
|
|
||
|
|
use crate::conversation::llm::{Author as LLMAuthor, Message as LLMMessage};
|
||
|
|
|
||
|
|
pub fn convert_llm_messages_to_openai_messages(
|
||
|
|
conversation_messages: Vec<LLMMessage>,
|
||
|
|
) -> Vec<ChatCompletionRequestMessage> {
|
||
|
|
let mut openai_conversation_messages: Vec<ChatCompletionRequestMessage> =
|
||
|
|
Vec::with_capacity(conversation_messages.len());
|
||
|
|
|
||
|
|
for message in conversation_messages {
|
||
|
|
openai_conversation_messages.push(convert_llm_message_to_openai_message(message));
|
||
|
|
}
|
||
|
|
|
||
|
|
openai_conversation_messages
|
||
|
|
}
|
||
|
|
|
||
|
|
fn convert_llm_message_to_openai_message(llm_message: LLMMessage) -> ChatCompletionRequestMessage {
|
||
|
|
match llm_message.author {
|
||
|
|
LLMAuthor::Prompt => ChatCompletionRequestSystemMessageArgs::default()
|
||
|
|
.content(llm_message.message_text)
|
||
|
|
.build()
|
||
|
|
.expect("Failed building OpenAI system message")
|
||
|
|
.into(),
|
||
|
|
LLMAuthor::Assistant => ChatCompletionRequestAssistantMessageArgs::default()
|
||
|
|
.content(llm_message.message_text)
|
||
|
|
.build()
|
||
|
|
.expect("Failed building OpenAI assistant message")
|
||
|
|
.into(),
|
||
|
|
LLMAuthor::User => ChatCompletionRequestUserMessageArgs::default()
|
||
|
|
.content(llm_message.message_text)
|
||
|
|
.build()
|
||
|
|
.expect("Failed building OpenAI user message")
|
||
|
|
.into(),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub(super) fn convert_string_to_enum<T>(value: &str) -> Result<T, String>
|
||
|
|
where
|
||
|
|
T: serde::de::DeserializeOwned,
|
||
|
|
{
|
||
|
|
// This is a hacky way to construct an enum from the string we have.
|
||
|
|
let enum_result: serde_json::Result<T> = serde_json::from_str(&format!("\"{}\"", value));
|
||
|
|
match enum_result {
|
||
|
|
Ok(enum_result) => Ok(enum_result),
|
||
|
|
Err(err) => {
|
||
|
|
tracing::debug!(?err, "Failed to parse into enum");
|
||
|
|
|
||
|
|
Err(format!("The value ({}) is not supported.", value))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|