@lobehub/lobehub 2.0.0-next.374 → 2.0.0-next.376

package/packages/database/src/schemas/agent.ts

@@ -105,6 +105,8 @@ export const agentsKnowledgeBases = pgTable(
   (t) => [
     primaryKey({ columns: [t.agentId, t.knowledgeBaseId] }),
     index('agents_knowledge_bases_agent_id_idx').on(t.agentId),
+    index('agents_knowledge_bases_knowledge_base_id_idx').on(t.knowledgeBaseId),
+    index('agents_knowledge_bases_user_id_idx').on(t.userId),
   ],
 );
 
@@ -127,5 +129,7 @@ export const agentsFiles = pgTable(
   (t) => [
     primaryKey({ columns: [t.fileId, t.agentId, t.userId] }),
     index('agents_files_agent_id_idx').on(t.agentId),
+    index('agents_files_file_id_idx').on(t.fileId),
+    index('agents_files_user_id_idx').on(t.userId),
   ],
 );

package/packages/database/src/schemas/chatGroup.ts

@@ -52,6 +52,7 @@ export const chatGroups = pgTable(
   },
   (t) => [
     uniqueIndex('chat_groups_client_id_user_id_unique').on(t.clientId, t.userId),
+    index('chat_groups_user_id_idx').on(t.userId),
     index('chat_groups_group_id_idx').on(t.groupId),
   ],
 );
@@ -95,6 +96,7 @@ export const chatGroupsAgents = pgTable(
   },
   (t) => ({
     pk: primaryKey({ columns: [t.chatGroupId, t.agentId] }),
+    userIdIdx: index('chat_groups_agents_user_id_idx').on(t.userId),
   }),
 );
 

package/packages/database/src/schemas/file.ts

@@ -166,6 +166,8 @@ export const files = pgTable(
       fileHashIdx: index('file_hash_idx').on(table.fileHash),
       userIdIdx: index('files_user_id_idx').on(table.userId),
       parentIdIdx: index('files_parent_id_idx').on(table.parentId),
+      chunkTaskIdIdx: index('files_chunk_task_id_idx').on(table.chunkTaskId),
+      embeddingTaskIdIdx: index('files_embedding_task_id_idx').on(table.embeddingTaskId),
       clientIdUnique: uniqueIndex('files_client_id_user_id_unique').on(
         table.clientId,
         table.userId,
@@ -231,5 +233,7 @@ export const knowledgeBaseFiles = pgTable(
   (t) => [
     primaryKey({ columns: [t.knowledgeBaseId, t.fileId] }),
     index('knowledge_base_files_kb_id_idx').on(t.knowledgeBaseId),
+    index('knowledge_base_files_user_id_idx').on(t.userId),
+    index('knowledge_base_files_file_id_idx').on(t.fileId),
   ],
 );

package/packages/database/src/schemas/generation.ts

@@ -134,6 +134,7 @@ export const generations = pgTable(
   (t) => [
     index('generations_user_id_idx').on(t.userId),
     index('generations_batch_id_idx').on(t.generationBatchId),
+    index('generations_file_id_idx').on(t.fileId),
   ],
 );
 

package/packages/database/src/schemas/message.ts

@@ -70,8 +70,11 @@ export const messageGroups = pgTable(
   },
   (t) => [
     uniqueIndex('message_groups_client_id_user_id_unique').on(t.clientId, t.userId),
+    index('message_groups_user_id_idx').on(t.userId),
     index('message_groups_topic_id_idx').on(t.topicId),
     index('message_groups_type_idx').on(t.type),
+    index('message_groups_parent_group_id_idx').on(t.parentGroupId),
+    index('message_groups_parent_message_id_idx').on(t.parentMessageId),
   ],
 );
 
@@ -174,6 +177,7 @@ export const messagePlugins = pgTable(
   },
   (t) => [
     uniqueIndex('message_plugins_client_id_user_id_unique').on(t.clientId, t.userId),
+    index('message_plugins_user_id_idx').on(t.userId),
     index('message_plugins_tool_call_id_idx').on(t.toolCallId),
   ],
 );
