mirrors/zed
1
0
Fork 0
mirror of https://github.com/zed-industries/zed.git synced 2024-12-24 17:28:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

revert change to return only the text and inside return all text inside markdown blocks

Browse source
This commit is contained in:
KCaverly 2023-10-18 10:35:11 -04:00
parent 13c7bbbac6
commit ed8a2c8793
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
crates/assistant/src/prompts.rs
View file

@ -243,7 +243,7 @@ pub fn generate_content_prompt(
} }
prompts.push("Never make remarks about the output.".to_string()); prompts.push("Never make remarks about the output.".to_string());
prompts.push("Do not return any text, except the generated code.".to_string()); prompts.push("Do not return any text, except the generated code.".to_string());
prompts.push("Do not wrap your text in a Markdown block".to_string()); prompts.push("Always wrap your code in a Markdown block".to_string());
let current_messages = [ChatCompletionRequestMessage { let current_messages = [ChatCompletionRequestMessage {
role: "user".to_string(), role: "user".to_string(),
@ -256,7 +256,11 @@ pub fn generate_content_prompt(
tiktoken_rs::num_tokens_from_messages(model, &current_messages) tiktoken_rs::num_tokens_from_messages(model, &current_messages)
{ {
let max_token_count = tiktoken_rs::model::get_context_size(model); let max_token_count = tiktoken_rs::model::get_context_size(model);
let intermediate_token_count = max_token_count - current_token_count; let intermediate_token_count = if max_token_count > current_token_count {
max_token_count - current_token_count
} else {
0
};
if intermediate_token_count < RESERVED_TOKENS_FOR_GENERATION { if intermediate_token_count < RESERVED_TOKENS_FOR_GENERATION {
0 0