ZhenYi/gitdataai
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

feat(agent): inject project context and sender info into AI chat messages

Browse Source
This commit is contained in:
ZhenYi 2026-04-19 01:04:19 +08:00
parent b740e2884d
commit 66006d842e
1 changed files with 32 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
libs/agent/chat/service.rs
View File

@ -559,6 +559,38 @@ impl ChatService {
messages.push(mem.to_system_message());
}
// Inject project context so the AI knows which project it is operating in.
let project_info = format!(
"Current Project:\n{}\nDescription: {}\nPublic: {}",
request.project.display_name,
request.project.description.as_deref().unwrap_or("(none)"),
if request.project.is_public { "yes" } else { "no" }
);
messages.push(ChatCompletionRequestMessage::System(
ChatCompletionRequestSystemMessage {
content: async_openai::types::chat::ChatCompletionRequestSystemMessageContent::Text(project_info),
..Default::default()
},
));
// Inject sender info so the AI knows who is asking.
let mut sender_parts = vec![format!("**Sender:** {}", request.sender.username)];
if let Some(ref display_name) = request.sender.display_name {
sender_parts.push(display_name.clone());
}
if let Some(ref org) = request.sender.organization {
sender_parts.push(format!("({})", org));
}
let sender_display = sender_parts.join(" ");
messages.push(ChatCompletionRequestMessage::System(
ChatCompletionRequestSystemMessage {
content: async_openai::types::chat::ChatCompletionRequestSystemMessageContent::Text(
format!("The person sending the next message:\n{}", sender_display),
),
..Default::default()
},
));
messages.push(ChatCompletionRequestMessage::User(
ChatCompletionRequestUserMessage {
content: async_openai::types::chat::ChatCompletionRequestUserMessageContent::Text(