@@ -194,6 +198,7 @@ export const messageTTS = pgTable(
   },
   (t) => ({
     clientIdUnique: uniqueIndex('message_tts_client_id_user_id_unique').on(t.clientId, t.userId),
+    userIdIdx: index('message_tts_user_id_idx').on(t.userId),
   }),
 );
 
@@ -216,6 +221,7 @@ export const messageTranslates = pgTable(
       t.clientId,
       t.userId,
     ),
+    userIdIdx: index('message_translates_user_id_idx').on(t.userId),
   }),
 );
 
@@ -236,6 +242,8 @@ export const messagesFiles = pgTable(
   },
   (t) => ({
     pk: primaryKey({ columns: [t.fileId, t.messageId] }),
+    userIdIdx: index('messages_files_user_id_idx').on(t.userId),
+    messageIdIdx: index('messages_files_message_id_idx').on(t.messageId),
   }),
 );
 
@@ -259,6 +267,9 @@ export const messageQueries = pgTable(
       t.clientId,
       t.userId,
     ),
+    userIdIdx: index('message_queries_user_id_idx').on(t.userId),
+    messageIdIdx: index('message_queries_message_id_idx').on(t.messageId),
+    embeddingsIdIdx: index('message_queries_embeddings_id_idx').on(t.embeddingsId),
   }),
 );
 
@@ -277,6 +288,9 @@ export const messageQueryChunks = pgTable(
   },
   (t) => ({
     pk: primaryKey({ columns: [t.chunkId, t.messageId, t.queryId] }),
+    userIdIdx: index('message_query_chunks_user_id_idx').on(t.userId),
+    messageIdIdx: index('message_query_chunks_message_id_idx').on(t.messageId),
+    queryIdIdx: index('message_query_chunks_query_id_idx').on(t.queryId),
   }),
 );
 export type NewMessageFileChunk = typeof messageQueryChunks.$inferInsert;
@@ -294,5 +308,7 @@ export const messageChunks = pgTable(
   },
   (t) => ({
     pk: primaryKey({ columns: [t.chunkId, t.messageId] }),
+    userIdIdx: index('message_chunks_user_id_idx').on(t.userId),
+    messageIdIdx: index('message_chunks_message_id_idx').on(t.messageId),
   }),
 );

package/packages/database/src/schemas/nextauth.ts

@@ -1,4 +1,4 @@
-import { boolean, integer, pgTable, primaryKey, text, timestamp } from 'drizzle-orm/pg-core';
+import { boolean, index, integer, pgTable, primaryKey, text, timestamp } from 'drizzle-orm/pg-core';
 
 import { users } from './user';
 
@@ -33,6 +33,7 @@ export const nextauthAccounts = pgTable(
     compositePk: primaryKey({
       columns: [account.provider, account.providerAccountId],
     }),
+    userIdIdx: index('nextauth_accounts_user_id_idx').on(account.userId),
   }),
 );
 
