@yugenlab/vaayu 0.1.0 → 0.1.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -133,8 +133,12 @@ async function runMigrations(pool) {
133
133
  await pool.query("ALTER TABLE agent_runs ADD COLUMN IF NOT EXISTS agent_id TEXT");
134
134
  await pool.query("ALTER TABLE agent_runs ADD COLUMN IF NOT EXISTS parent_agent_id TEXT");
135
135
  await pool.query(
136
- "CREATE TABLE IF NOT EXISTS session_routes (session_id TEXT PRIMARY KEY, agent_id TEXT, lane_id TEXT, queue_mode TEXT, announce_mode TEXT, updated_at TEXT)"
136
+ "CREATE TABLE IF NOT EXISTS session_routes (session_id TEXT PRIMARY KEY, agent_id TEXT, lane_id TEXT, queue_mode TEXT, announce_mode TEXT, pin_provider_id TEXT, pin_model TEXT, pin_reason TEXT, pin_until TEXT, updated_at TEXT)"
137
137
  );
138
+ await pool.query("ALTER TABLE session_routes ADD COLUMN IF NOT EXISTS pin_provider_id TEXT");
139
+ await pool.query("ALTER TABLE session_routes ADD COLUMN IF NOT EXISTS pin_model TEXT");
140
+ await pool.query("ALTER TABLE session_routes ADD COLUMN IF NOT EXISTS pin_reason TEXT");
141
+ await pool.query("ALTER TABLE session_routes ADD COLUMN IF NOT EXISTS pin_until TEXT");
138
142
  await pool.query(
139
143
  "CREATE INDEX IF NOT EXISTS session_routes_updated_idx ON session_routes(updated_at)"
140
144
  );
@@ -1238,14 +1242,30 @@ var createSessionInboxStore = (pool) => {
1238
1242
  };
1239
1243
 
1240
1244
  // packages/storage/src/postgres/session-routes.ts
1241
- var selectRoute = "SELECT session_id, agent_id, lane_id, queue_mode, announce_mode, updated_at ";
1245
+ var selectRoute = "SELECT session_id, agent_id, lane_id, queue_mode, announce_mode, pin_provider_id, pin_model, pin_reason, pin_until, updated_at ";
1246
+ var mapQueueMode = (value) => {
1247
+ if (value === "followup" || value === "collect" || value === "steer" || value === "drop" || value === "reject") {
1248
+ return value;
1249
+ }
1250
+ return void 0;
1251
+ };
1252
+ var mapAnnounceMode = (value) => {
1253
+ if (value === "hub" || value === "channel" || value === "none") {
1254
+ return value;
1255
+ }
1256
+ return void 0;
1257
+ };
1242
1258
  var mapRoute = (row) => {
1243
1259
  return {
1244
1260
  sessionId: String(row.session_id),
1245
1261
  agentId: row.agent_id ? String(row.agent_id) : void 0,
1246
1262
  laneId: row.lane_id ? String(row.lane_id) : void 0,
1247
- queueMode: row.queue_mode ? String(row.queue_mode) : void 0,
1248
- announceMode: row.announce_mode ? String(row.announce_mode) : void 0,
1263
+ queueMode: mapQueueMode(row.queue_mode),
1264
+ announceMode: mapAnnounceMode(row.announce_mode),
1265
+ pinProviderId: row.pin_provider_id ? String(row.pin_provider_id) : void 0,
1266
+ pinModel: row.pin_model ? String(row.pin_model) : void 0,
1267
+ pinReason: row.pin_reason ? String(row.pin_reason) : void 0,
1268
+ pinUntil: row.pin_until ? String(row.pin_until) : void 0,
1249
1269
  updatedAt: row.updated_at ? String(row.updated_at) : (/* @__PURE__ */ new Date()).toISOString()
1250
1270
  };
1251
1271
  };
@@ -1260,13 +1280,17 @@ var createSessionRouteStore = (pool) => {
1260
1280
  async setSessionRoute(sessionId, route) {
1261
1281
  const updatedAt = (/* @__PURE__ */ new Date()).toISOString();
1262
1282
  const result = await pool.query(
1263
- `INSERT INTO session_routes (session_id, agent_id, lane_id, queue_mode, announce_mode, updated_at) VALUES ($1,$2,$3,$4,$5,$6) ON CONFLICT (session_id) DO UPDATE SET agent_id = EXCLUDED.agent_id, lane_id = EXCLUDED.lane_id, queue_mode = EXCLUDED.queue_mode, announce_mode = EXCLUDED.announce_mode, updated_at = EXCLUDED.updated_at RETURNING ${selectRoute.replace("SELECT ", "")}`,
1283
+ `INSERT INTO session_routes (session_id, agent_id, lane_id, queue_mode, announce_mode, pin_provider_id, pin_model, pin_reason, pin_until, updated_at) VALUES ($1,$2,$3,$4,$5,$6,$7,$8,$9,$10) ON CONFLICT (session_id) DO UPDATE SET agent_id = EXCLUDED.agent_id, lane_id = EXCLUDED.lane_id, queue_mode = EXCLUDED.queue_mode, announce_mode = EXCLUDED.announce_mode, pin_provider_id = EXCLUDED.pin_provider_id, pin_model = EXCLUDED.pin_model, pin_reason = EXCLUDED.pin_reason, pin_until = EXCLUDED.pin_until, updated_at = EXCLUDED.updated_at RETURNING ${selectRoute.replace("SELECT ", "")}`,
1264
1284
  [
1265
1285
  sessionId,
1266
1286
  route.agentId ?? null,
1267
1287
  route.laneId ?? null,
1268
1288
  route.queueMode ?? null,
1269
1289
  route.announceMode ?? null,
1290
+ route.pinProviderId ?? null,
1291
+ route.pinModel ?? null,
1292
+ route.pinReason ?? null,
1293
+ route.pinUntil ?? null,
1270
1294
  updatedAt
1271
1295
  ]
1272
1296
  );
@@ -1476,4 +1500,4 @@ var PostgresStorage = class _PostgresStorage {
1476
1500
  export {
1477
1501
  PostgresStorage
1478
1502
  };
1479
- //# sourceMappingURL=chunk-MINFB5LT.js.map
1503
+ //# sourceMappingURL=chunk-C76USAC5.js.map
@@ -118,10 +118,30 @@ function initAgentSchema(db) {
118
118
  lane_id TEXT,
119
119
  queue_mode TEXT,
120
120
  announce_mode TEXT,
121
+ pin_provider_id TEXT,
122
+ pin_model TEXT,
123
+ pin_reason TEXT,
124
+ pin_until TEXT,
121
125
  updated_at TEXT
122
126
  )
123
127
  `);
124
128
  db.exec(`CREATE INDEX IF NOT EXISTS session_routes_updated_idx ON session_routes(updated_at)`);
129
+ try {
130
+ db.exec(`ALTER TABLE session_routes ADD COLUMN pin_provider_id TEXT`);
131
+ } catch {
132
+ }
133
+ try {
134
+ db.exec(`ALTER TABLE session_routes ADD COLUMN pin_model TEXT`);
135
+ } catch {
136
+ }
137
+ try {
138
+ db.exec(`ALTER TABLE session_routes ADD COLUMN pin_reason TEXT`);
139
+ } catch {
140
+ }
141
+ try {
142
+ db.exec(`ALTER TABLE session_routes ADD COLUMN pin_until TEXT`);
143
+ } catch {
144
+ }
125
145
  db.exec(`
126
146
  CREATE TABLE IF NOT EXISTS agent_run_events (
127
147
  id TEXT PRIMARY KEY,
@@ -2353,23 +2373,39 @@ async function listSessionInbox(db, options) {
2353
2373
  function prepareSessionRouteStatements(db) {
2354
2374
  return {
2355
2375
  get: db.prepare(
2356
- "SELECT session_id, agent_id, lane_id, queue_mode, announce_mode, updated_at FROM session_routes WHERE session_id = @session_id"
2376
+ "SELECT session_id, agent_id, lane_id, queue_mode, announce_mode, pin_provider_id, pin_model, pin_reason, pin_until, updated_at FROM session_routes WHERE session_id = @session_id"
2357
2377
  ),
2358
2378
  upsert: db.prepare(
2359
- "INSERT INTO session_routes (session_id, agent_id, lane_id, queue_mode, announce_mode, updated_at) VALUES (@session_id, @agent_id, @lane_id, @queue_mode, @announce_mode, @updated_at) ON CONFLICT(session_id) DO UPDATE SET agent_id = excluded.agent_id, lane_id = excluded.lane_id, queue_mode = excluded.queue_mode, announce_mode = excluded.announce_mode, updated_at = excluded.updated_at"
2379
+ "INSERT INTO session_routes (session_id, agent_id, lane_id, queue_mode, announce_mode, pin_provider_id, pin_model, pin_reason, pin_until, updated_at) VALUES (@session_id, @agent_id, @lane_id, @queue_mode, @announce_mode, @pin_provider_id, @pin_model, @pin_reason, @pin_until, @updated_at) ON CONFLICT(session_id) DO UPDATE SET agent_id = excluded.agent_id, lane_id = excluded.lane_id, queue_mode = excluded.queue_mode, announce_mode = excluded.announce_mode, pin_provider_id = excluded.pin_provider_id, pin_model = excluded.pin_model, pin_reason = excluded.pin_reason, pin_until = excluded.pin_until, updated_at = excluded.updated_at"
2360
2380
  ),
2361
2381
  list: db.prepare(
2362
- "SELECT session_id, agent_id, lane_id, queue_mode, announce_mode, updated_at FROM session_routes ORDER BY updated_at DESC LIMIT @limit"
2382
+ "SELECT session_id, agent_id, lane_id, queue_mode, announce_mode, pin_provider_id, pin_model, pin_reason, pin_until, updated_at FROM session_routes ORDER BY updated_at DESC LIMIT @limit"
2363
2383
  )
2364
2384
  };
2365
2385
  }
2386
+ var mapQueueMode = (value) => {
2387
+ if (value === "followup" || value === "collect" || value === "steer" || value === "drop" || value === "reject") {
2388
+ return value;
2389
+ }
2390
+ return void 0;
2391
+ };
2392
+ var mapAnnounceMode = (value) => {
2393
+ if (value === "hub" || value === "channel" || value === "none") {
2394
+ return value;
2395
+ }
2396
+ return void 0;
2397
+ };
2366
2398
  var mapRoute = (row) => {
2367
2399
  return {
2368
2400
  sessionId: String(row.session_id),
2369
2401
  agentId: row.agent_id ? String(row.agent_id) : void 0,
2370
2402
  laneId: row.lane_id ? String(row.lane_id) : void 0,
2371
- queueMode: row.queue_mode ? String(row.queue_mode) : void 0,
2372
- announceMode: row.announce_mode ? String(row.announce_mode) : void 0,
2403
+ queueMode: mapQueueMode(row.queue_mode),
2404
+ announceMode: mapAnnounceMode(row.announce_mode),
2405
+ pinProviderId: row.pin_provider_id ? String(row.pin_provider_id) : void 0,
2406
+ pinModel: row.pin_model ? String(row.pin_model) : void 0,
2407
+ pinReason: row.pin_reason ? String(row.pin_reason) : void 0,
2408
+ pinUntil: row.pin_until ? String(row.pin_until) : void 0,
2373
2409
  updatedAt: row.updated_at ? String(row.updated_at) : (/* @__PURE__ */ new Date()).toISOString()
2374
2410
  };
2375
2411
  };
@@ -2384,6 +2420,10 @@ async function upsertSessionRoute(s, record) {
2384
2420
  lane_id: record.laneId ?? null,
2385
2421
  queue_mode: record.queueMode ?? null,
2386
2422
  announce_mode: record.announceMode ?? null,
2423
+ pin_provider_id: record.pinProviderId ?? null,
2424
+ pin_model: record.pinModel ?? null,
2425
+ pin_reason: record.pinReason ?? null,
2426
+ pin_until: record.pinUntil ?? null,
2387
2427
  updated_at: record.updatedAt
2388
2428
  });
2389
2429
  return record;
@@ -2750,4 +2790,4 @@ export {
2750
2790
  deleteReaction,
2751
2791
  SqliteStorage
2752
2792
  };
2753
- //# sourceMappingURL=chunk-SMVJRPAH.js.map
2793
+ //# sourceMappingURL=chunk-PRXQW76U.js.map
@@ -1,8 +1,8 @@
1
1
  import {
2
2
  PostgresStorage
3
- } from "./chunk-MINFB5LT.js";
3
+ } from "./chunk-C76USAC5.js";
4
4
  import "./chunk-IGKYKEKT.js";
5
5
  export {
6
6
  PostgresStorage
7
7
  };
8
- //# sourceMappingURL=postgres-3ZXBYTPC.js.map
8
+ //# sourceMappingURL=postgres-WLH3D5HG.js.map
@@ -72,7 +72,7 @@ import {
72
72
  updateSession,
73
73
  updateSession2,
74
74
  updateSyncState
75
- } from "./chunk-SMVJRPAH.js";
75
+ } from "./chunk-PRXQW76U.js";
76
76
  import "./chunk-IGKYKEKT.js";
77
77
  export {
78
78
  SqliteStorage,
@@ -149,4 +149,4 @@ export {
149
149
  updateSession,
150
150
  updateSyncState
151
151
  };
152
- //# sourceMappingURL=sqlite-JPF5TICX.js.map
152
+ //# sourceMappingURL=sqlite-DHUQGPR5.js.map