@goscribe/server 1.3.4 → 1.5.0

package/prisma/schema.prisma

@@ -65,6 +65,19 @@ enum ActivityLogStatus {
   FAILURE
 }
 
+enum KnowledgeBaseStatus {
+  READY
+  INDEXING
+  ERROR
+}
+
+enum KnowledgeBaseDocumentStatus {
+  PENDING
+  PROCESSING
+  READY
+  FAILED
+}
+
 //
 // NextAuth-compatible auth models (minimal)
 //
@@ -106,6 +119,8 @@ model User {
   idempotencyRecords        IdempotencyRecord[]
   activityLogs              ActivityLog[]               @relation("ActivityLogActor")
   passwordResetTokens       PasswordResetToken[]
+  knowledgeBases            KnowledgeBase[]             @relation("UserKnowledgeBases")
+  workspaceKnowledgeBases   WorkspaceKnowledgeBase[]    @relation("UserWorkspaceKnowledgeBases")
 }
 
 /// One-time tokens for forgot-password flow (token stored as SHA-256 hash of the secret from the email link).
@@ -128,7 +143,7 @@ model Role {
 }
 
 model Notification {
-  id          String                 @id @default(cuid())
+  id          String               @id @default(cuid())
   userId      String
   actorUserId String?
   workspaceId String?
@@ -138,22 +153,22 @@ model Notification {
   content     String?
   actionUrl   String?
   metadata    Json?
-  priority    NotificationPriority   @default(NORMAL)
+  priority    NotificationPriority @default(NORMAL)
   sourceId    String?
-  read        Boolean                @default(false)
+  read        Boolean              @default(false)
   readAt      DateTime?
   deliveredAt DateTime?
-  createdAt   DateTime               @default(now())
-  updatedAt   DateTime               @updatedAt
-  user        User                   @relation(fields: [userId], references: [id], onDelete: Cascade)
-  actor       User?                  @relation("NotificationActor", fields: [actorUserId], references: [id], onDelete: SetNull)
-  workspace   Workspace?             @relation(fields: [workspaceId], references: [id], onDelete: SetNull)
+  createdAt   DateTime             @default(now())
+  updatedAt   DateTime             @updatedAt
+  user        User                 @relation(fields: [userId], references: [id], onDelete: Cascade)
+  actor       User?                @relation("NotificationActor", fields: [actorUserId], references: [id], onDelete: SetNull)
+  workspace   Workspace?           @relation(fields: [workspaceId], references: [id], onDelete: SetNull)
 
+  @@unique([userId, type, sourceId])
   @@index([userId, read, createdAt(sort: Desc)])
   @@index([userId, createdAt(sort: Desc)])
   @@index([type, createdAt(sort: Desc)])
   @@index([workspaceId])
-  @@unique([userId, type, sourceId])
 }
 
 model Session {
@@ -212,6 +227,7 @@ model Workspace {
   invitations          WorkspaceInvitation[]
   notifications        Notification[]
   activityLogs         ActivityLog[]
+  knowledgeBases       WorkspaceKnowledgeBase[]
   createdAt            DateTime              @default(now())
   updatedAt            DateTime              @updatedAt
 
@@ -243,15 +259,15 @@ model Chat {
 }
 
 model CopilotConversation {
-  id          String    @id @default(cuid())
+  id          String           @id @default(cuid())
   workspaceId String
-  workspace   Workspace @relation(fields: [workspaceId], references: [id], onDelete: Cascade)
+  workspace   Workspace        @relation(fields: [workspaceId], references: [id], onDelete: Cascade)
   userId      String
-  user        User      @relation(fields: [userId], references: [id], onDelete: Cascade)
-  title       String    @default("New Chat")
+  user        User             @relation(fields: [userId], references: [id], onDelete: Cascade)
+  title       String           @default("New Chat")
   messages    CopilotMessage[]
-  createdAt   DateTime  @default(now())
-  updatedAt   DateTime  @updatedAt
+  createdAt   DateTime         @default(now())
+  updatedAt   DateTime         @updatedAt
 
   @@index([workspaceId, userId, updatedAt])
 }
@@ -440,15 +456,15 @@ model WorksheetQuestionProgress {
 }
 
 model WorksheetPreset {
-  id          String    @id @default(cuid())
+  id          String     @id @default(cuid())
   userId      String?
   workspaceId String?
   name        String
-  isSystem    Boolean   @default(false)
+  isSystem    Boolean    @default(false)
   config      Json
-  createdAt   DateTime  @default(now())
-  updatedAt   DateTime  @updatedAt
-  user        User?     @relation(fields: [userId], references: [id], onDelete: Cascade)
+  createdAt   DateTime   @default(now())
+  updatedAt   DateTime   @updatedAt
+  user        User?      @relation(fields: [userId], references: [id], onDelete: Cascade)
   workspace   Workspace? @relation(fields: [workspaceId], references: [id], onDelete: Cascade)
 
   @@index([userId, workspaceId])
@@ -512,18 +528,18 @@ model PodcastSegment {
 }
 
 model Invoice {
-  id               String      @id @default(cuid())
+  id               String        @id @default(cuid())
   userId           String
   subscriptionId   String?
-  stripeInvoiceId  String      @unique
+  stripeInvoiceId  String        @unique
   amountPaid       Int
   status           String
   invoicePdfUrl    String?
   hostedInvoiceUrl String?
   paidAt           DateTime?
-  type             InvoiceType @default(SUBSCRIPTION)
-  createdAt        DateTime    @default(now())
-  user             User        @relation(fields: [userId], references: [id], onDelete: Cascade)
+  type             InvoiceType   @default(SUBSCRIPTION)
+  createdAt        DateTime      @default(now())
+  user             User          @relation(fields: [userId], references: [id], onDelete: Cascade)
   subscription     Subscription? @relation(fields: [subscriptionId], references: [id])
 
   @@index([userId])
@@ -596,48 +612,48 @@ model UserCredit {
 }
 
 model IdempotencyRecord {
-  id               String   @id @default(cuid())
-  userId           String
-  planId           String?               // The plan being purchased
-  resourceType     ArtifactType?         // The resource for top-ups
-  stripeSessionId  String?               // The Stripe Checkout Session ID
-  status           String   @default("pending") // pending, completed, failed, expired
-  activeLockKey    String?  @unique      // Unique lock: "user_id_plan_id" (Nullified when done)
-  
-  createdAt        DateTime @default(now())
-  updatedAt        DateTime @updatedAt
-  user             User     @relation(fields: [userId], references: [id], onDelete: Cascade)
+  id              String        @id @default(cuid())
+  userId          String
+  planId          String? // The plan being purchased
+  resourceType    ArtifactType? // The resource for top-ups
+  stripeSessionId String? // The Stripe Checkout Session ID
+  status          String        @default("pending") // pending, completed, failed, expired
+  activeLockKey   String?       @unique // Unique lock: "user_id_plan_id" (Nullified when done)
+
+  createdAt DateTime @default(now())
+  updatedAt DateTime @updatedAt
+  user      User     @relation(fields: [userId], references: [id], onDelete: Cascade)
 
   @@index([userId, status])
   @@index([activeLockKey])
 }
 
 model StripeEvent {
-  id               String   @id @default(cuid())
-  stripeEventId    String   @unique      // Stripe event ID (e.g. evt_...)
-  type             String                // Event type (e.g. checkout.session.completed)
-  status           String   @default("pending") // pending, processed, failed
-  error            String?               // Error message if processing fails
-  processedAt      DateTime?
-  createdAt        DateTime @default(now())
-  updatedAt        DateTime @updatedAt
+  id            String    @id @default(cuid())
+  stripeEventId String    @unique // Stripe event ID (e.g. evt_...)
+  type          String // Event type (e.g. checkout.session.completed)
+  status        String    @default("pending") // pending, processed, failed
+  error         String? // Error message if processing fails
+  processedAt   DateTime?
+  createdAt     DateTime  @default(now())
+  updatedAt     DateTime  @updatedAt
 
   @@index([stripeEventId])
 }
 
 /// Append-only system / user activity (tRPC and explicit events). Retention via admin or scheduled job.
 model ActivityLog {
-  id                 String               @id @default(cuid())
-  createdAt          DateTime             @default(now())
+  id                 String              @id @default(cuid())
+  createdAt          DateTime            @default(now())
   actorUserId        String?
-  actor              User?                @relation("ActivityLogActor", fields: [actorUserId], references: [id], onDelete: SetNull)
+  actor              User?               @relation("ActivityLogActor", fields: [actorUserId], references: [id], onDelete: SetNull)
   actorEmailSnapshot String?
   action             String
   category           ActivityLogCategory
   resourceType       String?
   resourceId         String?
   workspaceId        String?
-  workspace          Workspace?           @relation(fields: [workspaceId], references: [id], onDelete: SetNull)
+  workspace          Workspace?          @relation(fields: [workspaceId], references: [id], onDelete: SetNull)
   trpcPath           String?
   httpMethod         String?
   status             ActivityLogStatus
@@ -653,3 +669,89 @@ model ActivityLog {
   @@index([category, createdAt(sort: Desc)])
 }
 
+/// A retrieval-augmented knowledge base. Knowledge bases live in a global
+/// catalog managed by System Admins; workspaces opt-in by attaching one
+/// (see `WorkspaceKnowledgeBase`). When `isCurated = true` the KB shows up
+/// in the user-facing catalog search so any workspace can add it.
+///
+/// Each knowledge base owns uploaded documents (PDFs, etc.) which are split
+/// into chunks with pgvector embeddings for semantic search.
+model KnowledgeBase {
+  id             String                   @id @default(cuid())
+  createdById    String?
+  createdBy      User?                    @relation("UserKnowledgeBases", fields: [createdById], references: [id], onDelete: SetNull)
+  name           String
+  description    String?
+  isCurated      Boolean                  @default(false)
+  status         KnowledgeBaseStatus      @default(READY)
+  embeddingModel String                   @default("text-embedding-3-small")
+  embeddingDim   Int                      @default(1536)
+  meta           Json?
+  documents      KnowledgeBaseDocument[]
+  workspaces     WorkspaceKnowledgeBase[]
+  createdAt      DateTime                 @default(now())
+  updatedAt      DateTime                 @updatedAt
+
+  @@index([isCurated, updatedAt(sort: Desc)])
+}
+
+/// Many-to-many join between Workspace and KnowledgeBase. A row exists
+/// whenever a workspace has "added" a knowledge base from the catalog (or
+/// when an admin attaches one directly).
+model WorkspaceKnowledgeBase {
+  id              String        @id @default(cuid())
+  workspaceId     String
+  workspace       Workspace     @relation(fields: [workspaceId], references: [id], onDelete: Cascade)
+  knowledgeBaseId String
+  knowledgeBase   KnowledgeBase @relation(fields: [knowledgeBaseId], references: [id], onDelete: Cascade)
+  addedById       String?
+  addedBy         User?         @relation("UserWorkspaceKnowledgeBases", fields: [addedById], references: [id], onDelete: SetNull)
+  createdAt       DateTime      @default(now())
+
+  @@unique([workspaceId, knowledgeBaseId])
+  @@index([workspaceId])
+  @@index([knowledgeBaseId])
+}
+
+/// A single uploaded source document inside a knowledge base.
+model KnowledgeBaseDocument {
+  id              String                      @id @default(cuid())
+  knowledgeBaseId String
+  knowledgeBase   KnowledgeBase               @relation(fields: [knowledgeBaseId], references: [id], onDelete: Cascade)
+  name            String
+  mimeType        String
+  size            Int
+  bucket          String?
+  objectKey       String?
+  status          KnowledgeBaseDocumentStatus @default(PENDING)
+  errorMessage    String?
+  numChunks       Int                         @default(0)
+  numPages        Int?
+  meta            Json?
+  chunks          KnowledgeBaseChunk[]
+  createdAt       DateTime                    @default(now())
+  updatedAt       DateTime                    @updatedAt
+
+  @@index([knowledgeBaseId])
+  @@index([knowledgeBaseId, createdAt(sort: Desc)])
+}
+
+/// A single text chunk inside a document. The `embedding` column is a
+/// pgvector value; Prisma cannot generate types for it, so similarity
+/// search must be performed via raw SQL (`$queryRaw`). The dimension is
+/// fixed at 1536 to match `text-embedding-3-small`.
+model KnowledgeBaseChunk {
+  id              String                      @id @default(cuid())
+  documentId      String
+  document        KnowledgeBaseDocument       @relation(fields: [documentId], references: [id], onDelete: Cascade)
+  knowledgeBaseId String
+  chunkIndex      Int
+  pageNumber      Int?
+  content         String
+  tokenCount      Int?
+  embedding       Unsupported("vector(1536)")?
+  createdAt       DateTime                    @default(now())
+
+  @@index([knowledgeBaseId])
+  @@index([documentId, chunkIndex])
+}

package/prisma/seed.mjs

@@ -0,0 +1,67 @@
+import { PrismaClient } from '@prisma/client';
+
+/** Keep in sync with server/src/lib/curated-kb-seed.ts */
+const CURATED_KB_SEEDS = [
+  { slug: 'ap-biology', name: 'AP Biology', description: 'Cell biology, genetics, evolution, and ecology aligned with the AP Biology curriculum.', category: 'AP', subject: 'biology' },
+  { slug: 'ap-chemistry', name: 'AP Chemistry', description: 'Atomic structure, bonding, thermodynamics, and kinetics for AP Chemistry.', category: 'AP', subject: 'chemistry' },
+  { slug: 'ap-physics-1', name: 'AP Physics 1', description: 'Algebra-based mechanics, waves, and electricity for AP Physics 1.', category: 'AP', subject: 'physics' },
+  { slug: 'ap-calculus-ab', name: 'AP Calculus AB', description: 'Limits, derivatives, integrals, and the Fundamental Theorem of Calculus.', category: 'AP', subject: 'calculus' },
+  { slug: 'ap-calculus-bc', name: 'AP Calculus BC', description: 'Series, parametric equations, polar coordinates, and advanced integration.', category: 'AP', subject: 'calculus' },
+  { slug: 'ap-us-history', name: 'AP US History', description: 'Colonial America through the modern era with primary source analysis.', category: 'AP', subject: 'history' },
+  { slug: 'ap-world-history', name: 'AP World History', description: 'Global civilizations, trade networks, and historical thinking skills.', category: 'AP', subject: 'history' },
+  { slug: 'ap-english-lang', name: 'AP English Language', description: 'Rhetorical analysis, argumentation, and synthesis writing strategies.', category: 'AP', subject: 'english' },
+  { slug: 'ap-english-lit', name: 'AP English Literature', description: 'Literary analysis, poetry, prose fiction, and critical essay writing.', category: 'AP', subject: 'literature' },
+  { slug: 'ap-psychology', name: 'AP Psychology', description: 'Biological bases of behavior, cognition, development, and social psychology.', category: 'AP', subject: 'psychology' },
+  { slug: 'ap-environmental-science', name: 'AP Environmental Science', description: 'Ecosystems, biodiversity, pollution, and sustainability concepts.', category: 'AP', subject: 'environmental' },
+  { slug: 'ap-macroeconomics', name: 'AP Macroeconomics', description: 'GDP, inflation, fiscal and monetary policy, and international trade.', category: 'AP', subject: 'economics' },
+  { slug: 'ib-biology-hl', name: 'IB Biology HL', description: 'Higher Level biology: molecular biology, genetics, ecology, and human physiology.', category: 'IB', subject: 'biology' },
+  { slug: 'ib-chemistry-hl', name: 'IB Chemistry HL', description: 'Higher Level chemistry: organic, inorganic, physical, and analytical topics.', category: 'IB', subject: 'chemistry' },
+  { slug: 'ib-physics-hl', name: 'IB Physics HL', description: 'Higher Level physics: mechanics, fields, waves, and nuclear physics.', category: 'IB', subject: 'physics' },
+  { slug: 'ib-math-aa-hl', name: 'IB Math AA HL', description: 'Analysis & Approaches HL: calculus, proof, vectors, and complex numbers.', category: 'IB', subject: 'math' },
+  { slug: 'ib-math-ai-hl', name: 'IB Math AI HL', description: 'Applications & Interpretation HL: statistics, modelling, and technology use.', category: 'IB', subject: 'math' },
+  { slug: 'ib-history-hl', name: 'IB History HL', description: 'Higher Level history: 20th century world history and historical investigation.', category: 'IB', subject: 'history' },
+  { slug: 'ib-english-a-hl', name: 'IB English A HL', description: 'Language & Literature HL: textual analysis, comparative study, and IO prep.', category: 'IB', subject: 'literature' },
+  { slug: 'ib-economics-hl', name: 'IB Economics HL', description: 'Microeconomics, macroeconomics, international economics, and development.', category: 'IB', subject: 'economics' },
+  { slug: 'ib-psychology-hl', name: 'IB Psychology HL', description: 'Biological, cognitive, sociocultural, and abnormal psychology at HL depth.', category: 'IB', subject: 'psychology' },
+];
+
+const prisma = new PrismaClient();
+
+async function main() {
+  for (const seed of CURATED_KB_SEEDS) {
+    const existing = await prisma.knowledgeBase.findFirst({
+      where: { meta: { path: ['slug'], equals: seed.slug } },
+    });
+
+    if (existing) {
+      await prisma.knowledgeBase.update({
+        where: { id: existing.id },
+        data: {
+          name: seed.name,
+          description: seed.description,
+          isCurated: true,
+          meta: { slug: seed.slug, category: seed.category, subject: seed.subject },
+        },
+      });
+      continue;
+    }
+
+    await prisma.knowledgeBase.create({
+      data: {
+        name: seed.name,
+        description: seed.description,
+        isCurated: true,
+        meta: { slug: seed.slug, category: seed.category, subject: seed.subject },
+      },
+    });
+  }
+
+  console.log(`Seeded ${CURATED_KB_SEEDS.length} curated knowledge bases.`);
+}
+
+main()
+  .catch((e) => {
+    console.error(e);
+    process.exit(1);
+  })
+  .finally(() => prisma.$disconnect());

package/scripts/debug/README.md

@@ -0,0 +1,4 @@
+# Debug scripts
+
+Ad-hoc scripts used once during development / debugging. Kept out of `src/`
+so the build surface doesn't pick them up. Safe to delete if not needed.

package/src/README.md

@@ -0,0 +1,63 @@
+# Server source layout
+
+This server uses a layered architecture. Please keep new code in the right
+layer so routers don't grow back to 1000+ lines again.
+
+## Layers
+
+```
+ tRPC Client ->  routers/*.ts        Zod input + middleware + glue
+              -> services/*.ts       business logic and orchestration
+              -> repositories/*.ts   Prisma queries
+              -> lib/*               infra: email, pusher, stripe, storage,
+                                     inference, logger, errors, env, etc.
+```
+
+- `routers/` — **thin**. Each procedure:
+  1. Declares its Zod input schema.
+  2. Selects a procedure type (`authedProcedure`, `verifiedProcedure`, etc.).
+  3. Instantiates a service (`new FooService(ctx.db)`) and returns its result.
+  4. Optionally handles transport-only concerns (e.g. cookies in
+     `auth.ts`).
+  5. Does **not** call Prisma directly. Does **not** contain business
+     logic beyond trivial response shaping.
+- `services/` — **business logic**. One service per domain aggregate.
+  Constructor signature is `(db: PrismaClient)` via `BaseService`. Services
+  can depend on other services and on repositories, but must **not**
+  depend on `Context`, `Request`, or other HTTP types.
+- `repositories/` — **data access**. Thin Prisma wrappers per aggregate.
+  Expose intent-revealing methods (`findAccessibleById`, `listOwned`) so
+  the shape of `where` clauses doesn't leak into services.
+- `lib/` — infrastructure and pure utilities: Prisma client singleton,
+  Stripe SDK, email transport, Pusher client, Supabase client, logger,
+  shared error classes, env validation, retry helper, validation helpers.
+  **Not** domain services.
+
+## Conventions
+
+- **File naming**: `{kebab-case-domain}.service.ts`,
+  `{kebab-case-domain}.repository.ts` (singular).
+- **Class naming**: `FooService`, `FooRepository`.
+- **Constructor injection**: every service/repository takes a
+  `PrismaClient` in its constructor. Instantiate per-request from the
+  router (`new FooService(ctx.db)`).
+- **Workspace access**: use `workspaceAccessWhere(userId)` from
+  `repositories/workspace.repository.ts` for member-inclusive access, or
+  `workspaceOwnerWhere(userId)` for owner-only. Don't hand-roll
+  `OR: [{ ownerId }, …]` again.
+- **Errors**: throw `TRPCError` for HTTP-shaped errors in services. They
+  propagate cleanly back through the router.
+
+## Folders
+
+- `context.ts` — tRPC context builder (session + db + req/res).
+- `trpc.ts` — procedure types and middleware (auth, verification, plan
+  limits, activity logging).
+- `routers/` — one file per top-level router, plus `_app.ts` composition.
+- `services/` — domain services (router → service → repository).
+- `repositories/` — Prisma access per aggregate.
+- `lib/` — infrastructure and shared utilities.
+- `types/` — shared TS types.
+- `scripts/` — operational tooling (cron-style tasks).
+
+See `REFACTOR_NOTES.md` at the repo root for known follow-up work.

package/src/lib/ai/config.ts

@@ -0,0 +1,34 @@
+export const aiConfig = {
+  llm: {
+    apiKey: process.env.INFERENCE_API_KEY,
+    baseUrl: process.env.INFERENCE_BASE_URL,
+    model: process.env.INFERENCE_MODEL ?? 'command-a-03-2025',
+  },
+  embeddings: {
+    apiKey:
+      process.env.OPENAI_API_KEY,
+    baseUrl:
+      undefined,
+    model: process.env.EMBEDDING_MODEL ?? 'text-embedding-3-small',
+    dim: Number(process.env.EMBEDDING_DIM ?? 1536),
+  },
+  inferenceBackend: {
+    url: process.env.INFERENCE_API_URL,
+    mockEnabled: process.env.DONT_TEST_INFERENCE === 'true',
+    mockDelayMs: process.env.DONT_TEST_INFERENCE === 'true' ? 10_000 : 0,
+  },
+} as const;
+
+export function getInferenceBackendUploadUrl(): string {
+  const base = aiConfig.inferenceBackend.url;
+  if (!base) {
+    throw new Error('INFERENCE_API_URL is not configured');
+  }
+  return `${base.replace(/\/$/, '')}/upload`;
+}
+
+export function tryGetInferenceBackendUploadUrl(): string | null {
+  const base = aiConfig.inferenceBackend.url;
+  if (!base) return null;
+  return `${base.replace(/\/$/, '')}/upload`;
+}

package/src/lib/ai/embedding-client.ts

@@ -0,0 +1,47 @@
+import OpenAI from 'openai';
+import { aiConfig } from './config.js';
+
+const client = new OpenAI({
+  apiKey: aiConfig.embeddings.apiKey,
+  baseURL: aiConfig.embeddings.baseUrl,
+});
+
+export const DEFAULT_EMBEDDING_MODEL = aiConfig.embeddings.model;
+export const DEFAULT_EMBEDDING_DIM = aiConfig.embeddings.dim;
+
+export interface EmbedOptions {
+  model?: string;
+}
+
+export async function embedTexts(
+  texts: string[],
+  opts: EmbedOptions = {},
+): Promise<number[][]> {
+  const model = opts.model ?? DEFAULT_EMBEDDING_MODEL;
+
+  if (texts.length === 0) return [];
+
+  const BATCH = 96;
+  const out: number[][] = new Array(texts.length);
+  for (let i = 0; i < texts.length; i += BATCH) {
+    const batch = texts.slice(i, i + BATCH);
+
+    const res = await client.embeddings.create({ model, input: batch });
+    res.data.forEach((row, j) => {
+      out[i + j] = row.embedding as number[];
+    });
+  }
+  return out;
+}
+
+export async function embedQuery(
+  text: string,
+  opts: EmbedOptions = {},
+): Promise<number[]> {
+  const [vec] = await embedTexts([text], opts);
+  return vec;
+}
+
+export function toVectorLiteral(vec: number[]): string {
+  return `[${vec.map((n) => (Number.isFinite(n) ? n : 0)).join(',')}]`;
+}

package/src/lib/ai/index.ts

@@ -0,0 +1,62 @@
+import { aiConfig } from './config.js';
+import {
+  complete,
+  completeText,
+} from './llm-client.js';
+import {
+  DEFAULT_EMBEDDING_DIM,
+  DEFAULT_EMBEDDING_MODEL,
+  embedQuery,
+  embedTexts,
+  toVectorLiteral,
+} from './embedding-client.js';
+import { extractJson, parseJsonField } from './json-parse.js';
+import { isAiMockMode, mockDelay } from './mock.js';
+import { inferenceBackend } from './inference-backend/client.js';
+
+export const ai = {
+  config: aiConfig,
+  isMockMode: isAiMockMode,
+  mockDelay,
+  llm: {
+    complete,
+    completeText,
+  },
+  embed: {
+    texts: embedTexts,
+    query: embedQuery,
+    toVectorLiteral,
+    DEFAULT_MODEL: DEFAULT_EMBEDDING_MODEL,
+    DEFAULT_DIM: DEFAULT_EMBEDDING_DIM,
+  },
+  backend: inferenceBackend,
+  parse: {
+    extractJson,
+    parseJsonField,
+  },
+};
+
+export type { ChatMessage, ChatRole, CompleteOptions } from './types.js';
+export type {
+  AISession,
+  ProcessFileResult,
+  PodcastSpeaker,
+  StudyGuideSegment,
+  WorksheetGenerationOptions,
+} from './inference-backend/types.js';
+
+export {
+  aiConfig,
+  complete,
+  completeText,
+  DEFAULT_EMBEDDING_DIM,
+  DEFAULT_EMBEDDING_MODEL,
+  embedQuery,
+  embedTexts,
+  extractJson,
+  inferenceBackend,
+  isAiMockMode,
+  mockDelay,
+  parseJsonField,
+  toVectorLiteral,
+};