@@ -42,13 +43,19 @@ export const nextauthAccounts = pgTable(
  * which will enable remote logout and other features.
  * @see {@link https://authjs.dev/guides/creating-a-database-adapter#database-session-management | NextAuth Doc}
  */
-export const nextauthSessions = pgTable(`nextauth_sessions`, {
-  expires: timestamp('expires', { mode: 'date' }).notNull(),
-  sessionToken: text('sessionToken').primaryKey(),
-  userId: text('user_id')
-    .notNull()
-    .references(() => users.id, { onDelete: 'cascade' }),
-});
+export const nextauthSessions = pgTable(
+  `nextauth_sessions`,
+  {
+    expires: timestamp('expires', { mode: 'date' }).notNull(),
+    sessionToken: text('sessionToken').primaryKey(),
+    userId: text('user_id')
+      .notNull()
+      .references(() => users.id, { onDelete: 'cascade' }),
+  },
+  (t) => ({
+    userIdIdx: index('nextauth_sessions_user_id_idx').on(t.userId),
+  }),
+);
 
 /**
  * @description This table stores nextauth verification tokens.

package/packages/database/src/schemas/oidc.ts

@@ -1,5 +1,5 @@
 /* eslint-disable sort-keys-fix/sort-keys-fix  */
-import { boolean, jsonb, pgTable, primaryKey, text, varchar } from 'drizzle-orm/pg-core';
+import { boolean, index, jsonb, pgTable, primaryKey, text, varchar } from 'drizzle-orm/pg-core';
 import { createInsertSchema, createSelectSchema } from 'drizzle-zod';
 
 import { timestamps, timestamptz } from './_helpers';
@@ -9,68 +9,92 @@ import { users } from './user';
  * OIDC authorization code
  * One of the models that oidc-provider needs to persist
  */
-export const oidcAuthorizationCodes = pgTable('oidc_authorization_codes', {
-  id: varchar('id', { length: 255 }).primaryKey(),
-  data: jsonb('data').notNull(),
-  expiresAt: timestamptz('expires_at').notNull(),
-  consumedAt: timestamptz('consumed_at'),
-  userId: text('user_id')
-    .references(() => users.id, { onDelete: 'cascade' })
-    .notNull(),
-  clientId: varchar('client_id', { length: 255 }).notNull(),
-  grantId: varchar('grant_id', { length: 255 }),
-  ...timestamps,
-});
+export const oidcAuthorizationCodes = pgTable(
+  'oidc_authorization_codes',
+  {
+    id: varchar('id', { length: 255 }).primaryKey(),
+    data: jsonb('data').notNull(),
+    expiresAt: timestamptz('expires_at').notNull(),
+    consumedAt: timestamptz('consumed_at'),
+    userId: text('user_id')
+      .references(() => users.id, { onDelete: 'cascade' })
+      .notNull(),
+    clientId: varchar('client_id', { length: 255 }).notNull(),
+    grantId: varchar('grant_id', { length: 255 }),
+    ...timestamps,
+  },
+  (t) => ({
+    userIdIdx: index('oidc_authorization_codes_user_id_idx').on(t.userId),
+  }),
+);
 
 /**
  * OIDC access token
  * One of the models that oidc-provider needs to persist
  */
-export const oidcAccessTokens = pgTable('oidc_access_tokens', {
-  id: varchar('id', { length: 255 }).primaryKey(),
-  data: jsonb('data').notNull(),
-  expiresAt: timestamptz('expires_at').notNull(),
-  consumedAt: timestamptz('consumed_at'),
-  userId: text('user_id')
-    .references(() => users.id, { onDelete: 'cascade' })
-    .notNull(),
-  clientId: varchar('client_id', { length: 255 }).notNull(),
-  grantId: varchar('grant_id', { length: 255 }),
-  ...timestamps,
-});
+export const oidcAccessTokens = pgTable(
+  'oidc_access_tokens',
+  {
+    id: varchar('id', { length: 255 }).primaryKey(),
+    data: jsonb('data').notNull(),
+    expiresAt: timestamptz('expires_at').notNull(),
+    consumedAt: timestamptz('consumed_at'),
+    userId: text('user_id')
+      .references(() => users.id, { onDelete: 'cascade' })
+      .notNull(),
+    clientId: varchar('client_id', { length: 255 }).notNull(),
+    grantId: varchar('grant_id', { length: 255 }),
+    ...timestamps,
+  },
+  (t) => ({
+    userIdIdx: index('oidc_access_tokens_user_id_idx').on(t.userId),
+  }),
+);
 
 /**
  * OIDC refresh token
  * One of the models that oidc-provider needs to persist
  */
-export const oidcRefreshTokens = pgTable('oidc_refresh_tokens', {
-  id: varchar('id', { length: 255 }).primaryKey(),
-  data: jsonb('data').notNull(),
-  expiresAt: timestamptz('expires_at').notNull(),
-  consumedAt: timestamptz('consumed_at'),
-  userId: text('user_id')
-    .references(() => users.id, { onDelete: 'cascade' })
-    .notNull(),
-  clientId: varchar('client_id', { length: 255 }).notNull(),
-  grantId: varchar('grant_id', { length: 255 }),
-  ...timestamps,
-});
+export const oidcRefreshTokens = pgTable(
+  'oidc_refresh_tokens',
+  {
+    id: varchar('id', { length: 255 }).primaryKey(),
+    data: jsonb('data').notNull(),
+    expiresAt: timestamptz('expires_at').notNull(),
+    consumedAt: timestamptz('consumed_at'),
+    userId: text('user_id')
+      .references(() => users.id, { onDelete: 'cascade' })
+      .notNull(),
+    clientId: varchar('client_id', { length: 255 }).notNull(),
+    grantId: varchar('grant_id', { length: 255 }),
+    ...timestamps,
+  },
+  (t) => ({
+    userIdIdx: index('oidc_refresh_tokens_user_id_idx').on(t.userId),
+  }),
+);
 
 /**
  * OIDC device code
  * One of the models that oidc-provider needs to persist
  */
-export const oidcDeviceCodes = pgTable('oidc_device_codes', {
-  id: varchar('id', { length: 255 }).primaryKey(),
-  data: jsonb('data').notNull(),
-  expiresAt: timestamptz('expires_at').notNull(),
-  consumedAt: timestamptz('consumed_at'),
-  userId: text('user_id').references(() => users.id, { onDelete: 'cascade' }),
-  clientId: varchar('client_id', { length: 255 }).notNull(),
-  grantId: varchar('grant_id', { length: 255 }),
-  userCode: varchar('user_code', { length: 255 }),
-  ...timestamps,
-});
+export const oidcDeviceCodes = pgTable(
+  'oidc_device_codes',
+  {
+    id: varchar('id', { length: 255 }).primaryKey(),
+    data: jsonb('data').notNull(),
+    expiresAt: timestamptz('expires_at').notNull(),
+    consumedAt: timestamptz('consumed_at'),
+    userId: text('user_id').references(() => users.id, { onDelete: 'cascade' }),
+    clientId: varchar('client_id', { length: 255 }).notNull(),
+    grantId: varchar('grant_id', { length: 255 }),
+    userCode: varchar('user_code', { length: 255 }),
+    ...timestamps,
+  },
+  (t) => ({
+    userIdIdx: index('oidc_device_codes_user_id_idx').on(t.userId),
+  }),
+);
 
 /**
  * OIDC interaction session
@@ -87,17 +111,23 @@ export const oidcInteractions = pgTable('oidc_interactions', {
  * OIDC grant record
  * One of the models that oidc-provider needs to persist
  */
-export const oidcGrants = pgTable('oidc_grants', {
-  id: varchar('id', { length: 255 }).primaryKey(),
-  data: jsonb('data').notNull(),
-  expiresAt: timestamptz('expires_at').notNull(),
-  consumedAt: timestamptz('consumed_at'),
-  userId: text('user_id')
-    .references(() => users.id, { onDelete: 'cascade' })
-    .notNull(),
-  clientId: varchar('client_id', { length: 255 }).notNull(),
-  ...timestamps,
-});
+export const oidcGrants = pgTable(
+  'oidc_grants',
+  {
+    id: varchar('id', { length: 255 }).primaryKey(),
+    data: jsonb('data').notNull(),
+    expiresAt: timestamptz('expires_at').notNull(),
+    consumedAt: timestamptz('consumed_at'),
+    userId: text('user_id')
+      .references(() => users.id, { onDelete: 'cascade' })
+      .notNull(),
+    clientId: varchar('client_id', { length: 255 }).notNull(),
+    ...timestamps,
+  },
+  (t) => ({
+    userIdIdx: index('oidc_grants_user_id_idx').on(t.userId),
+  }),
+);
 
 /**
  * OIDC client configuration
@@ -126,15 +156,21 @@ export const oidcClients = pgTable('oidc_clients', {
  * OIDC session
  * One of the models that oidc-provider needs to persist
  */
-export const oidcSessions = pgTable('oidc_sessions', {
-  id: varchar('id', { length: 255 }).primaryKey(),
-  data: jsonb('data').notNull(),
-  expiresAt: timestamptz('expires_at').notNull(),
-  userId: text('user_id')
-    .references(() => users.id, { onDelete: 'cascade' })
-    .notNull(),
-  ...timestamps,
-});
+export const oidcSessions = pgTable(
+  'oidc_sessions',
+  {
+    id: varchar('id', { length: 255 }).primaryKey(),
+    data: jsonb('data').notNull(),
+    expiresAt: timestamptz('expires_at').notNull(),
+    userId: text('user_id')
+      .references(() => users.id, { onDelete: 'cascade' })
+      .notNull(),
+    ...timestamps,
+  },
+  (t) => ({
+    userIdIdx: index('oidc_sessions_user_id_idx').on(t.userId),
+  }),
+);
 
 /**
  * OIDC authorization consent record

package/packages/database/src/schemas/rag.ts

@@ -61,6 +61,9 @@ export const unstructuredChunks = pgTable(
       t.clientId,
       t.userId,
     ),
+    userIdIdx: index('unstructured_chunks_user_id_idx').on(t.userId),
+    compositeIdIdx: index('unstructured_chunks_composite_id_idx').on(t.compositeId),
+    fileIdIdx: index('unstructured_chunks_file_id_idx').on(t.fileId),
   }),
 );
 
@@ -116,6 +119,7 @@ export const documentChunks = pgTable(
     primaryKey({ columns: [t.documentId, t.chunkId] }),
     index('document_chunks_document_id_idx').on(t.documentId),
     index('document_chunks_chunk_id_idx').on(t.chunkId),
+    index('document_chunks_user_id_idx').on(t.userId),
   ],
 );
 

package/packages/database/src/schemas/ragEvals.ts

@@ -1,102 +1,126 @@
 /* eslint-disable sort-keys-fix/sort-keys-fix  */
 import { DEFAULT_MODEL } from '@lobechat/const';
 import { EvalEvaluationStatus } from '@lobechat/types';
-import { integer, jsonb, pgTable, text, uuid } from 'drizzle-orm/pg-core';
+import { index, integer, jsonb, pgTable, text, uuid } from 'drizzle-orm/pg-core';
 
 import { timestamps } from './_helpers';
 import { knowledgeBases } from './file';
 import { embeddings } from './rag';
 import { users } from './user';
 
-export const evalDatasets = pgTable('rag_eval_datasets', {
-  id: integer('id').generatedAlwaysAsIdentity({ startWith: 30_000 }).primaryKey(),
+export const evalDatasets = pgTable(
+  'rag_eval_datasets',
+  {
+    id: integer('id').generatedAlwaysAsIdentity({ startWith: 30_000 }).primaryKey(),
 
-  description: text('description'),
-  name: text('name').notNull(),
+    description: text('description'),
+    name: text('name').notNull(),
 
-  knowledgeBaseId: text('knowledge_base_id').references(() => knowledgeBases.id, {
-    onDelete: 'cascade',
-  }),
-  userId: text('user_id').references(() => users.id, { onDelete: 'cascade' }),
+    knowledgeBaseId: text('knowledge_base_id').references(() => knowledgeBases.id, {
+      onDelete: 'cascade',
+    }),
+    userId: text('user_id').references(() => users.id, { onDelete: 'cascade' }),
 
-  ...timestamps,
-});
+    ...timestamps,
+  },
+  (t) => ({
+    userIdIdx: index('rag_eval_datasets_user_id_idx').on(t.userId),
+  }),
+);
 
 export type NewEvalDatasetsItem = typeof evalDatasets.$inferInsert;
 export type EvalDatasetsSelectItem = typeof evalDatasets.$inferSelect;
 
-export const evalDatasetRecords = pgTable('rag_eval_dataset_records', {
-  id: integer('id').generatedAlwaysAsIdentity().primaryKey(),
-  datasetId: integer('dataset_id')
-    .references(() => evalDatasets.id, { onDelete: 'cascade' })
-    .notNull(),
-
-  ideal: text('ideal'),
-  question: text('question'),
-  referenceFiles: text('reference_files').array(),
-  metadata: jsonb('metadata'),
-
-  userId: text('user_id').references(() => users.id, { onDelete: 'cascade' }),
-  ...timestamps,
-});
+export const evalDatasetRecords = pgTable(
+  'rag_eval_dataset_records',
+  {
+    id: integer('id').generatedAlwaysAsIdentity().primaryKey(),
+    datasetId: integer('dataset_id')
+      .references(() => evalDatasets.id, { onDelete: 'cascade' })
+      .notNull(),
+
+    ideal: text('ideal'),
+    question: text('question'),
+    referenceFiles: text('reference_files').array(),
+    metadata: jsonb('metadata'),
+
+    userId: text('user_id').references(() => users.id, { onDelete: 'cascade' }),
+    ...timestamps,
+  },
+  (t) => ({
+    userIdIdx: index('rag_eval_dataset_records_user_id_idx').on(t.userId),
+  }),
+);
 
 export type NewEvalDatasetRecordsItem = typeof evalDatasetRecords.$inferInsert;
 export type EvalDatasetRecordsSelectItem = typeof evalDatasetRecords.$inferSelect;
 
-export const evalEvaluation = pgTable('rag_eval_evaluations', {
-  id: integer('id').generatedAlwaysAsIdentity().primaryKey(),
-  name: text('name').notNull(),
-  description: text('description'),
-
-  evalRecordsUrl: text('eval_records_url'),
-  status: text('status').$defaultFn(() => EvalEvaluationStatus.Pending),
-  error: jsonb('error'),
-
-  datasetId: integer('dataset_id')
-    .references(() => evalDatasets.id, { onDelete: 'cascade' })
-    .notNull(),
-  knowledgeBaseId: text('knowledge_base_id').references(() => knowledgeBases.id, {
-    onDelete: 'cascade',
+export const evalEvaluation = pgTable(
+  'rag_eval_evaluations',
+  {
+    id: integer('id').generatedAlwaysAsIdentity().primaryKey(),
+    name: text('name').notNull(),
+    description: text('description'),
+
+    evalRecordsUrl: text('eval_records_url'),
+    status: text('status').$defaultFn(() => EvalEvaluationStatus.Pending),
+    error: jsonb('error'),
+
+    datasetId: integer('dataset_id')
+      .references(() => evalDatasets.id, { onDelete: 'cascade' })
+      .notNull(),
+    knowledgeBaseId: text('knowledge_base_id').references(() => knowledgeBases.id, {
+      onDelete: 'cascade',
+    }),
+    languageModel: text('language_model').$defaultFn(() => DEFAULT_MODEL),
+    embeddingModel: text('embedding_model'),
+
+    userId: text('user_id').references(() => users.id, { onDelete: 'cascade' }),
+    ...timestamps,
+  },
+  (t) => ({
+    userIdIdx: index('rag_eval_evaluations_user_id_idx').on(t.userId),
   }),
-  languageModel: text('language_model').$defaultFn(() => DEFAULT_MODEL),
-  embeddingModel: text('embedding_model'),
-
-  userId: text('user_id').references(() => users.id, { onDelete: 'cascade' }),
-  ...timestamps,
-});
+);
 
 export type NewEvalEvaluationItem = typeof evalEvaluation.$inferInsert;
 export type EvalEvaluationSelectItem = typeof evalEvaluation.$inferSelect;
 
-export const evaluationRecords = pgTable('rag_eval_evaluation_records', {
-  id: integer('id').generatedAlwaysAsIdentity().primaryKey(),
-
-  question: text('question').notNull(),
-  answer: text('answer'),
-  context: text('context').array(),
-  ideal: text('ideal'),
-
-  status: text('status').$defaultFn(() => EvalEvaluationStatus.Pending),
-  error: jsonb('error'),
-
-  languageModel: text('language_model'),
-  embeddingModel: text('embedding_model'),
-
-  questionEmbeddingId: uuid('question_embedding_id').references(() => embeddings.id, {
-    onDelete: 'set null',
+export const evaluationRecords = pgTable(
+  'rag_eval_evaluation_records',
+  {
+    id: integer('id').generatedAlwaysAsIdentity().primaryKey(),
+
+    question: text('question').notNull(),
+    answer: text('answer'),
+    context: text('context').array(),
+    ideal: text('ideal'),
+
+    status: text('status').$defaultFn(() => EvalEvaluationStatus.Pending),
+    error: jsonb('error'),
+
+    languageModel: text('language_model'),
+    embeddingModel: text('embedding_model'),
+
+    questionEmbeddingId: uuid('question_embedding_id').references(() => embeddings.id, {
+      onDelete: 'set null',
+    }),
+
+    duration: integer('duration'),
+    datasetRecordId: integer('dataset_record_id')
+      .references(() => evalDatasetRecords.id, { onDelete: 'cascade' })
+      .notNull(),
+    evaluationId: integer('evaluation_id')
+      .references(() => evalEvaluation.id, { onDelete: 'cascade' })
+      .notNull(),
+
+    userId: text('user_id').references(() => users.id, { onDelete: 'cascade' }),
+    ...timestamps,
+  },
+  (t) => ({
+    userIdIdx: index('rag_eval_evaluation_records_user_id_idx').on(t.userId),
   }),
-
-  duration: integer('duration'),
-  datasetRecordId: integer('dataset_record_id')
-    .references(() => evalDatasetRecords.id, { onDelete: 'cascade' })
-    .notNull(),
-  evaluationId: integer('evaluation_id')
-    .references(() => evalEvaluation.id, { onDelete: 'cascade' })
-    .notNull(),
-
-  userId: text('user_id').references(() => users.id, { onDelete: 'cascade' }),
-  ...timestamps,
-});
+);
 
 export type NewEvaluationRecordsItem = typeof evaluationRecords.$inferInsert;
 export type EvaluationRecordsSelectItem = typeof evaluationRecords.$inferSelect;

package/packages/database/src/schemas/relations.ts

@@ -31,6 +31,7 @@ export const agentsToSessions = pgTable(
     primaryKey({ columns: [t.agentId, t.sessionId] }),
     index('agents_to_sessions_session_id_idx').on(t.sessionId),
     index('agents_to_sessions_agent_id_idx').on(t.agentId),
+    index('agents_to_sessions_user_id_idx').on(t.userId),
   ],
 );
 
@@ -49,6 +50,9 @@ export const filesToSessions = pgTable(
   },
   (t) => ({
     pk: primaryKey({ columns: [t.fileId, t.sessionId] }),
+    userIdIdx: index('files_to_sessions_user_id_idx').on(t.userId),
+    fileIdIdx: index('files_to_sessions_file_id_idx').on(t.fileId),
+    sessionIdIdx: index('files_to_sessions_session_id_idx').on(t.sessionId),
   }),
 );
 
@@ -64,6 +68,9 @@ export const fileChunks = pgTable(
   },
   (t) => ({
     pk: primaryKey({ columns: [t.fileId, t.chunkId] }),
+    userIdIdx: index('file_chunks_user_id_idx').on(t.userId),
+    fileIdIdx: index('file_chunks_file_id_idx').on(t.fileId),
+    chunkIdIdx: index('file_chunks_chunk_id_idx').on(t.chunkId),
   }),
 );
 export type NewFileChunkItem = typeof fileChunks.$inferInsert;

package/packages/database/src/schemas/session.ts

@@ -29,6 +29,7 @@ export const sessionGroups = pgTable(
       table.clientId,
       table.userId,
     ),
+    userIdIdx: index('session_groups_user_id_idx').on(table.userId),
   }),
 );