zed-industries/zed
1
1
Fork 0
mirror of https://github.com/zed-industries/zed.git synced 2024-12-28 22:24:47 +03:00
Code Issues Packages Projects Releases Wiki Activity

added initial placeholder for truncation without a valid strategy

Browse Source
This commit is contained in:
KCaverly 2023-10-18 16:23:53 -04:00
parent 473067db31
commit 32853c2044
2 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
crates/ai/src/templates/file_context.rs
View File

@ -1,3 +1,4 @@
use anyhow::anyhow;
use language::ToOffset;
use crate::templates::base::PromptArguments;
@ -12,6 +13,12 @@ impl PromptTemplate for FileContext {
args: &PromptArguments,
max_token_length: Option<usize>,
) -> anyhow::Result<(String, usize)> {
if max_token_length.is_some() {
return Err(anyhow!(
"no truncation strategy established for file_context template"
));
}
let mut prompt = String::new();
// Add Initial Preamble

7
crates/ai/src/templates/generate.rs
View File

@ -18,6 +18,12 @@ impl PromptTemplate for GenerateInlineContent {
args: &PromptArguments,
max_token_length: Option<usize>,
) -> anyhow::Result<(String, usize)> {
if max_token_length.is_some() {
return Err(anyhow!(
"no truncation strategy established for generating inline content template"
));
}
let Some(user_prompt) = &args.user_prompt else {
return Err(anyhow!("user prompt not provided"));
};
@ -83,6 +89,7 @@ impl PromptTemplate for GenerateInlineContent {
}
let token_count = args.model.count_tokens(&prompt)?;
anyhow::Ok((prompt, token_count))
}
}