@pepys/source-telemetry 0.0.1 → 0.1.1

package/README.md

@@ -28,6 +28,8 @@ bun add @pepys/source-telemetry
 - `telemetryPlugin`: the plug-in, to add to a deployment.
 - `telemetryModelRequestConsumer` and `telemetryTurnCostConsumer`: the consumers of the log record
   facts, with their SQL schemas.
+- `sessionModelRequestsQuery`: the query `telemetry_session_model_requests`, which returns the model
+  requests of one session that the named sender stored.
 
 ## Documentation
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pepys/source-telemetry",
-  "version": "0.0.1",
+  "version": "0.1.1",
   "description": "The pepys telemetry plug-in: the OTLP log records of coding agents, with their model requests and turn costs",
   "license": "MIT",
   "repository": {
@@ -19,14 +19,14 @@
     "access": "public"
   },
   "dependencies": {
-    "@pepys/kit": "0.0.1",
+    "@pepys/kit": "0.1.1",
     "@zgeoff/std": "0.4.3",
     "tiny-invariant": "1.3.3",
     "zod": "4.6.7"
   },
   "devDependencies": {
-    "@pepys/cli": "0.0.1",
-    "@pepys/ingest": "0.0.1",
-    "@pepys/testing": "0.0.1"
+    "@pepys/cli": "0.1.1",
+    "@pepys/ingest": "0.1.1",
+    "@pepys/testing": "0.1.1"
   }
 }

package/src/index.ts

@@ -20,6 +20,7 @@ export { defineTelemetryProduct } from './products/define-telemetry-product.ts';
 export { products } from './products/products.ts';
 export type { ProductName } from './products/products.ts';
 export type { TelemetryProduct } from './products/types.ts';
+export { sessionModelRequestsQuery } from './queries/session-model-requests-query.ts';
 export { telemetryModelRequestConsumer } from './telemetry-model-request-consumer.ts';
 export { telemetryPlugin, telemetrySource } from './telemetry-plugin.ts';
 export { telemetryTurnCostConsumer } from './telemetry-turn-cost-consumer.ts';

package/src/queries/session-model-requests-query.ts

@@ -0,0 +1,55 @@
+import { defineQuery } from '@pepys/kit';
+import type { EntityID } from '@pepys/kit';
+import { toTagged } from '@zgeoff/std';
+import { z } from 'zod';
+import { telemetryModelRequestConsumer } from '../telemetry-model-request-consumer.ts';
+
+const modelRequestSchema = z.object({
+  id: z.string(),
+  product: z.string(),
+  model: z.string(),
+  input_tokens: z.number(),
+  cache_read_tokens: z.number(),
+  cache_creation_tokens: z.number(),
+  output_tokens: z.number(),
+  cost_usd: z.number().nullable(),
+  at: z.string(),
+});
+
+// The sender writes the session of a record, so a row counts only when the caller names the sender
+// that owns the session, as another plug-in's fact gives it.
+export const sessionModelRequestsQuery = defineQuery({
+  name: 'telemetry_session_model_requests',
+  description:
+    'Returns the model requests of one telemetry session, oldest first, that the named sender stored: the sender is the issuer and the subject of the identity that proved each export. A row from any other sender is left out.',
+  input: z.object({
+    session: z.string().min(1),
+    issuer: z.string().min(1),
+    subject: z.string().min(1),
+  }),
+  output: z.array(modelRequestSchema),
+  reads: [telemetryModelRequestConsumer],
+  handler: async (call) => {
+    const rows = await call.db
+      .selectFrom('telemetry_model_request')
+      .select([
+        'id',
+        'product',
+        'model',
+        'input_tokens',
+        'cache_read_tokens',
+        'cache_creation_tokens',
+        'output_tokens',
+        'cost_usd',
+        'at',
+      ])
+      .where('session_id', '=', toTagged<EntityID>(call.input.session))
+      .where('origin_issuer', '=', call.input.issuer)
+      .where('origin_subject', '=', call.input.subject)
+      .orderBy('at')
+      .orderBy('id')
+      .execute();
+
+    return rows;
+  },
+});

package/src/telemetry-plugin.ts

@@ -2,6 +2,7 @@ import { definePlugin, defineSource } from '@pepys/kit';
 import { logRecordFamily } from './log-record-family.ts';
 import { receiveOtlpLogs } from './otlp/receive-otlp-logs.ts';
 import { products } from './products/products.ts';
+import { sessionModelRequestsQuery } from './queries/session-model-requests-query.ts';
 import { telemetryModelRequestConsumer } from './telemetry-model-request-consumer.ts';
 import { telemetryTurnCostConsumer } from './telemetry-turn-cost-consumer.ts';
 
@@ -39,4 +40,5 @@ export const telemetryPlugin = definePlugin({
   name: 'telemetry',
   sources: [telemetrySource],
   consumers: [telemetryModelRequestConsumer, telemetryTurnCostConsumer],
+  queries: [sessionModelRequestsQuery],
 });