get-observations-by-ids-relevance.test.ts

import { describe, it, expect, beforeEach, afterEach } from 'bun:test';
import { SessionStore } from '../../../src/services/sqlite/SessionStore.js';

describe('SessionStore.*ByIds — orderBy: "relevance" preserves caller ID order (#2153)', () => {
  let store: SessionStore;

  beforeEach(() => {
    store = new SessionStore(':memory:');
  });

  afterEach(() => {
    store.close();
  });

  it('getObservationsByIds returns rows in caller-provided ID order when orderBy is "relevance"', () => {
    const sdkId = store.createSDKSession('content-relevance', 'p', 'prompt');
    store.updateMemorySessionId(sdkId, 'session-relevance');

    const baseTs = 1_700_000_000_000;
    const inserted: number[] = [];
    for (let i = 0; i < 5; i++) {
      const result = store.storeObservations(
        'session-relevance',
        'p',
        [{
          type: 'test',
          title: `obs-${i}`,
          subtitle: null,
          facts: [`fact ${i}`],
          narrative: null,
          concepts: [],
          files_read: [],
          files_modified: [],
        }],
        null,
        i,
        0,
        baseTs + i * 1000,
      );
      inserted.push(result.observationIds[0]);
    }

    const callerOrder = [...inserted].reverse();
    const results = store.getObservationsByIds(callerOrder, { orderBy: 'relevance' });

    expect(results.map(r => r.id)).toEqual(callerOrder);
  });

  it('applies a relevance limit after restoring the caller-provided ID order', () => {
    const sdkId = store.createSDKSession('content-relevance-limit', 'p', 'prompt');
    store.updateMemorySessionId(sdkId, 'session-relevance-limit');

    const baseTs = 1_700_000_000_000;
    const inserted: number[] = [];
    for (let i = 0; i < 5; i++) {
      const result = store.storeObservations(
        'session-relevance-limit',
        'p',
        [{
          type: 'test',
          title: `obs-limit-${i}`,
          subtitle: null,
          facts: [`fact ${i}`],
          narrative: null,
          concepts: [],
          files_read: [],
          files_modified: [],
        }],
        null,
        i,
        0,
        baseTs + i * 1000,
      );
      inserted.push(result.observationIds[0]);
    }

    const callerOrder = [...inserted].reverse();
    const results = store.getObservationsByIds(callerOrder, { orderBy: 'relevance', limit: 2 });

    expect(results.map(r => r.id)).toEqual(callerOrder.slice(0, 2));
  });

  it('getObservationsByIds still respects date_desc when orderBy defaults', () => {
    const sdkId = store.createSDKSession('content-date', 'p', 'prompt');
    store.updateMemorySessionId(sdkId, 'session-date');
    const baseTs = 1_700_000_000_000;
    const inserted: number[] = [];
    for (let i = 0; i < 3; i++) {
      const result = store.storeObservations(
        'session-date',
        'p',
        [{
          type: 'test',
          title: `obs-${i}`,
          subtitle: null,
          facts: [`fact ${i}`],
          narrative: null,
          concepts: [],
          files_read: [],
          files_modified: [],
        }],
        null,
        i,
        0,
        baseTs + i * 1000,
      );
      inserted.push(result.observationIds[0]);
    }

    const callerOrder = [...inserted].reverse(); 
    const results = store.getObservationsByIds(callerOrder);
    expect(results.map(r => r.id)).toEqual([...inserted].reverse());
  });
});