@sixb/pg 0.1.6 → 0.1.7

package/dist/migrations.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"migrations.d.ts","sourceRoot":"","sources":["../src/migrations.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,YAAY,CAAA;AACjD,OAAO,KAAK,EAGV,YAAY,EACZ,aAAa,EACb,oBAAoB,EACrB,MAAM,oBAAoB,CAAA;AAmE3B,OAAO,KAAK,EAAE,GAAG,EAAa,MAAM,aAAa,CAAA;AAEjD,MAAM,WAAW,wBAAwB;IACvC,IAAI,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;CACrC;AAED,eAAO,MAAM,2BAA2B,wBAAwB,CAAA;AAEhE,wBAAgB,KAAK,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,aAAa,CAAC,wBAAwB,CAAC,CAE1F;AAED,wBAAgB,MAAM,CACpB,EAAE,EAAE,MAAM,EACV,EAAE,EAAE,CAAC,OAAO,EAAE,wBAAwB,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,EAC/D,OAAO,GAAE,oBAAyB,GACjC,aAAa,CAAC,wBAAwB,CAAC,CAEzC;AAED,wBAAgB,8BAA8B,CAC5C,GAAG,EAAE,GAAG,EACR,UAAU,EAAE,MAAM,GACjB,SAAS,eAAe,EAAE,CAQ5B;AAED,wBAAgB,sBAAsB,CAAC,MAAM,EAAE;IAC7C,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAA;IACjB,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAA;IAC3B,QAAQ,CAAC,UAAU,EAAE,YAAY,CAAC,wBAAwB,CAAC,CAAA;CAC5D,GAAG,eAAe,CAqBlB;AAED,wBAAsB,sBAAsB,CAAC,GAAG,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAMxF;AAED,wBAAsB,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAE5E;AAED,wBAAgB,UAAU,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAKrD;AA6KD,eAAO,MAAM,yBAAyB,wCAgCpC,CAAA"}
+{"version":3,"file":"migrations.d.ts","sourceRoot":"","sources":["../src/migrations.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,YAAY,CAAA;AACjD,OAAO,KAAK,EAGV,YAAY,EACZ,aAAa,EACb,oBAAoB,EACrB,MAAM,oBAAoB,CAAA;AA+E3B,OAAO,KAAK,EAAE,GAAG,EAAa,MAAM,aAAa,CAAA;AAEjD,MAAM,WAAW,wBAAwB;IACvC,IAAI,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;CACrC;AAED,eAAO,MAAM,2BAA2B,wBAAwB,CAAA;AAEhE,wBAAgB,KAAK,CAAC,EAAE,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,aAAa,CAAC,wBAAwB,CAAC,CAE1F;AAED,wBAAgB,MAAM,CACpB,EAAE,EAAE,MAAM,EACV,EAAE,EAAE,CAAC,OAAO,EAAE,wBAAwB,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,EAC/D,OAAO,GAAE,oBAAyB,GACjC,aAAa,CAAC,wBAAwB,CAAC,CAEzC;AAED,wBAAgB,8BAA8B,CAC5C,GAAG,EAAE,GAAG,EACR,UAAU,EAAE,MAAM,GACjB,SAAS,eAAe,EAAE,CAQ5B;AAED,wBAAgB,sBAAsB,CAAC,MAAM,EAAE;IAC7C,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAA;IACjB,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAA;IAC3B,QAAQ,CAAC,UAAU,EAAE,YAAY,CAAC,wBAAwB,CAAC,CAAA;CAC5D,GAAG,eAAe,CAqBlB;AAED,wBAAsB,sBAAsB,CAAC,GAAG,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAMxF;AAED,wBAAsB,UAAU,CAAC,GAAG,EAAE,GAAG,EAAE,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAE5E;AAED,wBAAgB,UAAU,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAKrD;AA6KD,eAAO,MAAM,yBAAyB,wCAsCpC,CAAA"}

package/dist/pg-ai-cost-storage.d.ts

@@ -1,9 +1,10 @@
-import { type AiAccountingOverview, type AiCostStorage, type AiCostSummary, type AiModelCallCostRecord, type ListAiModelCallAccountingInput, type ListAiModelCallAccountingResult, type QueryAiAccountingOverviewInput, type SummarizeAiCostExecutionsInput } from "@sixb/core/storage";
+import { type AiAccountingOverview, type AiCostStorage, type AiCostSummary, type AiModelCallCostRecord, type ListAiModelCallAccountingInput, type ListAiModelCallAccountingResult, type ListAiModelCallGroupsInput, type ListAiModelCallGroupsResult, type QueryAiAccountingOverviewInput, type SummarizeAiCostExecutionsInput } from "@sixb/core/storage";
 import type { PgStoreClient } from "./transactions";
 /** PostgreSQL-backed immutable valuations stored in one append-only table. */
 export declare class PgAiCostStorage implements AiCostStorage {
     private readonly sql;
     constructor(sql: PgStoreClient);
+    listModelCallGroups(input: ListAiModelCallGroupsInput): Promise<ListAiModelCallGroupsResult>;
     recordModelCallCost(input: AiModelCallCostRecord): Promise<void>;
     summarizeExecutions(input: SummarizeAiCostExecutionsInput): Promise<readonly AiCostSummary[]>;
     queryProjectOverview(input: QueryAiAccountingOverviewInput): Promise<AiAccountingOverview>;

package/dist/pg-ai-cost-storage.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"pg-ai-cost-storage.d.ts","sourceRoot":"","sources":["../src/pg-ai-cost-storage.ts"],"names":[],"mappings":"AAaA,OAAO,EACL,KAAK,oBAAoB,EACzB,KAAK,aAAa,EAElB,KAAK,aAAa,EAClB,KAAK,qBAAqB,EAE1B,KAAK,8BAA8B,EACnC,KAAK,+BAA+B,EACpC,KAAK,8BAA8B,EACnC,KAAK,8BAA8B,EACpC,MAAM,oBAAoB,CAAA;AAC3B,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAA;AAGnD,8EAA8E;AAC9E,qBAAa,eAAgB,YAAW,aAAa;IACvC,OAAO,CAAC,QAAQ,CAAC,GAAG;gBAAH,GAAG,EAAE,aAAa;IAEzC,mBAAmB,CAAC,KAAK,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC;IA6BhE,mBAAmB,CACvB,KAAK,EAAE,8BAA8B,GACpC,OAAO,CAAC,SAAS,aAAa,EAAE,CAAC;IAsB9B,oBAAoB,CAAC,KAAK,EAAE,8BAA8B,GAAG,OAAO,CAAC,oBAAoB,CAAC;IA6G1F,cAAc,CAClB,KAAK,EAAE,8BAA8B,GACpC,OAAO,CAAC,+BAA+B,CAAC;CAsF5C"}
+{"version":3,"file":"pg-ai-cost-storage.d.ts","sourceRoot":"","sources":["../src/pg-ai-cost-storage.ts"],"names":[],"mappings":"AAcA,OAAO,EACL,KAAK,oBAAoB,EACzB,KAAK,aAAa,EAElB,KAAK,aAAa,EAClB,KAAK,qBAAqB,EAE1B,KAAK,8BAA8B,EACnC,KAAK,+BAA+B,EACpC,KAAK,0BAA0B,EAC/B,KAAK,2BAA2B,EAChC,KAAK,8BAA8B,EACnC,KAAK,8BAA8B,EACpC,MAAM,oBAAoB,CAAA;AAG3B,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAA;AAGnD,8EAA8E;AAC9E,qBAAa,eAAgB,YAAW,aAAa;IACvC,OAAO,CAAC,QAAQ,CAAC,GAAG;gBAAH,GAAG,EAAE,aAAa;IAEzC,mBAAmB,CACvB,KAAK,EAAE,0BAA0B,GAChC,OAAO,CAAC,2BAA2B,CAAC;IAIjC,mBAAmB,CAAC,KAAK,EAAE,qBAAqB,GAAG,OAAO,CAAC,IAAI,CAAC;IA6BhE,mBAAmB,CACvB,KAAK,EAAE,8BAA8B,GACpC,OAAO,CAAC,SAAS,aAAa,EAAE,CAAC;IAsB9B,oBAAoB,CAAC,KAAK,EAAE,8BAA8B,GAAG,OAAO,CAAC,oBAAoB,CAAC;IAoH1F,cAAc,CAClB,KAAK,EAAE,8BAA8B,GACpC,OAAO,CAAC,+BAA+B,CAAC;CAwF5C"}

package/dist/pg-ai-limit-storage.d.ts

@@ -0,0 +1,34 @@
+import type { AiLimitPolicy, AiLimitPolicyStatus, AiLimitStorage, AiModelCallReservation, CreateAiLimitPolicyInput, DeleteAiLimitPolicyInput, GetAiLimitPolicyInput, ListAiLimitPoliciesInput, ListAiLimitPolicyStatusesInput, MarkAiModelCallReservationUnknownInput, ReconcileAiModelCallInput, RecordAiModelCallLimitActualsInput, ReserveAiModelCallInput, ReserveAiModelCallResult, UpdateAiLimitPolicyInput } from "@sixb/core/storage";
+import { type PgStoreClient } from "./transactions";
+/** PostgreSQL-backed editable AI limits and atomic model-call reservations. */
+export declare class PgAiLimitStorage implements AiLimitStorage {
+    private readonly sql;
+    private readonly operations;
+    constructor(sql: PgStoreClient);
+    /** @internal Keep queued work inside the lifetime of the owning SQL transaction. */
+    settleOperations(): Promise<void>;
+    createPolicy(input: CreateAiLimitPolicyInput): Promise<AiLimitPolicy>;
+    updatePolicy(input: UpdateAiLimitPolicyInput): Promise<AiLimitPolicy>;
+    deletePolicy(input: DeleteAiLimitPolicyInput): Promise<boolean>;
+    getPolicy(input: GetAiLimitPolicyInput): Promise<AiLimitPolicy | null>;
+    listPolicies(input: ListAiLimitPoliciesInput): Promise<readonly AiLimitPolicy[]>;
+    listPolicyStatuses(input: ListAiLimitPolicyStatusesInput): Promise<readonly AiLimitPolicyStatus[]>;
+    reserveModelCall(input: ReserveAiModelCallInput): Promise<ReserveAiModelCallResult>;
+    recordModelCallActuals(input: RecordAiModelCallLimitActualsInput): Promise<void>;
+    reconcileModelCall(input: ReconcileAiModelCallInput): Promise<AiModelCallReservation>;
+    markReservationUnknown(input: MarkAiModelCallReservationUnknownInput): Promise<AiModelCallReservation>;
+    private withProjectLock;
+    private policyRows;
+    private requirePolicy;
+    private statusForPolicy;
+    private changeCounters;
+    private moveEstimateCounters;
+    private reconcileEstimateCounters;
+    private insertReservation;
+    private updateReservation;
+    private findReservation;
+    private accountingEntries;
+    private requireAccountingEntry;
+    private assertExecutionExists;
+}
+//# sourceMappingURL=pg-ai-limit-storage.d.ts.map

package/dist/pg-ai-limit-storage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pg-ai-limit-storage.d.ts","sourceRoot":"","sources":["../src/pg-ai-limit-storage.ts"],"names":[],"mappings":"AAwBA,OAAO,KAAK,EAEV,aAAa,EACb,mBAAmB,EAGnB,cAAc,EAEd,sBAAsB,EACtB,wBAAwB,EACxB,wBAAwB,EACxB,qBAAqB,EACrB,wBAAwB,EACxB,8BAA8B,EAC9B,sCAAsC,EACtC,yBAAyB,EACzB,kCAAkC,EAClC,uBAAuB,EACvB,wBAAwB,EACxB,wBAAwB,EACzB,MAAM,oBAAoB,CAAA;AAG3B,OAAO,EAAoB,KAAK,aAAa,EAAoB,MAAM,gBAAgB,CAAA;AAEvF,+EAA+E;AAC/E,qBAAa,gBAAiB,YAAW,cAAc;IAGzC,OAAO,CAAC,QAAQ,CAAC,GAAG;IAFhC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAA6B;gBAE3B,GAAG,EAAE,aAAa;IAE/C,oFAAoF;IACpF,gBAAgB,IAAI,OAAO,CAAC,IAAI,CAAC;IAI3B,YAAY,CAAC,KAAK,EAAE,wBAAwB,GAAG,OAAO,CAAC,aAAa,CAAC;IAsBrE,YAAY,CAAC,KAAK,EAAE,wBAAwB,GAAG,OAAO,CAAC,aAAa,CAAC;IAiBrE,YAAY,CAAC,KAAK,EAAE,wBAAwB,GAAG,OAAO,CAAC,OAAO,CAAC;IAY/D,SAAS,CAAC,KAAK,EAAE,qBAAqB,GAAG,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC;IAStE,YAAY,CAAC,KAAK,EAAE,wBAAwB,GAAG,OAAO,CAAC,SAAS,aAAa,EAAE,CAAC;IAOhF,kBAAkB,CACtB,KAAK,EAAE,8BAA8B,GACpC,OAAO,CAAC,SAAS,mBAAmB,EAAE,CAAC;IAapC,gBAAgB,CAAC,KAAK,EAAE,uBAAuB,GAAG,OAAO,CAAC,wBAAwB,CAAC;IAqGnF,sBAAsB,CAAC,KAAK,EAAE,kCAAkC,GAAG,OAAO,CAAC,IAAI,CAAC;IA0ChF,kBAAkB,CAAC,KAAK,EAAE,yBAAyB,GAAG,OAAO,CAAC,sBAAsB,CAAC;IA+BrF,sBAAsB,CAC1B,KAAK,EAAE,sCAAsC,GAC5C,OAAO,CAAC,sBAAsB,CAAC;YAepB,eAAe;IAe7B,OAAO,CAAC,UAAU;YAYJ,aAAa;YAeb,eAAe;YA2Cf,cAAc;YA8Cd,oBAAoB;YA2BpB,yBAAyB;YAmCzB,iBAAiB;YAkBjB,iBAAiB;YAgBjB,eAAe;YAmBf,iBAAiB;YA8CjB,sBAAsB;YAwCtB,qBAAqB;CAgBpC"}

package/dist/pg-ai-usage-storage.d.ts

@@ -1,3 +1,4 @@
+import type { ReadonlyJsonObject } from "@sixb/core/storage";
 import { type AiModelCallUsageRecord, type AiUsageExecutionSummary, type AiUsageStorage, type GetLatestAiModelCallForExecutionInput, type RecordAiModelCallInput, type RecordAiModelCallResult, type SummarizeAiUsageExecutionInput, type SummarizeAiUsageExecutionsInput } from "@sixb/core/storage";
 import { type PgStoreClient } from "./transactions";
 /** PostgreSQL-backed append-only model-call usage ledger. */
@@ -14,4 +15,31 @@ export declare class PgAiUsageStorage implements AiUsageStorage {
     private executionRows;
     private assertExecutionExists;
 }
+interface AiUsageRow {
+    readonly project_id: string;
+    readonly id: string;
+    readonly execution_id: string;
+    readonly attempt: number | string;
+    readonly call_id: string;
+    readonly provider_id: string;
+    readonly provider_ids: Record<string, unknown> | string | null;
+    readonly requested_model_id: string;
+    readonly requested_reasoning: string | Record<string, unknown> | null;
+    readonly response_model_id: string | null;
+    readonly response_id: string;
+    readonly input_tokens: number | string | null;
+    readonly output_tokens: number | string | null;
+    readonly total_tokens: number | string | null;
+    readonly uncached_input_tokens: number | string | null;
+    readonly cache_read_input_tokens: number | string | null;
+    readonly cache_write_input_tokens: number | string | null;
+    readonly text_output_tokens: number | string | null;
+    readonly reasoning_output_tokens: number | string | null;
+    readonly reporting_status: AiModelCallUsageRecord["usage"]["reportingStatus"];
+    readonly raw_usage: ReadonlyJsonObject | string | null;
+    readonly occurred_at: Date | string;
+    readonly recorded_at: Date | string;
+}
+export declare function requestedReasoningFromRow(value: Exclude<AiUsageRow["requested_reasoning"], null>): AiModelCallUsageRecord["requestedReasoning"];
+export {};
 //# sourceMappingURL=pg-ai-usage-storage.d.ts.map

package/dist/pg-ai-usage-storage.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"pg-ai-usage-storage.d.ts","sourceRoot":"","sources":["../src/pg-ai-usage-storage.ts"],"names":[],"mappings":"AACA,OAAO,EAEL,KAAK,sBAAsB,EAC3B,KAAK,uBAAuB,EAC5B,KAAK,cAAc,EAInB,KAAK,qCAAqC,EAE1C,KAAK,sBAAsB,EAC3B,KAAK,uBAAuB,EAC5B,KAAK,8BAA8B,EACnC,KAAK,+BAA+B,EACrC,MAAM,oBAAoB,CAAA;AAE3B,OAAO,EAAE,KAAK,aAAa,EAAoB,MAAM,gBAAgB,CAAA;AAErE,6DAA6D;AAC7D,qBAAa,gBAAiB,YAAW,cAAc;IACzC,OAAO,CAAC,QAAQ,CAAC,GAAG;gBAAH,GAAG,EAAE,aAAa;IAEzC,eAAe,CAAC,KAAK,EAAE,sBAAsB,GAAG,OAAO,CAAC,uBAAuB,CAAC;IAgFhF,kBAAkB,CACtB,KAAK,EAAE,8BAA8B,GACpC,OAAO,CAAC,uBAAuB,CAAC;IAQ7B,qBAAqB,CACzB,KAAK,EAAE,qCAAqC,GAC3C,OAAO,CAAC,sBAAsB,GAAG,IAAI,CAAC;IAYnC,mBAAmB,CACvB,KAAK,EAAE,+BAA+B,GACrC,OAAO,CAAC,SAAS,uBAAuB,EAAE,CAAC;YAShC,oBAAoB;YAkBpB,WAAW;YAYX,WAAW;IA6BzB,OAAO,CAAC,aAAa;YAiBP,qBAAqB;CAgBpC"}
+{"version":3,"file":"pg-ai-usage-storage.d.ts","sourceRoot":"","sources":["../src/pg-ai-usage-storage.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAA;AAC5D,OAAO,EAEL,KAAK,sBAAsB,EAC3B,KAAK,uBAAuB,EAC5B,KAAK,cAAc,EAInB,KAAK,qCAAqC,EAE1C,KAAK,sBAAsB,EAC3B,KAAK,uBAAuB,EAC5B,KAAK,8BAA8B,EACnC,KAAK,+BAA+B,EACrC,MAAM,oBAAoB,CAAA;AAE3B,OAAO,EAAE,KAAK,aAAa,EAAoB,MAAM,gBAAgB,CAAA;AAErE,6DAA6D;AAC7D,qBAAa,gBAAiB,YAAW,cAAc;IACzC,OAAO,CAAC,QAAQ,CAAC,GAAG;gBAAH,GAAG,EAAE,aAAa;IAEzC,eAAe,CAAC,KAAK,EAAE,sBAAsB,GAAG,OAAO,CAAC,uBAAuB,CAAC;IAwFhF,kBAAkB,CACtB,KAAK,EAAE,8BAA8B,GACpC,OAAO,CAAC,uBAAuB,CAAC;IAQ7B,qBAAqB,CACzB,KAAK,EAAE,qCAAqC,GAC3C,OAAO,CAAC,sBAAsB,GAAG,IAAI,CAAC;IAYnC,mBAAmB,CACvB,KAAK,EAAE,+BAA+B,GACrC,OAAO,CAAC,SAAS,uBAAuB,EAAE,CAAC;YAShC,oBAAoB;YAkBpB,WAAW;YAYX,WAAW;IAuCzB,OAAO,CAAC,aAAa;YAiBP,qBAAqB;CAgBpC;AAED,UAAU,UAAU;IAClB,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAA;IAC3B,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAA;IACnB,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAA;IAC7B,QAAQ,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAAA;IACjC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAA;IACxB,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAA;IAC5B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,GAAG,IAAI,CAAA;IAC9D,QAAQ,CAAC,kBAAkB,EAAE,MAAM,CAAA;IACnC,QAAQ,CAAC,mBAAmB,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAA;IACrE,QAAQ,CAAC,iBAAiB,EAAE,MAAM,GAAG,IAAI,CAAA;IACzC,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAA;IAC5B,QAAQ,CAAC,YAAY,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAA;IAC7C,QAAQ,CAAC,aAAa,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAA;IAC9C,QAAQ,CAAC,YAAY,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAA;IAC7C,QAAQ,CAAC,qBAAqB,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAA;IACtD,QAAQ,CAAC,uBAAuB,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAA;IACxD,QAAQ,CAAC,wBAAwB,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAA;IACzD,QAAQ,CAAC,kBAAkB,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAA;IACnD,QAAQ,CAAC,uBAAuB,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAAA;IACxD,QAAQ,CAAC,gBAAgB,EAAE,sBAAsB,CAAC,OAAO,CAAC,CAAC,iBAAiB,CAAC,CAAA;IAC7E,QAAQ,CAAC,SAAS,EAAE,kBAAkB,GAAG,MAAM,GAAG,IAAI,CAAA;IACtD,QAAQ,CAAC,WAAW,EAAE,IAAI,GAAG,MAAM,CAAA;IACnC,QAAQ,CAAC,WAAW,EAAE,IAAI,GAAG,MAAM,CAAA;CACpC;AA8CD,wBAAgB,yBAAyB,CACvC,KAAK,EAAE,OAAO,CAAC,UAAU,CAAC,qBAAqB,CAAC,EAAE,IAAI,CAAC,GACtD,sBAAsB,CAAC,oBAAoB,CAAC,CAS9C"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sixb/pg",
-  "version": "0.1.6",
+  "version": "0.1.7",
   "description": "PostgreSQL storage provider for Sixb",
   "keywords": [
     "sixb",
@@ -33,17 +33,17 @@
   "scripts": {
     "build": "bun ../../scripts/build-package.ts",
     "typecheck": "tsc --noEmit",
-    "test:e2e": "bun test --preload ./tests/setup.ts ./tests/*.e2e.ts",
+    "test:e2e": "bun ../../scripts/ci-guard.ts --stall 90 --max 600 bun test --preload ./tests/setup.ts ./tests/*.e2e.ts",
     "prepack": "bun run build"
   },
   "dependencies": {
     "postgres": "^3.4.5"
   },
   "peerDependencies": {
-    "@sixb/core": "0.1.6"
+    "@sixb/core": "0.1.7"
   },
   "devDependencies": {
-    "@sixb/core": "0.1.6",
+    "@sixb/core": "0.1.7",
     "@types/bun": "latest",
     "typescript": "^5.7.0"
   },

package/src/agents/rows.ts

@@ -7,7 +7,10 @@ import {
   type AgentRunDiagnostic,
   type AgentRunRecord,
   type AgentThreadRecord,
+  type ConversationAgentRunSpec,
   coerceAgentRunFinishReason,
+  type SubagentRunResult,
+  type SubagentRunSpec,
 } from "@sixb/core/storage"
 import type { SQLClient, SqlParameter } from "../pg-client"
 
@@ -17,7 +20,6 @@ import type { SQLClient, SqlParameter } from "../pg-client"
 export interface AgentThreadRow {
   project_id: string
   id: string
-  agent_id: string
   owner_principal_type: Principal["type"]
   owner_principal_id: string
   title: string | null
@@ -33,9 +35,13 @@ export interface AgentRunRow {
   project_id: string
   id: string
   execution_id: string
-  thread_id: string
-  agent_id: string
-  trigger_message_id: string
+  kind: AgentRunRecord["kind"]
+  thread_id: string | null
+  trigger_message_id: string | null
+  parent_run_id: string | null
+  spawn_key: string | null
+  spec: ConversationAgentRunSpec | SubagentRunSpec | string | null
+  result: SubagentRunResult | string | null
   requester_group_ids: string[] | string
   status: AgentRunRecord["status"]
   model_id: string | null
@@ -89,7 +95,6 @@ export function rowToThreadRecord(row: AgentThreadRow): AgentThreadRecord {
   return {
     id: row.id,
     projectId: row.project_id,
-    agentId: row.agent_id,
     ownerPrincipal: { type: row.owner_principal_type, id: row.owner_principal_id },
     title: row.title ?? undefined,
     status: row.status,
@@ -102,13 +107,10 @@ export function rowToThreadRecord(row: AgentThreadRow): AgentThreadRecord {
 }
 
 export function rowToRunRecord(row: AgentRunRow): AgentRunRecord {
-  return {
+  const base = {
     id: row.id,
     projectId: row.project_id,
     executionId: row.execution_id,
-    threadId: row.thread_id,
-    agentId: row.agent_id,
-    triggerMessageId: row.trigger_message_id,
     requesterGroupIds:
       typeof row.requester_group_ids === "string"
         ? (JSON.parse(row.requester_group_ids) as string[])
@@ -130,6 +132,29 @@ export function rowToRunRecord(row: AgentRunRow): AgentRunRecord {
     startedAt: row.started_at ? new Date(row.started_at) : undefined,
     completedAt: row.completed_at ? new Date(row.completed_at) : undefined,
   }
+  if (row.kind === "conversation") {
+    if (!row.thread_id || !row.trigger_message_id) {
+      throw new Error(`[SixbPg] Conversational Agent run '${row.id}' has invalid persisted fields.`)
+    }
+    return {
+      ...base,
+      kind: "conversation",
+      threadId: row.thread_id,
+      triggerMessageId: row.trigger_message_id,
+      ...(row.spec === null ? {} : { spec: parseJsonColumn<ConversationAgentRunSpec>(row.spec) }),
+    }
+  }
+  if (!row.parent_run_id || !row.spawn_key || row.spec === null) {
+    throw new Error(`[SixbPg] Subagent run '${row.id}' has invalid persisted fields.`)
+  }
+  return {
+    ...base,
+    kind: "subagent",
+    parentRunId: row.parent_run_id,
+    spawnKey: row.spawn_key,
+    spec: parseJsonColumn<SubagentRunSpec>(row.spec),
+    ...(row.result === null ? {} : { result: parseJsonColumn<SubagentRunResult>(row.result) }),
+  }
 }
 
 export function rowToMessageRecord(row: AgentMessageRow): AgentMessageRecord {
@@ -203,6 +228,10 @@ function normalizeDiagnostics(
   return typeof value === "string" ? (JSON.parse(value) as AgentRunDiagnostic[]) : value
 }
 
+function parseJsonColumn<T>(value: unknown): T {
+  return typeof value === "string" ? (JSON.parse(value) as T) : (value as T)
+}
+
 // ── offset pagination (threads / messages order by their own column, not started_at) ─────────────
 
 export async function queryAgentList<TRow>(input: {

package/src/agents/runs.ts

@@ -1,4 +1,10 @@
-import { assertAgentRunExecution } from "@sixb/core/internal/agent-run-storage-provider"
+import {
+  assertAgentRunExecution,
+  assertConversationAgentRunSpec,
+  assertCreateSubagentRunInput,
+  assertSubagentRunResult,
+  subagentRunMatchesCreateInput,
+} from "@sixb/core/internal/agent-run-storage-provider"
 import { normalizeRequesterGroupIds } from "@sixb/core/internal/auth"
 import { serializeSixbFailure } from "@sixb/core/internal/errors"
 import {
@@ -7,7 +13,9 @@ import {
   type AgentRunStore,
   AgentStorageError,
   type ConfirmAgentRunExecutionOwnershipInput,
+  type ConversationAgentRunRecord,
   type CreateAgentRunInput,
+  type CreateSubagentRunInput,
   type ExecutionStorage,
   type FinishAgentRunInput,
   type FinishQueuedAgentRunInput,
@@ -15,6 +23,7 @@ import {
   type ListAgentRunsResult,
   type ReclaimAgentRunInput,
   type StartAgentRunInput,
+  type SubagentRunRecord,
 } from "@sixb/core/storage"
 import type { SQLClient, SqlParameter } from "../pg-client"
 import { appendRunListFilters, hasEmptyStatuses, queryRunList } from "../run-list-query"
@@ -30,14 +39,14 @@ export class PgAgentRunStore implements AgentRunStore {
     private readonly executions: ExecutionStorage
   ) {}
 
-  async create(input: CreateAgentRunInput): Promise<AgentRunRecord> {
+  async create(input: CreateAgentRunInput): Promise<ConversationAgentRunRecord> {
+    assertConversationAgentRunSpec(input.spec, "SixbPg")
     const createdAt = input.createdAt ?? new Date()
     await assertAgentRunExecution({
       executions: this.executions,
       projectId: input.projectId,
       executionId: input.executionId,
       runId: input.id,
-      agentId: input.agentId,
     })
 
     try {
@@ -54,7 +63,6 @@ export class PgAgentRunStore implements AgentRunStore {
             `[SixbPg] Agent thread '${input.threadId}' not found for project '${input.projectId}'.`
           )
         }
-
         if (thread.active_run_id !== null) {
           throw new AgentStorageError(
             "active_run_exists",
@@ -67,9 +75,10 @@ export class PgAgentRunStore implements AgentRunStore {
             project_id,
             id,
             execution_id,
+            kind,
             thread_id,
-            agent_id,
             trigger_message_id,
+            spec,
             requester_group_ids,
             status,
             attempt,
@@ -78,9 +87,10 @@ export class PgAgentRunStore implements AgentRunStore {
             ${input.projectId},
             ${input.id},
             ${input.executionId},
+            ${"conversation"},
             ${input.threadId},
-            ${input.agentId},
             ${input.triggerMessageId},
+            ${JSON.stringify(input.spec)}::text::jsonb,
             ${JSON.stringify(normalizeRequesterGroupIds(input.requesterGroupIds))}::text::jsonb,
             ${"queued"},
             ${0},
@@ -94,7 +104,11 @@ export class PgAgentRunStore implements AgentRunStore {
           WHERE project_id = ${input.projectId} AND id = ${input.threadId}
         `
 
-        return rowToRunRecord(row)
+        const created = rowToRunRecord(row)
+        if (created.kind !== "conversation") {
+          throw new Error(`[SixbPg] Agent run insert '${input.id}' returned the wrong run kind.`)
+        }
+        return created
       })
     } catch (error) {
       if (isUniqueViolation(error)) {
@@ -108,6 +122,108 @@ export class PgAgentRunStore implements AgentRunStore {
     }
   }
 
+  async createSubagent(input: CreateSubagentRunInput): Promise<SubagentRunRecord> {
+    assertCreateSubagentRunInput(input, "SixbPg")
+    const execution = await assertAgentRunExecution({
+      executions: this.executions,
+      projectId: input.projectId,
+      executionId: input.executionId,
+      runId: input.id,
+    })
+
+    try {
+      return await runPgTransaction(this.sql, async (tx) => {
+        const [parent] = await tx<AgentRunRow[]>`
+          SELECT * FROM agent_runs
+          WHERE project_id = ${input.projectId} AND id = ${input.parentRunId}
+          FOR UPDATE
+        `
+        if (!parent || parent.kind !== "conversation") {
+          throw new AgentStorageError(
+            "run_not_found",
+            `[SixbPg] Parent Agent run '${input.parentRunId}' was not found.`
+          )
+        }
+
+        const [existingRow] = await tx<AgentRunRow[]>`
+          SELECT * FROM agent_runs
+          WHERE project_id = ${input.projectId} AND id = ${input.id}
+        `
+        if (existingRow) {
+          const existing = rowToRunRecord(existingRow)
+          if (subagentRunMatchesCreateInput(existing, input)) return existing
+          throw new AgentStorageError(
+            "duplicate_id",
+            `[SixbPg] Subagent run '${input.id}' already exists with different immutable inputs.`
+          )
+        }
+        if (parent.status !== "running") {
+          throw new AgentStorageError(
+            "invalid_state",
+            `[SixbPg] Parent Agent run '${parent.id}' is not running (status '${parent.status}').`
+          )
+        }
+        if (
+          execution.source.type !== "execution" ||
+          execution.source.executionId !== parent.execution_id
+        ) {
+          throw new AgentStorageError(
+            "invalid_input",
+            `[SixbPg] Subagent execution '${execution.id}' is not a child of parent execution '${parent.execution_id}'.`
+          )
+        }
+        if (parent.execution_token !== input.parentExecutionToken) {
+          throw new AgentStorageError(
+            "execution_lost",
+            `[SixbPg] Parent Agent run '${parent.id}' is no longer owned by this execution.`
+          )
+        }
+        const parentRecord = rowToRunRecord(parent)
+
+        const [countRow] = await tx<{ count: string | number }[]>`
+          SELECT COUNT(*)::bigint AS count FROM agent_runs
+          WHERE project_id = ${input.projectId}
+            AND parent_run_id = ${input.parentRunId}
+            AND status IN (${"queued"}, ${"running"})
+        `
+        const activeChildren = Number(countRow?.count ?? 0)
+        if (activeChildren >= input.maxActiveChildren) {
+          throw new AgentStorageError(
+            "active_child_limit",
+            `[SixbPg] Agent run '${parent.id}' already has ${activeChildren} active subagents.`
+          )
+        }
+
+        const [row] = await tx<AgentRunRow[]>`
+          INSERT INTO agent_runs (
+            project_id, id, execution_id, kind, parent_run_id, spawn_key, spec,
+            requester_group_ids, status, attempt, created_at
+          ) VALUES (
+            ${input.projectId}, ${input.id}, ${input.executionId}, ${"subagent"},
+            ${input.parentRunId}, ${input.spawnKey},
+            ${JSON.stringify(input.spec)}::text::jsonb,
+            ${JSON.stringify(parentRecord.requesterGroupIds)}::text::jsonb,
+            ${"queued"}, ${0}, ${input.createdAt ?? new Date()}
+          )
+          RETURNING *
+        `
+        const created = rowToRunRecord(row)
+        if (created.kind !== "subagent") {
+          throw new Error(`[SixbPg] Subagent insert '${input.id}' returned the wrong run kind.`)
+        }
+        return created
+      })
+    } catch (error) {
+      if (isUniqueViolation(error)) {
+        throw new AgentStorageError(
+          "duplicate_id",
+          `[SixbPg] Subagent run '${input.id}' conflicts with an existing durable spawn.`
+        )
+      }
+      throw error
+    }
+  }
+
   async start(input: StartAgentRunInput): Promise<AgentRunRecord> {
     const startedAt = input.startedAt ?? new Date()
     return runPgTransaction(this.sql, async (tx) => {
@@ -198,6 +314,20 @@ export class PgAgentRunStore implements AgentRunStore {
     return runPgTransaction(this.sql, async (tx) => {
       const run = await this.lockRunning(tx, input.projectId, input.id)
 
+      if (input.status === "succeeded" && run.kind === "subagent") {
+        assertSubagentRunResult(input.result, run.id, "SixbPg")
+      }
+      if (
+        input.status === "succeeded" &&
+        run.kind === "conversation" &&
+        input.result !== undefined
+      ) {
+        throw new AgentStorageError(
+          "invalid_input",
+          `[SixbPg] Conversational Agent run '${run.id}' cannot persist a subagent result.`
+        )
+      }
+
       if (run.execution_token !== input.executionToken) {
         throw new AgentStorageError(
           "execution_lost",
@@ -213,6 +343,7 @@ export class PgAgentRunStore implements AgentRunStore {
           finish_reason = ${input.finishReason ?? null},
           error = ${errorValue}::text::jsonb,
           diagnostics = ${input.diagnostics === undefined ? null : JSON.stringify(input.diagnostics)}::text::jsonb,
+          result = ${input.status === "succeeded" && input.result !== undefined ? JSON.stringify(input.result) : null}::text::jsonb,
           execution_token = ${null},
           execution_queue_lease_expires_at = ${null},
           completed_at = ${completedAt}
@@ -262,7 +393,7 @@ export class PgAgentRunStore implements AgentRunStore {
   }
 
   async list(input: ListAgentRunsInput): Promise<ListAgentRunsResult> {
-    if (hasEmptyStatuses(input)) {
+    if (hasEmptyStatuses(input) || input.kinds?.length === 0) {
       return { runs: [], hasMore: false, total: 0 }
     }
 
@@ -270,14 +401,19 @@ export class PgAgentRunStore implements AgentRunStore {
     const params: SqlParameter[] = [input.projectId]
     let index = 2
 
+    if (input.kinds) {
+      whereClauses.push(`kind = ANY($${index++}::text[])`)
+      params.push(input.kinds as string[])
+    }
+
     if (input.threadId) {
       whereClauses.push(`thread_id = $${index++}`)
       params.push(input.threadId)
     }
 
-    if (input.agentId) {
-      whereClauses.push(`agent_id = $${index++}`)
-      params.push(input.agentId)
+    if (input.parentRunId) {
+      whereClauses.push(`parent_run_id = $${index++}`)
+      params.push(input.parentRunId)
     }
 
     index = appendRunListFilters(
@@ -336,6 +472,7 @@ export class PgAgentRunStore implements AgentRunStore {
   }
 
   private async releaseThread(sql: SQLClient, run: AgentRunRow, completedAt: Date): Promise<void> {
+    if (run.kind !== "conversation") return
     await sql`
       UPDATE agent_threads SET active_run_id = ${null}, updated_at = ${completedAt}
       WHERE project_id = ${run.project_id} AND id = ${run.thread_id} AND active_run_id = ${run.id}

package/src/agents/threads.ts

@@ -23,7 +23,6 @@ export class PgAgentThreadStore implements AgentThreadStore {
         INSERT INTO agent_threads (
           project_id,
           id,
-          agent_id,
           owner_principal_type,
           owner_principal_id,
           title,
@@ -33,7 +32,6 @@ export class PgAgentThreadStore implements AgentThreadStore {
         ) VALUES (
           ${input.projectId},
           ${input.id},
-          ${input.agentId},
           ${input.ownerPrincipal.type},
           ${input.ownerPrincipal.id},
           ${input.title ?? null},
@@ -66,10 +64,7 @@ export class PgAgentThreadStore implements AgentThreadStore {
   }
 
   async list(input: ListAgentThreadsInput): Promise<ListAgentThreadsResult> {
-    if (
-      (input.statuses !== undefined && input.statuses.length === 0) ||
-      (input.agentIds !== undefined && input.agentIds.length === 0)
-    ) {
+    if (input.statuses !== undefined && input.statuses.length === 0) {
       return { threads: [], hasMore: false, total: 0 }
     }
 
@@ -77,17 +72,6 @@ export class PgAgentThreadStore implements AgentThreadStore {
     const params: SqlParameter[] = [input.projectId]
     let index = 2
 
-    if (input.agentId) {
-      whereClauses.push(`agent_id = $${index++}`)
-      params.push(input.agentId)
-    }
-
-    if (input.agentIds) {
-      const placeholders = input.agentIds.map(() => `$${index++}`)
-      whereClauses.push(`agent_id IN (${placeholders.join(", ")})`)
-      params.push(...input.agentIds)
-    }
-
     if (input.statuses) {
       const placeholders = input.statuses.map(() => `$${index++}`)
       whereClauses.push(`status IN (${placeholders.join(", ")})`)

package/src/ai-cost-groups.ts

@@ -0,0 +1,102 @@
+import {
+  type AiModelCallGroupRow,
+  aiModelCallGroupFragmentFromRow,
+  buildAiModelCallGroupsFromFragments,
+  normalizeAiModelCallAccountingQuery,
+} from "@sixb/core/internal/ai-cost-storage-provider"
+import type { ListAiModelCallGroupsInput, ListAiModelCallGroupsResult } from "@sixb/core/storage"
+import type { PgStoreClient } from "./transactions"
+
+/** Paginate initiating executions, then aggregate their matching calls without loading usage payloads. */
+export async function listAiModelCallGroups(
+  sql: PgStoreClient,
+  input: ListAiModelCallGroupsInput
+): Promise<ListAiModelCallGroupsResult> {
+  const query = normalizeAiModelCallAccountingQuery(input)
+  const parameters = [
+    query.projectId,
+    query.from.toISOString(),
+    query.to.toISOString(),
+    query.providerId ?? null,
+    query.modelId ?? null,
+    query.valuationStatus ?? null,
+  ]
+  const [count] = await sql.unsafe<{ total: string }[]>(COUNT_SQL, parameters)
+  const rows = await sql.unsafe<AiModelCallGroupRow[]>(PAGE_SQL, [
+    ...parameters,
+    query.limit,
+    query.offset,
+  ])
+  const total = Number(count?.total)
+  if (!Number.isSafeInteger(total) || total < 0)
+    throw new RangeError("[Sixb] Invalid model-call group count.")
+  const items = buildAiModelCallGroupsFromFragments(rows.map(aiModelCallGroupFragmentFromRow))
+  return { items, total, hasMore: query.offset + items.length < total }
+}
+
+const FILTERED_SQL = `
+WITH filtered AS (
+  SELECT COALESCE(root_agent.execution_id, usage.execution_id) AS root_execution_id,
+    usage.execution_id, usage.provider_id, usage.requested_model_id, usage.occurred_at,
+    usage.total_tokens,
+    COALESCE(cost.status, 'unvalued') AS valuation_status,
+    cost.currency AS amount_currency, cost.amount_nanos,
+    root_agent.id AS root_agent_run_id,
+    root_agent.thread_id AS root_thread_id,
+    direct_agent.kind AS direct_kind,
+    direct_agent.id AS direct_run_id,
+    direct_agent.parent_run_id AS parent_run_id,
+    workflow_node.node_id AS workflow_agent_step_id,
+    workflow_agent.node_run_id AS workflow_node_run_id,
+    workflow_node.workflow_id AS workflow_id,
+    workflow_node.workflow_run_id AS workflow_run_id
+  FROM ai_model_call_usage AS usage
+  LEFT JOIN agent_runs AS direct_agent
+    ON direct_agent.project_id = usage.project_id AND direct_agent.execution_id = usage.execution_id
+  LEFT JOIN agent_runs AS root_agent
+    ON root_agent.project_id = usage.project_id
+    AND root_agent.id = COALESCE(direct_agent.parent_run_id, direct_agent.id)
+  LEFT JOIN workflow_agent_node_runs AS workflow_agent
+    ON workflow_agent.project_id = usage.project_id AND workflow_agent.execution_id = usage.execution_id
+  LEFT JOIN workflow_node_runs AS workflow_node
+    ON workflow_node.project_id = workflow_agent.project_id AND workflow_node.id = workflow_agent.node_run_id
+  LEFT JOIN ai_model_call_valuations AS cost
+    ON cost.project_id = usage.project_id AND cost.usage_record_id = usage.id
+  WHERE usage.project_id = $1
+    AND usage.occurred_at >= $2::timestamptz
+    AND usage.occurred_at < $3::timestamptz
+    AND ($4::text IS NULL OR usage.provider_id = $4)
+    AND ($5::text IS NULL OR usage.requested_model_id = $5)
+    AND ($6::text IS NULL OR COALESCE(cost.status, 'unvalued') = $6)
+), roots AS (
+  SELECT root_execution_id, MAX(occurred_at) AS last_call_at
+  FROM filtered GROUP BY root_execution_id
+)
+`
+const COUNT_SQL = `${FILTERED_SQL} SELECT CAST(COUNT(*) AS TEXT) AS total FROM roots`
+const PAGE_SQL = `${FILTERED_SQL}, page AS (
+  SELECT root_execution_id FROM roots
+  ORDER BY last_call_at DESC, root_execution_id COLLATE "C" ASC LIMIT $7 OFFSET $8
+)
+SELECT filtered.root_execution_id, execution_id, provider_id, requested_model_id,
+  MIN(occurred_at) AS first_call_at, MAX(occurred_at) AS last_call_at,
+  CAST(COUNT(*) AS TEXT) AS call_count,
+  CAST(COUNT(total_tokens) AS TEXT) AS token_count,
+  CAST(SUM(total_tokens) AS TEXT) AS total_tokens,
+  valuation_status, amount_currency,
+  CAST(SUM(amount_nanos / 1000000000000) AS TEXT) AS amount_high,
+  CAST(SUM((amount_nanos / 1000000) % 1000000) AS TEXT) AS amount_middle,
+  CAST(SUM(amount_nanos % 1000000) AS TEXT) AS amount_low,
+  MAX(root_agent_run_id) AS root_agent_run_id,
+  MAX(root_thread_id) AS root_thread_id,
+  MAX(direct_kind) AS direct_kind,
+  MAX(direct_run_id) AS direct_run_id,
+  MAX(parent_run_id) AS parent_run_id,
+  MAX(workflow_agent_step_id) AS workflow_agent_step_id,
+  MAX(workflow_node_run_id) AS workflow_node_run_id,
+  MAX(workflow_id) AS workflow_id,
+  MAX(workflow_run_id) AS workflow_run_id
+FROM filtered JOIN page ON page.root_execution_id = filtered.root_execution_id
+GROUP BY filtered.root_execution_id, execution_id, provider_id, requested_model_id,
+  valuation_status, amount_currency
+`