sscoble/pi-mono
1
0
Fork 0
mirror of https://github.com/badlogic/pi-mono.git synced 2026-11-05 23:37:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

docs(agent): clarify branch query helpers

Browse source
This commit is contained in:
Vegard Stikbakke 2026-08-07 09:55:13 +02:00
parent 80ef7ff0f4
commit f562a1a44f
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
packages/agent/src/harness/session/session.ts
View file

@ -223,6 +223,7 @@ export class Session<TMetadata extends SessionMetadata = SessionMetadata> implem
return this.queryLog(options);
}
/** Returns the lane's current leaf, or null when empty. Throws when the lane does not exist. */
private async getLeafIdForLane(lane: string): Promise<string | null> {
const pointer = (await this.getLanes()).find((candidate) => candidate.lane === lane);
if (!pointer) throw new SessionError("invalid_lane", `Lane not found: ${lane}`);
@ -235,14 +236,18 @@ export class Session<TMetadata extends SessionMetadata = SessionMetadata> implem
return this.storage.findEntries(resultLimit === query.limit ? query : { ...query, limit: resultLimit });
}
/**
* Queries from `query.start` toward the root, defaulting to the lane's current leaf.
* `resultLimit` lets single-entry queries cap results without changing the caller's query.
*/
private async queryBranchEntries(
lane: string,
defaultLane: string,
query: EntryQuery & BranchBounds = {},
resultLimit = query.limit,
): Promise<Entry[]> {
assertValidLimit(query.limit);
assertValidCursor(query.cursor?.afterSeq);
const start = query.start ?? (await this.getLeafIdForLane(lane));
const start = query.start ?? (await this.getLeafIdForLane(defaultLane));
if (start === null) return [];
const storageQuery = resultLimit === query.limit ? query : { ...query, limit: resultLimit };
return this.storage.findEntriesOnBranch({ ...storageQuery, start });