Pick any line to see why it is the way it is: the commit, the pull request and issue it came from, and what the agent was thinking.
| The sidebar asks billing whether it is on, not for the account, so no page waits on the card processor | 1 | //! Messages to agents at work. People steer the agent on a pull request; |
| 2 | //! agents ask each other questions and hand each other work, and answer. | |
| 3 | //! Each reaches its agent at the agent's next step: the sandbox asks for | |
| 4 | //! undelivered messages after each tool call and before it stops. | |
| 5 | ||
| 6 | use g1t_contracts::access::Capability; | |
| 7 | use g1t_contracts::time::rfc3339; | |
| 8 | use g1t_contracts::work::*; | |
| 9 | use g1t_contracts::{FailureCode, Outcome, PrincipalKind, new_id}; | |
| 10 | use g1t_kit::now_ms; | |
| 11 | use serde::Deserialize; | |
| 12 | use worker::Result; | |
| 13 | ||
| 14 | use crate::Work; | |
| 15 | ||
| 16 | /// The longest message an agent is sent. | |
| 17 | const MAX_MESSAGE_CHARS: usize = 4000; | |
| 18 | ||
| 19 | /// Kinds an agent may send another. | |
| 20 | const ASKS: [&str; 2] = ["question", "handoff"]; | |
| 21 | ||
| 22 | /// How long an agent woken to answer holds its pull request: nothing else | |
| 23 | /// starts on it meanwhile. Answering everything lets go sooner. | |
| 24 | const ANSWER_MINUTES: u64 = 20; | |
| 25 | ||
| 26 | #[derive(Deserialize)] | |
| 27 | struct MessageRow { | |
| 28 | id: String, | |
| 29 | #[allow(dead_code)] | |
| 30 | pull_id: String, | |
| 31 | author_name: String, | |
| 32 | body: String, | |
| 33 | created_at: String, | |
| 34 | delivered_at: Option<String>, | |
| 35 | kind: String, | |
| 36 | from_number: Option<u32>, | |
| 37 | to_number: Option<u32>, | |
| 38 | answer: Option<String>, | |
| 39 | declined: u32, | |
| 40 | } | |
| 41 | ||
| 42 | impl From<MessageRow> for AgentMessage { | |
| 43 | fn from(row: MessageRow) -> Self { | |
| 44 | AgentMessage { | |
| 45 | id: row.id, | |
| 46 | author: row.author_name, | |
| 47 | body: row.body, | |
| 48 | created_at: row.created_at, | |
| 49 | delivered_at: row.delivered_at, | |
| 50 | kind: row.kind, | |
| 51 | from_number: row.from_number, | |
| 52 | to_number: row.to_number.unwrap_or_default(), | |
| 53 | answer: row.answer, | |
| 54 | declined: row.declined != 0, | |
| 55 | hint: None, | |
| 56 | } | |
| 57 | } | |
| 58 | } | |
| 59 | ||
| 60 | /// Who sent a message, in a sentence's words. | |
| 61 | fn sender(message: &AgentMessage) -> String { | |
| 62 | match message.from_number { | |
| 63 | Some(number) => format!("the agent on #{number}"), | |
| 64 | None => message.author.clone(), | |
| 65 | } | |
| 66 | } | |
| 67 | ||
| 68 | impl Work { | |
| 69 | pub(crate) async fn locate_pull(&self, a: LocatePullArgs) -> Result<Outcome<LocatedPull>> { | |
| 70 | let missing = || Outcome::fail(FailureCode::NotFound, "Pull request not found."); | |
| 71 | let Some(pull) = self.pull_by_id(&a.id).await? else { | |
| 72 | return Ok(missing()); | |
| 73 | }; | |
| 74 | let repo: Outcome<g1t_contracts::repos::Repo> = g1t_kit::call( | |
| 75 | &self.repos, | |
| 76 | "get_by_id", | |
| 77 | &g1t_contracts::repos::GetByIdArgs { | |
| 78 | id: pull.repo_id.clone(), | |
| 79 | viewer: a.viewer, | |
| 80 | }, | |
| 81 | ) | |
| 82 | .await?; | |
| 83 | let Outcome::Ok(repo) = repo else { | |
| 84 | return Ok(missing()); | |
| 85 | }; | |
| 86 | Ok(Outcome::Ok(LocatedPull { | |
| 87 | repo: g1t_contracts::repos::RepoPath { | |
| 88 | namespace: repo.namespace, | |
| 89 | name: repo.name, | |
| 90 | }, | |
| 91 | number: pull.number, | |
| 92 | title: pull.title, | |
| 93 | status: pull.status, | |
| 94 | })) | |
| 95 | } | |
| 96 | ||
| 97 | /// Every message sent to the agent on a pull request, oldest first. | |
| 98 | pub(crate) async fn messages(&self, pull_id: &str) -> Result<Vec<AgentMessage>> { | |
| 99 | if let Some(found) = self.prefetched_pull(pull_id) { | |
| 100 | return Ok(found | |
| 101 | .rows::<MessageRow>(crate::prefetch::Slot::Messages)? | |
| 102 | .into_iter() | |
| 103 | .map(AgentMessage::from) | |
| 104 | .collect()); | |
| 105 | } | |
| 106 | Ok(self | |
| 107 | .db | |
| 108 | .prepare("SELECT * FROM agent_messages WHERE pull_id = ? ORDER BY created_at, id") | |
| 109 | .bind(&[pull_id.into()])? | |
| 110 | .all() | |
| 111 | .await? | |
| 112 | .results::<MessageRow>()? | |
| 113 | .into_iter() | |
| 114 | .map(AgentMessage::from) | |
| 115 | .collect()) | |
| 116 | } | |
| 117 | ||
| 118 | pub(crate) async fn message_agent(&self, a: MessageAgentArgs) -> Result<Outcome<AgentMessage>> { | |
| 119 | let viewer = Some(a.actor.clone()); | |
| 120 | let (repo, pull) = match self.pull_at(&a.repo, a.number, &viewer).await? { | |
| 121 | Outcome::Ok(found) => found, | |
| 122 | Outcome::Fail(failure) => return Ok(Outcome::Fail(failure)), | |
| 123 | }; | |
| 124 | if let Outcome::Fail(failure) = crate::retired::writable(&repo) { | |
| 125 | return Ok(Outcome::Fail(failure)); | |
| 126 | } | |
| 127 | let from_agent = a.actor.kind == PrincipalKind::Agent; | |
| 128 | let kind = match (&a.kind, from_agent) { | |
| 129 | (Some(kind), true) if ASKS.contains(&kind.as_str()) => kind.clone(), | |
| 130 | (None, true) => "question".to_owned(), | |
| 131 | (_, true) => { | |
| 132 | return Ok(Outcome::fail( | |
| 133 | FailureCode::Invalid, | |
| 134 | "An agent sends a question or a handoff.", | |
| 135 | )); | |
| 136 | } | |
| 137 | (_, false) => "message".to_owned(), | |
| 138 | }; | |
| 139 | if from_agent && a.from_number.is_none() { | |
| 140 | return Ok(Outcome::fail( | |
| 141 | FailureCode::Invalid, | |
| 142 | "Say which pull request you are working on, as from_number; it is where the answer goes.", | |
| 143 | )); | |
| 144 | } | |
| 145 | if !a.actor.verified { | |
| 146 | return Ok(Outcome::fail(FailureCode::Forbidden, crate::UNVERIFIED)); | |
| 147 | } | |
| 148 | // Its owner (whoever asked g1t for it, or its author) may always | |
| 149 | // steer it; anyone else puts compute to work. | |
| 150 | if !pull.is_owned_by(&a.actor.id) | |
| 151 | && let Outcome::Fail(failure) = crate::allowed(Some(&a.actor), &repo, Capability::Run) | |
| 152 | { | |
| 153 | return Ok(Outcome::Fail(failure)); | |
| 154 | } | |
| 155 | if !pull.status.is_active() { | |
| 156 | return Ok(Outcome::fail( | |
| 157 | FailureCode::Conflict, | |
| 158 | "This pull request is no longer being worked on.", | |
| 159 | )); | |
| 160 | } | |
| 161 | let body = a.body.trim(); | |
| 162 | if body.is_empty() { | |
| 163 | return Ok(Outcome::fail(FailureCode::Invalid, "Write a message.")); | |
| 164 | } | |
| 165 | let body: String = body.chars().take(MAX_MESSAGE_CHARS).collect(); | |
| 166 | // An agent may name its issue rather than its pull request: the | |
| 167 | // answer goes to the issue's pull request that is still open. | |
| 168 | let from_number = match (from_agent, a.from_number) { | |
| 169 | (true, Some(from)) => match self.pull(&repo.id, from).await? { | |
| 170 | Some(_) => Some(from), | |
| 171 | None => self.db | |
| 172 | .prepare( | |
| 173 | "SELECT number AS value FROM pulls | |
| 174 | WHERE repo_id = ? AND issue_number = ? AND status IN ('draft', 'open') | |
| 175 | ORDER BY number DESC LIMIT 1", | |
| 176 | ) | |
| 177 | .bind(&[repo.id.as_str().into(), from.into()])? | |
| 178 | .first::<u32>(Some("value")) | |
| 179 | .await? | |
| 180 | .or(Some(from)), | |
| 181 | }, | |
| 182 | _ => None, | |
| 183 | }; | |
| 184 | if from_agent && from_number == Some(pull.number) { | |
| 185 | return Ok(Outcome::fail(FailureCode::Invalid, "That is your own pull request.")); | |
| 186 | } | |
| 187 | // Whether the agent asked is at work now, to read it soon. | |
| 188 | let at_work = pull.status == PullStatus::Draft | |
| 189 | || self | |
| 190 | .db | |
| 191 | .prepare( | |
| 192 | "SELECT 1 AS value FROM pulls | |
| 193 | WHERE id = ? AND working_on = 'revision' AND working_until > ?", | |
| 194 | ) | |
| 195 | .bind(&[pull.id.as_str().into(), rfc3339(now_ms()).into()])? | |
| 196 | .first::<u32>(Some("value")) | |
| 197 | .await? | |
| 198 | .is_some(); | |
| 199 | let now = now_ms(); | |
| 200 | let message = AgentMessage { | |
| 201 | id: new_id("msg", now), | |
| 202 | author: a.actor.username.clone(), | |
| 203 | body, | |
| 204 | created_at: rfc3339(now), | |
| 205 | delivered_at: None, | |
| 206 | kind, | |
| 207 | from_number, | |
| 208 | to_number: pull.number, | |
| 209 | answer: None, | |
| 210 | declined: false, | |
| 211 | hint: None, | |
| 212 | }; | |
| 213 | self.insert_message(&repo.id, &pull.id, &a.actor.id, &message).await?; | |
| 214 | let mut message = message; | |
| 215 | if from_agent && !at_work { | |
| 216 | // An open pull request that g1t has not stopped on: its agent is | |
| 217 | // woken to answer (see `wake_for_messages`). | |
| 218 | let wakeable = pull.status == PullStatus::Open | |
| 219 | && self | |
| 220 | .db | |
| 221 | .prepare("SELECT 1 AS value FROM pulls WHERE id = ? AND stalled IS NULL") | |
| 222 | .bind(&[pull.id.as_str().into()])? | |
| 223 | .first::<u32>(Some("value")) | |
| 224 | .await? | |
| 225 | .is_some(); | |
| 226 | message.hint = Some(if wakeable { | |
| 227 | self.publish("agent.asked", &repo.id, &a.actor, Self::pull_event(&pull)).await?; | |
| 228 | format!( | |
| 229 | "The agent on #{} was not at work, so g1t is waking it to answer; the answer reaches you at a later step. Its change is there to read meanwhile: get_pull_request and get_pull_request_changes on #{}.", | |
| 230 | pull.number, pull.number | |
| 231 | ) | |
| 232 | } else { | |
| 233 | format!( | |
| 234 | "The agent on #{} is not at work right now, so it will not answer soon. Its change is there to read: use get_pull_request and get_pull_request_changes on #{}, and decide from that.", | |
| 235 | pull.number, pull.number | |
| 236 | ) | |
| 237 | }); | |
| 238 | } | |
| 239 | let said = match (message.kind.as_str(), message.from_number) { | |
| 240 | ("question", Some(from)) => format!("was asked a question by the agent on #{from}"), | |
| 241 | ("handoff", Some(from)) => format!("was handed work by the agent on #{from}"), | |
| 242 | _ => "sent the agent a message".to_owned(), | |
| 243 | }; | |
| 244 | self.note(&repo.id, pull.number, (a.actor.id.as_str(), a.actor.username.as_str()), &said) | |
| 245 | .await?; | |
| 246 | Ok(Outcome::Ok(message)) | |
| 247 | } | |
| 248 | ||
| 249 | async fn insert_message( | |
| 250 | &self, | |
| 251 | repo_id: &str, | |
| 252 | pull_id: &str, | |
| 253 | author_id: &str, | |
| 254 | message: &AgentMessage, | |
| 255 | ) -> Result<()> { | |
| 256 | self.db | |
| 257 | .prepare( | |
| 258 | "INSERT INTO agent_messages | |
| 259 | (id, pull_id, repo_id, author_id, author_name, body, created_at, kind, | |
| 260 | from_number, to_number) | |
| 261 | VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", | |
| 262 | ) | |
| 263 | .bind(&[ | |
| 264 | message.id.as_str().into(), | |
| 265 | pull_id.into(), | |
| 266 | repo_id.into(), | |
| 267 | author_id.into(), | |
| 268 | message.author.as_str().into(), | |
| 269 | message.body.as_str().into(), | |
| 270 | message.created_at.as_str().into(), | |
| 271 | message.kind.as_str().into(), | |
| 272 | message | |
| 273 | .from_number | |
| 274 | .map_or(worker::wasm_bindgen::JsValue::NULL, |n| n.into()), | |
| 275 | message.to_number.into(), | |
| 276 | ])? | |
| 277 | .run() | |
| 278 | .await?; | |
| 279 | Ok(()) | |
| 280 | } | |
| 281 | ||
| 282 | /// The agent asked answers: the reply is recorded on the question and | |
| 283 | /// sent back to the asking agent as a message of its own. | |
| 284 | pub(crate) async fn answer_message(&self, a: AnswerMessageArgs) -> Result<Outcome<AgentMessage>> { | |
| 285 | let viewer = Some(a.actor.clone()); | |
| 286 | let repo = match self.repo(&a.repo, &viewer).await? { | |
| 287 | Outcome::Ok(repo) => repo, | |
| 288 | Outcome::Fail(failure) => return Ok(Outcome::Fail(failure)), | |
| 289 | }; | |
| 290 | if let Outcome::Fail(failure) = crate::retired::writable(&repo) { | |
| 291 | return Ok(Outcome::Fail(failure)); | |
| 292 | } | |
| 293 | if let Outcome::Fail(failure) = crate::allowed(Some(&a.actor), &repo, Capability::Run) { | |
| 294 | return Ok(Outcome::Fail(failure)); | |
| 295 | } | |
| 296 | let row = self | |
| 297 | .db | |
| 298 | .prepare("SELECT * FROM agent_messages WHERE id = ? AND repo_id = ?") | |
| 299 | .bind(&[a.id.as_str().into(), repo.id.as_str().into()])? | |
| 300 | .first::<MessageRow>(None) | |
| 301 | .await?; | |
| 302 | let Some(row) = row else { | |
| 303 | return Ok(Outcome::fail(FailureCode::NotFound, "No such message.")); | |
| 304 | }; | |
| 305 | let mut asked = AgentMessage::from(row); | |
| 306 | if !ASKS.contains(&asked.kind.as_str()) { | |
| 307 | return Ok(Outcome::fail(FailureCode::Invalid, "Only a question or a handoff is answered.")); | |
| 308 | } | |
| 309 | if asked.answer.is_some() { | |
| 310 | return Ok(Outcome::fail(FailureCode::Conflict, "It has been answered already.")); | |
| 311 | } | |
| 312 | let body: String = a.body.trim().chars().take(MAX_MESSAGE_CHARS).collect(); | |
| 313 | if body.is_empty() { | |
| 314 | return Ok(Outcome::fail(FailureCode::Invalid, "Write an answer.")); | |
| 315 | } | |
| 316 | let now = now_ms(); | |
| 317 | self.db | |
| 318 | .prepare("UPDATE agent_messages SET answer = ?, answered_at = ?, declined = ? WHERE id = ?") | |
| 319 | .bind(&[ | |
| 320 | body.as_str().into(), | |
| 321 | rfc3339(now).into(), | |
| 322 | u32::from(a.decline).into(), | |
| 323 | asked.id.as_str().into(), | |
| 324 | ])? | |
| 325 | .run() | |
| 326 | .await?; | |
| 327 | asked.answer = Some(body.clone()); | |
| 328 | asked.declined = a.decline; | |
| 329 | // An agent woken to answer lets go of its pull request once nothing | |
| 330 | // it was asked is left unanswered. | |
| 331 | self.db | |
| 332 | .prepare( | |
| 333 | "UPDATE pulls SET working_on = NULL, working_until = NULL | |
| 334 | WHERE id = (SELECT pull_id FROM agent_messages WHERE id = ?1) | |
| 335 | AND working_on = 'answer' | |
| 336 | AND NOT EXISTS ( | |
| 337 | SELECT 1 FROM agent_messages | |
| 338 | WHERE pull_id = pulls.id AND kind IN ('question', 'handoff') AND answer IS NULL)", | |
| 339 | ) | |
| 340 | .bind(&[asked.id.as_str().into()])? | |
| 341 | .run() | |
| 342 | .await?; | |
| 343 | // Back to whoever asked: the agent on the other pull request. | |
| 344 | if let Some(from) = asked.from_number { | |
| 345 | if let Some(back) = self.pull(&repo.id, from).await?.filter(|pull| pull.status.is_active()) { | |
| 346 | let reply = AgentMessage { | |
| 347 | id: new_id("msg", now), | |
| 348 | author: a.actor.username.clone(), | |
| 349 | body: if a.decline { format!("Declined: {body}") } else { body }, | |
| 350 | created_at: rfc3339(now), | |
| 351 | delivered_at: None, | |
| 352 | kind: "answer".to_owned(), | |
| 353 | from_number: Some(asked.to_number), | |
| 354 | to_number: from, | |
| 355 | answer: None, | |
| 356 | declined: a.decline, | |
| 357 | hint: None, | |
| 358 | }; | |
| 359 | self.insert_message(&repo.id, &back.id, &a.actor.id, &reply).await?; | |
| 360 | } | |
| 361 | let said = if asked.kind == "handoff" { | |
| 362 | if a.decline { "declined the handoff from" } else { "took on the handoff from" } | |
| 363 | } else { | |
| 364 | "answered the question from" | |
| 365 | }; | |
| 366 | self.note( | |
| 367 | &repo.id, | |
| 368 | asked.to_number, | |
| 369 | (a.actor.id.as_str(), a.actor.username.as_str()), | |
| 370 | &format!("{said} the agent on #{from}"), | |
| 371 | ) | |
| 372 | .await?; | |
| 373 | } | |
| 374 | Ok(Outcome::Ok(asked)) | |
| 375 | } | |
| 376 | ||
| 377 | /// Questions, handoffs and answers between the agents on these pull | |
| 378 | /// requests, newest first. | |
| 379 | pub(crate) async fn exchanges(&self, repo_id: &str, numbers: &[u32]) -> Result<Vec<AgentMessage>> { | |
| 380 | if numbers.is_empty() { | |
| 381 | return Ok(Vec::new()); | |
| 382 | } | |
| 383 | let rows = self | |
| 384 | .db | |
| 385 | .prepare( | |
| 386 | "SELECT * FROM agent_messages | |
| 387 | WHERE repo_id = ? AND kind IN ('question', 'handoff') | |
| 388 | ORDER BY created_at DESC LIMIT 50", | |
| 389 | ) | |
| 390 | .bind(&[repo_id.into()])? | |
| 391 | .all() | |
| 392 | .await? | |
| 393 | .results::<MessageRow>()?; | |
| 394 | Ok(rows | |
| 395 | .into_iter() | |
| 396 | .map(AgentMessage::from) | |
| 397 | .filter(|message| { | |
| 398 | numbers.contains(&message.to_number) | |
| 399 | || message.from_number.is_some_and(|from| numbers.contains(&from)) | |
| 400 | }) | |
| 401 | .collect()) | |
| 402 | } | |
| 403 | ||
| 404 | /// The undelivered messages, marked delivered and recorded in the | |
| 405 | /// session, for the agent's sandbox. | |
| 406 | pub(crate) async fn take_messages(&self, a: TakeMessagesArgs) -> Result<Outcome<Vec<AgentMessage>>> { | |
| 407 | if a.actor.kind != PrincipalKind::Agent { | |
| 408 | return Ok(Outcome::fail( | |
| 409 | FailureCode::Forbidden, | |
| 410 | "Only g1t's agents take messages.", | |
| 411 | )); | |
| 412 | } | |
| 413 | let viewer = Some(a.actor.clone()); | |
| 414 | let (_, pull) = match self.pull_at(&a.repo, a.number, &viewer).await? { | |
| 415 | Outcome::Ok(found) => found, | |
| 416 | Outcome::Fail(failure) => return Ok(Outcome::Fail(failure)), | |
| 417 | }; | |
| 418 | Ok(Outcome::Ok(self.deliver(&pull).await?)) | |
| 419 | } | |
| 420 | ||
| 421 | /// Wakes the agent on a pull request to answer what it was asked while | |
| 422 | /// it was not at work: claims a short step, and hands over its messages. | |
| 423 | pub(crate) async fn wake_for_messages(&self, a: WakeForMessagesArgs) -> Result<Option<Wake>> { | |
| 424 | let Some(pull) = self.pull_by_id(&a.pull_id).await? else { | |
| 425 | return Ok(None); | |
| 426 | }; | |
| 427 | // Only another agent's question or handoff wakes it; what people | |
| 428 | // say waits for its next step. | |
| 429 | let waiting = self | |
| 430 | .db | |
| 431 | .prepare( | |
| 432 | "SELECT 1 AS value FROM agent_messages | |
| 433 | WHERE pull_id = ? AND delivered_at IS NULL AND kind IN ('question', 'handoff') | |
| 434 | LIMIT 1", | |
| 435 | ) | |
| 436 | .bind(&[pull.id.as_str().into()])? | |
| 437 | .first::<u32>(Some("value")) | |
| 438 | .await? | |
| 439 | .is_some(); | |
| 440 | if !waiting || !self.claim(&pull.id, "answer", ANSWER_MINUTES, false).await? { | |
| 441 | return Ok(None); | |
| 442 | } | |
| 443 | let repo: Outcome<g1t_contracts::repos::Repo> = g1t_kit::call( | |
| 444 | &self.repos, | |
| 445 | "get_by_id", | |
| 446 | &g1t_contracts::repos::GetByIdArgs { | |
| 447 | id: pull.repo_id.clone(), | |
| 448 | viewer: self.owner_viewer(&pull).await?, | |
| 449 | }, | |
| 450 | ) | |
| 451 | .await?; | |
| 452 | let Outcome::Ok(repo) = crate::retired::unless_archived(repo) else { | |
| 453 | return Ok(None); | |
| 454 | }; | |
| Teams and CODEOWNERS, labels and milestones, dependency updates, the security suite, and a clearer top bar | 455 | let base = pull.base_branch(&repo.default_branch).to_owned(); |
| The sidebar asks billing whether it is on, not for the account, so no page waits on the card processor | 456 | let path = g1t_contracts::repos::RepoPath { |
| 457 | namespace: repo.namespace, | |
| 458 | name: repo.name, | |
| 459 | }; | |
| 460 | let issue = match pull.issue { | |
| 461 | Some(number) => self.issue(&pull.repo_id, number).await?, | |
| 462 | None => None, | |
| 463 | }; | |
| 464 | let messages = self.deliver(&pull).await?; | |
| 465 | let asking: Vec<String> = messages | |
| 466 | .iter() | |
| 467 | .filter_map(|message| message.from_number.map(|from| format!("#{from}"))) | |
| 468 | .collect(); | |
| 469 | self.note( | |
| 470 | &pull.repo_id, | |
| 471 | pull.number, | |
| 472 | (crate::lifecycle::POLICY_ACTOR_ID, crate::lifecycle::POLICY_ACTOR_NAME), | |
| 473 | &format!("woke g1t to answer the agent on {}", asking.join(", ")), | |
| 474 | ) | |
| 475 | .await?; | |
| 476 | Ok(Some(Wake { | |
| 477 | job: LifecycleJob { | |
| 478 | pull_id: pull.id, | |
| 479 | source: pull.fork.unwrap_or_else(|| path.clone()), | |
| 480 | repo: path, | |
| 481 | number: pull.number, | |
| 482 | author: pull.requested_by.unwrap_or(pull.author), | |
| 483 | branch: pull.branch, | |
| Teams and CODEOWNERS, labels and milestones, dependency updates, the security suite, and a clearer top bar | 484 | default_branch: base, |
| The sidebar asks billing whether it is on, not for the account, so no page waits on the card processor | 485 | title: pull.title, |
| 486 | description: pull.body.unwrap_or_default(), | |
| 487 | issue, | |
| 488 | feedback: String::new(), | |
| 489 | round: 0, | |
| 490 | }, | |
| 491 | messages, | |
| 492 | })) | |
| 493 | } | |
| 494 | ||
| 495 | /// Marks a pull request's undelivered messages delivered, records them | |
| 496 | /// in its session, and returns them, oldest first. | |
| 497 | async fn deliver(&self, pull: &Pull) -> Result<Vec<AgentMessage>> { | |
| 498 | let now = rfc3339(now_ms()); | |
| 499 | let mut taken: Vec<AgentMessage> = self | |
| 500 | .db | |
| 501 | .prepare( | |
| 502 | "UPDATE agent_messages SET delivered_at = ? | |
| 503 | WHERE pull_id = ? AND delivered_at IS NULL | |
| 504 | RETURNING *", | |
| 505 | ) | |
| 506 | .bind(&[now.as_str().into(), pull.id.as_str().into()])? | |
| 507 | .all() | |
| 508 | .await? | |
| 509 | .results::<MessageRow>()? | |
| 510 | .into_iter() | |
| 511 | .map(AgentMessage::from) | |
| 512 | .collect(); | |
| 513 | taken.sort_by(|a, b| a.created_at.cmp(&b.created_at)); | |
| 514 | if !taken.is_empty() { | |
| 515 | let entries: Vec<NewSessionEntry> = taken | |
| 516 | .iter() | |
| 517 | .map(|message| NewSessionEntry { | |
| 518 | kind: SessionEntryKind::Prompt, | |
| 519 | text: match message.kind.as_str() { | |
| 520 | "question" => format!("Question from {} ({}): {}", sender(message), message.id, message.body), | |
| 521 | "handoff" => format!("Work handed over by {} ({}): {}", sender(message), message.id, message.body), | |
| 522 | "answer" => format!("Answer from {}: {}", sender(message), message.body), | |
| 523 | _ => format!("Message from {}: {}", message.author, message.body), | |
| 524 | }, | |
| 525 | tool: None, | |
| 526 | commit: None, | |
| 527 | }) | |
| 528 | .collect(); | |
| 529 | self.append_entries(pull, &entries).await?; | |
| 530 | } | |
| 531 | Ok(taken) | |
| 532 | } | |
| 533 | } |
This file's history is long; its oldest lines are credited to the oldest commit read.