outlined when truncation is taking place in the prompt

KCaverly created 2 years ago

Change summary

crates/ai/src/templates/file_context.rs | 4 ++++
crates/ai/src/templates/generate.rs     | 3 ++-
crates/assistant/src/prompts.rs         | 2 --
3 files changed, 6 insertions(+), 3 deletions(-)

Detailed changes

crates/ai/src/templates/file_context.rs

@@ -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);

crates/ai/src/templates/generate.rs

@@ -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!(

crates/assistant/src/prompts.rs

@@ -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)
 }