Skip to content

Commit

Permalink
Merge pull request rustformers#351 from viirya/check_message_prompt
Browse files Browse the repository at this point in the history
Add placeholder check for message prompt
philpax authored Jul 5, 2023
2 parents 8a48238 + 8c4d59d commit a69d953
Showing 1 changed file with 18 additions and 2 deletions.
20 changes: 18 additions & 2 deletions binaries/llm-cli/src/cli_args.rs
Original file line number Diff line number Diff line change
@@ -203,9 +203,25 @@ impl Chat {
}

if let Some(message_prompt_file) = &self.message_prompt_file {
read_prompt_file(message_prompt_file)
read_prompt_file(message_prompt_file).and_then(|prompt| {
prompt
.contains("{{PROMPT}}")
.then_some(prompt)
.ok_or_else(|| {
eyre::eyre!(
"Message prompt file must contain a `{{{{PROMPT}}}}` placeholder, but it does not"
)
})
})
} else if let Some(message_prompt) = &self.message_prompt {
Ok(message_prompt.clone())
message_prompt
.contains("{{PROMPT}}")
.then(|| message_prompt.clone())
.ok_or_else(|| {
eyre::eyre!(
"Message prompt must contain a `{{{{PROMPT}}}}` placeholder, but it does not"
)
})
} else {
eyre::bail!("Must specify either --message-prompt or --message-prompt-file")
}

0 comments on commit a69d953

Please sign in to comment.