outlined when truncation is taking place in the prompt

This commit is contained in:
KCaverly
2023-10-19 14:33:52 -04:00
parent 178a84bcf6
commit 19c2df4822
3 changed files with 6 additions and 3 deletions
+4
View File
@@ -131,6 +131,10 @@ impl PromptTemplate for FileContext {
)?;
writeln!(prompt, "```{language_name}\n{context}\n```").unwrap();
if truncated {
writeln!(prompt, "Note the content has been truncated and only represents a portion of the file.").unwrap();
}
if let Some(selected_range) = &args.selected_range {
let start = selected_range.start.to_offset(buffer);
let end = selected_range.end.to_offset(buffer);
+2 -1
View File
@@ -50,7 +50,8 @@ impl PromptTemplate for GenerateInlineContent {
.unwrap();
} else {
writeln!(prompt, "Modify the user's selected {content_type} based upon the users prompt: '{user_prompt}'").unwrap();
writeln!(prompt, "You MUST reply with only the adjusted {content_type} (within the '<|START|' and '|END|>' spans), not the entire file.").unwrap();
writeln!(prompt, "You must reply with only the adjusted {content_type} (within the '<|START|' and '|END|>' spans) not the entire file.").unwrap();
writeln!(prompt, "Double check that you only return code and not the '<|START|' and '|END|'> spans").unwrap();
}
} else {
writeln!(
-2
View File
@@ -166,8 +166,6 @@ pub fn generate_content_prompt(
let chain = PromptChain::new(args, templates);
let (prompt, _) = chain.generate(true)?;
println!("PROMPT: {:?}", &prompt);
anyhow::Ok(prompt)
}