agent-trace 0.2.10 → 0.2.11

package/agent-trace.cjs

@@ -25077,6 +25077,18 @@ CREATE TABLE IF NOT EXISTS commits (
 );
 
 CREATE INDEX IF NOT EXISTS idx_commits_session ON commits(session_id);
+
+CREATE TABLE IF NOT EXISTS pull_requests (
+  session_id TEXT NOT NULL,
+  repo TEXT NOT NULL,
+  pr_number INTEGER NOT NULL,
+  state TEXT NOT NULL DEFAULT 'open',
+  url TEXT,
+  merged_at TEXT,
+  PRIMARY KEY (session_id, repo, pr_number)
+);
+
+CREATE INDEX IF NOT EXISTS idx_pull_requests_session ON pull_requests(session_id);
 `;
 function toJsonArray(value) {
   return JSON.stringify(value);
@@ -25259,6 +25271,37 @@ var SqliteClient = class {
     });
     transaction(rows);
   }
+  async upsertPullRequests(rows) {
+    if (rows.length === 0) return;
+    const upsert = this.db.prepare(`
+      INSERT INTO pull_requests
+        (session_id, repo, pr_number, state, url, merged_at)
+      VALUES (?, ?, ?, ?, ?, ?)
+      ON CONFLICT(session_id, repo, pr_number) DO UPDATE SET
+        state = excluded.state,
+        url = COALESCE(excluded.url, pull_requests.url),
+        merged_at = COALESCE(excluded.merged_at, pull_requests.merged_at)
+    `);
+    const transaction = this.db.transaction((prRows) => {
+      for (const row of prRows) {
+        upsert.run(
+          row.session_id,
+          row.repo,
+          row.pr_number,
+          row.state,
+          row.url,
+          row.merged_at
+        );
+      }
+    });
+    transaction(rows);
+  }
+  listPullRequestsBySessionId(sessionId) {
+    const rows = this.db.prepare(
+      "SELECT session_id, repo, pr_number, state, url, merged_at FROM pull_requests WHERE session_id = ? ORDER BY pr_number ASC"
+    ).all(sessionId);
+    return rows;
+  }
   listCommitsBySessionId(sessionId) {
     const rows = this.db.prepare(
       "SELECT sha, session_id, prompt_id, message, lines_added, lines_removed, committed_at FROM commits WHERE session_id = ? ORDER BY committed_at ASC"
@@ -25976,6 +26019,19 @@ function toPostgresCommitRow(trace, commit) {
 function toPostgresCommitRows(trace) {
   return trace.git.commits.map((commit) => toPostgresCommitRow(trace, commit));
 }
+function toPostgresPullRequestRow(trace, pr) {
+  return {
+    session_id: trace.sessionId,
+    repo: pr.repo,
+    pr_number: pr.prNumber,
+    state: pr.state,
+    url: toNullableString(pr.url),
+    merged_at: toNullableString(pr.mergedAt)
+  };
+}
+function toPostgresPullRequestRows(trace) {
+  return trace.git.pullRequests.map((pr) => toPostgresPullRequestRow(trace, pr));
+}
 function dedupeBySessionId(rows) {
   const bySession = /* @__PURE__ */ new Map();
   rows.forEach((row) => {
@@ -25990,6 +26046,13 @@ function dedupeBySha(rows) {
   });
   return [...bySha.values()];
 }
+function dedupePullRequests(rows) {
+  const byKey = /* @__PURE__ */ new Map();
+  rows.forEach((row) => {
+    byKey.set(`${row.session_id}:${row.repo}:${String(row.pr_number)}`, row);
+  });
+  return [...byKey.values()];
+}
 var PostgresSessionWriter = class {
   client;
   constructor(client) {
@@ -26007,9 +26070,11 @@ var PostgresSessionWriter = class {
     }
     const sessions = dedupeBySessionId(traces.map(toPostgresSessionRow));
     const commitRows = dedupeBySha(traces.flatMap((trace) => toPostgresCommitRows(trace)));
+    const prRows = dedupePullRequests(traces.flatMap((trace) => toPostgresPullRequestRows(trace)));
     const sessionsPromise = this.client.upsertSessions(sessions);
     const commitsPromise = this.client.upsertCommits(commitRows);
-    await Promise.all([sessionsPromise, commitsPromise]);
+    const prsPromise = this.client.upsertPullRequests(prRows);
+    await Promise.all([sessionsPromise, commitsPromise, prsPromise]);
     return {
       writtenSessions: sessions.length,
       writtenCommits: commitRows.length
@@ -28226,6 +28291,7 @@ var InMemoryRuntimeClickHouseClient = class {
 var InMemoryRuntimePostgresClient = class {
   sessionsById = /* @__PURE__ */ new Map();
   commitsBySha = /* @__PURE__ */ new Map();
+  pullRequestsByKey = /* @__PURE__ */ new Map();
   async upsertSessions(rows) {
     rows.forEach((row) => {
       this.sessionsById.set(row.session_id, row);
@@ -28236,12 +28302,20 @@ var InMemoryRuntimePostgresClient = class {
       this.commitsBySha.set(row.sha, row);
     });
   }
+  async upsertPullRequests(rows) {
+    rows.forEach((row) => {
+      this.pullRequestsByKey.set(`${row.session_id}:${row.repo}:${String(row.pr_number)}`, row);
+    });
+  }
   listSessions() {
     return [...this.sessionsById.values()];
   }
   listCommits() {
     return [...this.commitsBySha.values()];
   }
+  listPullRequests() {
+    return [...this.pullRequestsByKey.values()];
+  }
 };
 var InMemoryRuntimeSessionTraceClient = class {
   rows = [];
@@ -28851,10 +28925,21 @@ function hydrateFromSqlite(runtime, sqlite, limit, eventLimit) {
       const extra = commits.filter((c) => !pgShas.has(c.sha));
       commits = [...mapped, ...extra];
     }
+    let pullRequests = trace.git.pullRequests;
+    const pgPrs = sqlite.listPullRequestsBySessionId(row.session_id);
+    if (pgPrs.length > 0) {
+      pullRequests = pgPrs.map((pr) => ({
+        repo: pr.repo,
+        prNumber: pr.pr_number,
+        state: pr.state,
+        ...pr.url !== null ? { url: pr.url } : {},
+        ...pr.merged_at !== null ? { mergedAt: pr.merged_at } : {}
+      }));
+    }
     let hydratedTrace = {
       ...trace,
       timeline,
-      git: { ...trace.git, commits }
+      git: { ...trace.git, commits, pullRequests }
     };
     if (hydratedTrace.metrics.totalCostUsd === 0 && (hydratedTrace.metrics.totalInputTokens > 0 || hydratedTrace.metrics.totalOutputTokens > 0) && hydratedTrace.metrics.modelsUsed.length > 0) {
       const model = String(hydratedTrace.metrics.modelsUsed[0]);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-trace",
-  "version": "0.2.10",
+  "version": "0.2.11",
   "description": "Self-hosted observability for AI coding agents. One command, zero config.",
   "license": "Apache-2.0",
   "bin": {