Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Commitbccedef

Browse files
committed
Update prompts
Signed-off-by: Michael Yuan <michael@secondstate.io>
1 parenta0f739f commitbccedef

File tree

1 file changed

+16
-16
lines changed

1 file changed

+16
-16
lines changed

‎src/github-pr-summary.rs

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,8 @@ use llmservice_flows::{
1414
use std::env;
1515

1616
// The soft character limit of the input context size
17-
// THe codestral has a context length of32k, and we allow8k context here
18-
staticCHAR_SOFT_LIMIT:usize =8192;
17+
// THe codestral has a context length of16k tokens, and we allow16k chars of context here
18+
staticCHAR_SOFT_LIMIT:usize =16384;
1919

2020
#[no_mangle]
2121
#[tokio::main(flavor ="current_thread")]
@@ -89,7 +89,7 @@ async fn handler(payload: EventPayload) {
8989
// return;
9090
// }
9191
// TODO: Makeshift but operational
92-
if body.starts_with("Hello, I am a [code review bot]"){
92+
if body.starts_with("Hello, I am a [PR summary bot]"){
9393
log::info!("Ignore comment via bot");
9494
return;
9595
};
@@ -108,11 +108,11 @@ async fn handler(payload: EventPayload) {
108108
let issues = octo.issues(owner.clone(), repo.clone());
109109
letmut comment_id:CommentId =0u64.into();
110110
if new_commit{
111-
// Find the first "Hello, I am a [code review bot]" comment to update
111+
// Find the first "Hello, I am a [PR summary bot]" comment to update
112112
match issues.list_comments(pull_number).send().await{
113113
Ok(comments) =>{
114114
for cin comments.items{
115-
if c.body.unwrap_or_default().starts_with("Hello, I am a [code review bot]"){
115+
if c.body.unwrap_or_default().starts_with("Hello, I am a [PR summary bot]"){
116116
comment_id = c.id;
117117
break;
118118
}
@@ -125,7 +125,7 @@ async fn handler(payload: EventPayload) {
125125
}
126126
}else{
127127
// PR OPEN or Trigger phrase: create a new comment
128-
match issues.create_comment(pull_number,"Hello, I am a [code review bot](https://github.com/flows-network/github-pr-summary/) on [flows.network](https://flows.network/).\n\nIt could take a few minutes for me to analyze this PR. Relax, grab a cup of coffee and check back later. Thanks!").await{
128+
match issues.create_comment(pull_number,"Hello, I am a [PR summary bot](https://github.com/flows-network/github-pr-summary/) on [flows.network](https://flows.network/).\n\nIt could take a few minutes for me to analyze this PR. Relax, grab a cup of coffee and check back later. Thanks!").await{
129129
Ok(comment) =>{
130130
comment_id = comment.id;
131131
}
@@ -168,22 +168,22 @@ async fn handler(payload: EventPayload) {
168168
}
169169

170170
let chat_id =format!("PR#{pull_number}");
171-
let system =&format!("You are an experienced software developer. You will act as a reviewer for a GitHub Pull Request titled\"{}\".", title);
171+
let system =&format!("You are an experienced software developer. You will act as a reviewer for a GitHub Pull Request titled\"{}\". Please be as concise as possible while being accurate.", title);
172172
letmut lf =LLMServiceFlows::new(&llm_api_endpoint);
173-
lf.set_retry_times(3);
173+
//lf.set_retry_times(3);
174174

175175
letmut reviews:Vec<String> =Vec::new();
176176
letmut reviews_text =String::new();
177177
for(_i, commit)in commits.iter().enumerate(){
178178
let commit_hash =&commit[5..45];
179-
log::debug!("Sending patch toOpenAI: {}", commit_hash);
179+
log::debug!("Sending patch toLLM: {}", commit_hash);
180180
let co =ChatOptions{
181181
model:Some(&llm_model_name),
182182
restart:true,
183183
system_prompt:Some(system),
184184
..Default::default()
185185
};
186-
let question ="The following is a GitHub patch. Please summarize the key changesand identify potential problems. Start with the most important findings.\n\n".to_string() +truncate(commit,CHAR_SOFT_LIMIT);
186+
let question ="The following is a GitHub patch. Please summarize the key changesin concise points. Start with the most important findings.\n\n".to_string() +truncate(commit,CHAR_SOFT_LIMIT);
187187
match lf.chat_completion(&chat_id,&question,&co).await{
188188
Ok(r) =>{
189189
if reviews_text.len() <CHAR_SOFT_LIMIT{
@@ -196,33 +196,33 @@ async fn handler(payload: EventPayload) {
196196
review.push_str(&r.choice);
197197
review.push_str("\n\n");
198198
reviews.push(review);
199-
log::debug!("ReceivedOpenAI resp for patch: {}", commit_hash);
199+
log::debug!("ReceivedLLM resp for patch: {}", commit_hash);
200200
}
201201
Err(e) =>{
202-
log::error!("OpenAI returned an error for commit {commit_hash}: {}", e);
202+
log::error!("LLM returned an error for commit {commit_hash}: {}", e);
203203
}
204204
}
205205
}
206206

207207
letmut resp =String::new();
208-
resp.push_str("Hello, I am a [code review bot](https://github.com/flows-network/github-pr-summary/) on [flows.network](https://flows.network/). Here are my reviews of code commits in this PR.\n\n------\n\n");
208+
resp.push_str("Hello, I am a [PR summary bot](https://github.com/flows-network/github-pr-summary/) on [flows.network](https://flows.network/). Here are my reviews of code commits in this PR.\n\n------\n\n");
209209
if reviews.len() >1{
210-
log::debug!("Sending all reviews toOpenAI for summarization");
210+
log::debug!("Sending all reviews toLLM for summarization");
211211
let co =ChatOptions{
212212
model:Some(&llm_model_name),
213213
restart:true,
214214
system_prompt:Some(system),
215215
..Default::default()
216216
};
217-
let question ="Here is a set of summaries forsoftwaresource code patches. Each summary starts with a ------ line.Please writean overall summary considering all the individual summary. Please present the potential issues and errors first, following by the most important findings, in your summary.\n\n".to_string() +&reviews_text;
217+
let question ="Here is a set of summaries for source code patches in this PR. Each summary starts with a ------ line.Writean overall summary. Present the potential issues and errors first, following by the most important findings, in your summary.\n\n".to_string() +&reviews_text;
218218
match lf.chat_completion(&chat_id,&question,&co).await{
219219
Ok(r) =>{
220220
resp.push_str(&r.choice);
221221
resp.push_str("\n\n## Details\n\n");
222222
log::debug!("Received the overall summary");
223223
}
224224
Err(e) =>{
225-
log::error!("OpenAI returned an error for the overall summary: {}", e);
225+
log::error!("LLM returned an error for the overall summary: {}", e);
226226
}
227227
}
228228
}

0 commit comments

Comments
 (0)
[8]ページ先頭
©2009-2025 Movatter.jp