types.ts

import { z } from 'zod';
import type { Pool } from 'pg';

export interface ServerContext extends Record<string, unknown> {
  pgPool: Pool;
  schema: string;
}

export const zSource = z
  .string()
  .min(0)
  .nullable()
  .describe(
    'The source or origin of this memory. A deep URI to the origin of the fact is preferred (e.g., a specific URL, file path, or reference).',
  );

export const zMemory = z.object({
  id: z.string().describe('The unique identifier of this memory.'),
  content: z.string().describe('The content of this memory.'),
  source: zSource,
  created_at: z
    .date()
    .describe('The date and time when this memory was created.'),
  updated_at: z
    .date()
    .describe('The date and time when this memory was last updated.'),
});

export type Memory = z.infer<typeof zMemory>;

export const zKey = z
  .string()
  .min(1)
  .describe(
    'A unique identifier for the target set of memories. Can be any combination of user and application ids, as needed for scoping and personalization.',
  );