@lobehub/chat 1.98.2 → 1.99.0

package/src/components/KeyValueEditor/index.tsx

@@ -0,0 +1,203 @@
+import { ActionIcon, Icon } from '@lobehub/ui';
+import { Button } from 'antd';
+import { createStyles } from 'antd-style';
+import fastDeepEqual from 'fast-deep-equal';
+import { LucidePlus, LucideTrash } from 'lucide-react';
+import { memo, useEffect, useRef, useState } from 'react';
+import { useTranslation } from 'react-i18next';
+import { Flexbox } from 'react-layout-kit';
+import { v4 as uuidv4 } from 'uuid';
+
+import { FormInput } from '@/components/FormInput';
+
+import { KeyValueItem, localListToRecord, recordToLocalList } from './utils';
+
+const useStyles = createStyles(({ css, token }) => ({
+  container: css`
+    position: relative;
+
+    width: 100%;
+    padding: 12px;
+    border: 1px solid ${token.colorBorderSecondary};
+    border-radius: ${token.borderRadiusLG}px;
+  `,
+  input: css`
+    font-family: ${token.fontFamilyCode};
+    font-size: 12px;
+  `,
+  row: css`
+    margin-block-end: 8px;
+
+    &:last-child {
+      margin-block-end: 0;
+    }
+  `,
+  title: css`
+    margin-block-end: 8px;
+    color: ${token.colorTextTertiary};
+  `,
+}));
+
+export interface KeyValueEditorProps {
+  addButtonText?: string;
+  deleteTooltip?: string;
+  duplicateKeyErrorText?: string;
+  keyPlaceholder?: string;
+  onChange?: (value: Record<string, string>) => void;
+  value?: Record<string, string>;
+  valuePlaceholder?: string;
+}
+
+const KeyValueEditor = memo<KeyValueEditorProps>(
+  ({
+    value,
+    onChange,
+    keyPlaceholder,
+    valuePlaceholder,
+    addButtonText,
+    duplicateKeyErrorText,
+    deleteTooltip,
+  }) => {
+    const { styles } = useStyles();
+    const { t } = useTranslation('components');
+    const [items, setItems] = useState<KeyValueItem[]>(() => recordToLocalList(value));
+    const prevValueRef = useRef<Record<string, string> | undefined>(undefined);
+
+    useEffect(() => {
+      const externalRecord = value || {};
+      if (!fastDeepEqual(externalRecord, prevValueRef.current)) {
+        setItems(recordToLocalList(externalRecord));
+        prevValueRef.current = externalRecord;
+      }
+    }, [value]);
+
+    const triggerChange = (newItems: KeyValueItem[]) => {
+      const keysCount: Record<string, number> = {};
+      newItems.forEach((item) => {
+        const trimmedKey = item.key.trim();
+        if (trimmedKey) {
+          keysCount[trimmedKey] = (keysCount[trimmedKey] || 0) + 1;
+        }
+      });
+      setItems(
+        newItems.map((item) => ({
+          ...item,
+        })),
+      );
+      onChange?.(localListToRecord(newItems));
+    };
+
+    const handleAdd = () => {
+      const newItems = [...items, { id: uuidv4(), key: '', value: '' }];
+      triggerChange(newItems);
+    };
+
+    const handleRemove = (id: string) => {
+      const newItems = items.filter((item) => item.id !== id);
+      triggerChange(newItems);
+    };
+
+    const handleKeyChange = (id: string, newKey: string) => {
+      const newItems = items.map((item) => (item.id === id ? { ...item, key: newKey } : item));
+      triggerChange(newItems);
+    };
+
+    const handleValueChange = (id: string, newValue: string) => {
+      const newItems = items.map((item) => (item.id === id ? { ...item, value: newValue } : item));
+      triggerChange(newItems);
+    };
+
+    const getDuplicateKeys = (currentItems: KeyValueItem[]): Set<string> => {
+      const keys = new Set<string>();
+      const duplicates = new Set<string>();
+      currentItems.forEach((item) => {
+        const trimmedKey = item.key.trim();
+        if (trimmedKey) {
+          if (keys.has(trimmedKey)) {
+            duplicates.add(trimmedKey);
+          } else {
+            keys.add(trimmedKey);
+          }
+        }
+      });
+      return duplicates;
+    };
+    const duplicateKeys = getDuplicateKeys(items);
+
+    return (
+      <div className={styles.container}>
+        <Flexbox className={styles.title} gap={8} horizontal>
+          <Flexbox flex={1}>{keyPlaceholder || t('KeyValueEditor.keyPlaceholder')}</Flexbox>
+          <Flexbox flex={2}>{valuePlaceholder || t('KeyValueEditor.valuePlaceholder')}</Flexbox>
+          <Flexbox style={{ width: 30 }} />
+        </Flexbox>
+        <Flexbox width={'100%'}>
+          {items.map((item) => {
+            const isDuplicate = item.key.trim() && duplicateKeys.has(item.key.trim());
+            return (
+              <Flexbox
+                align="flex-start"
+                className={styles.row}
+                gap={8}
+                horizontal
+                key={item.id}
+                width={'100%'}
+              >
+                <Flexbox flex={1} style={{ position: 'relative' }}>
+                  <FormInput
+                    className={styles.input}
+                    onChange={(e) => handleKeyChange(item.id, e)}
+                    placeholder={keyPlaceholder || t('KeyValueEditor.keyPlaceholder')}
+                    status={isDuplicate ? 'error' : undefined}
+                    value={item.key}
+                    variant={'filled'}
+                  />
+                  {isDuplicate && (
+                    <div
+                      style={{
+                        bottom: '-16px',
+                        color: 'red',
+                        fontSize: '12px',
+                        position: 'absolute',
+                      }}
+                    >
+                      {duplicateKeyErrorText || t('KeyValueEditor.duplicateKeyError')}
+                    </div>
+                  )}
+                </Flexbox>
+                <Flexbox flex={2}>
+                  <FormInput
+                    className={styles.input}
+                    onChange={(value) => handleValueChange(item.id, value)}
+                    placeholder={valuePlaceholder || t('KeyValueEditor.valuePlaceholder')}
+                    value={item.value}
+                    variant={'filled'}
+                  />
+                </Flexbox>
+                <ActionIcon
+                  icon={LucideTrash}
+                  onClick={() => handleRemove(item.id)}
+                  size={'small'}
+                  style={{ marginTop: 4 }}
+                  title={deleteTooltip || t('KeyValueEditor.deleteTooltip')}
+                />
+              </Flexbox>
+            );
+          })}
+          <Button
+            block
+            icon={<Icon icon={LucidePlus} />}
+            onClick={handleAdd}
+            size={'small'}
+            style={{ marginTop: items.length > 0 ? 16 : 8 }}
+            type="dashed"
+          >
+            {addButtonText || t('KeyValueEditor.addButton')}
+          </Button>
+        </Flexbox>
+      </div>
+    );
+  },
+);
+
+export default KeyValueEditor;

package/src/components/KeyValueEditor/utils.ts

@@ -0,0 +1,42 @@
+import { v4 as uuidv4 } from 'uuid';
+
+export interface KeyValueItem {
+  id: string;
+  key: string;
+  value: string;
+}
+
+export const recordToLocalList = (
+  record: Record<string, string> | undefined | null = {},
+): KeyValueItem[] =>
+  Object.entries(record || {}).map(([key, val]) => ({
+    id: uuidv4(),
+    key,
+    value: typeof val === 'string' ? val : '',
+  }));
+
+export const localListToRecord = (
+  list: KeyValueItem[] | undefined | null = [],
+): Record<string, string> => {
+  const record: Record<string, string> = {};
+  const keys = new Set<string>();
+  (list || [])
+    .slice()
+    .reverse()
+    .forEach((item) => {
+      const trimmedKey = item.key.trim();
+      if (trimmedKey && !keys.has(trimmedKey)) {
+        record[trimmedKey] = typeof item.value === 'string' ? item.value : '';
+        keys.add(trimmedKey);
+      }
+    });
+  return Object.keys(record)
+    .reverse()
+    .reduce(
+      (acc, key) => {
+        acc[key] = record[key];
+        return acc;
+      },
+      {} as Record<string, string>,
+    );
+};

package/src/config/aiModels/fal.ts

@@ -0,0 +1,52 @@
+import { AIImageModelCard } from '@/types/aiModel';
+
+import { fluxKontextDevParamsSchema } from '../paramsSchemas/fal/flux-kontext-dev';
+import { fluxProKontextParamsSchema } from '../paramsSchemas/fal/flux-pro-kontext';
+import { fluxSchnellParamsSchema } from '../paramsSchemas/fal/flux-schnell';
+import { imagen4ParamsSchema } from '../paramsSchemas/fal/imagen4';
+
+const falImageModels: AIImageModelCard[] = [
+  {
+    description: 'Frontier image editing model.',
+    displayName: 'FLUX.1 Kontext Dev',
+    enabled: true,
+    id: 'flux-kontext/dev',
+    parameters: fluxKontextDevParamsSchema,
+    releasedAt: '2025-06-28',
+    type: 'image',
+  },
+  {
+    description:
+      'FLUX.1 Kontext [pro] 能够处理文本和参考图像作为输入，无缝实现目标性的局部编辑和复杂的整体场景变换。',
+    displayName: 'FLUX.1 Kontext [pro]',
+    enabled: true,
+    id: 'flux-pro/kontext',
+    parameters: fluxProKontextParamsSchema,
+    releasedAt: '2025-05-01',
+    type: 'image',
+  },
+  {
+    description:
+      'FLUX.1 [schnell] 是一个拥有120亿参数的流式转换器模型，能够在1到4步内从文本生成高质量图像，适合个人和商业用途。',
+    displayName: 'FLUX.1 Schnell',
+    enabled: true,
+    id: 'flux/schnell',
+    parameters: fluxSchnellParamsSchema,
+    releasedAt: '2024-08-01',
+    type: 'image',
+  },
+  {
+    description: 'Google 最高质量的图像生成模型',
+    displayName: 'Imagen 4',
+    enabled: true,
+    id: 'imagen4/preview',
+    organization: 'Deepmind',
+    parameters: imagen4ParamsSchema,
+    releasedAt: '2025-05-21',
+    type: 'image',
+  },
+];
+
+export const allModels = [...falImageModels];
+
+export default allModels;

package/src/config/aiModels/index.ts

@@ -10,6 +10,7 @@ import { default as bedrock } from './bedrock';
 import { default as cloudflare } from './cloudflare';
 import { default as cohere } from './cohere';
 import { default as deepseek } from './deepseek';
+import { default as fal } from './fal';
 import { default as fireworksai } from './fireworksai';
 import { default as giteeai } from './giteeai';
 import { default as github } from './github';
@@ -85,6 +86,7 @@ export const LOBE_DEFAULT_MODEL_LIST = buildDefaultModelList({
   cloudflare,
   cohere,
   deepseek,
+  fal,
   fireworksai,
   giteeai,
   github,
@@ -141,6 +143,7 @@ export { default as bedrock } from './bedrock';
 export { default as cloudflare } from './cloudflare';
 export { default as cohere } from './cohere';
 export { default as deepseek } from './deepseek';
+export { default as fal } from './fal';
 export { default as fireworksai } from './fireworksai';
 export { default as giteeai } from './giteeai';
 export { default as github } from './github';

package/src/config/aiModels/openai.ts

@@ -1,10 +1,11 @@
+import { gptImage1ParamsSchema } from '@/config/paramsSchemas/openai/gpt-image-1';
 import {
   AIChatModelCard,
   AIEmbeddingModelCard,
+  AIImageModelCard,
   AIRealtimeModelCard,
   AISTTModelCard,
   AITTSModelCard,
-  AIText2ImageModelCard,
 } from '@/types/aiModel';
 
 export const openaiChatModels: AIChatModelCard[] = [
@@ -691,7 +692,8 @@ export const openaiChatModels: AIChatModelCard[] = [
       vision: true,
     },
     contextWindowTokens: 200_000,
-    description: 'codex-mini-latest 是 o4-mini 的微调版本，专门用于 Codex CLI。对于直接通过 API 使用，我们推荐从 gpt-4.1 开始。',
+    description:
+      'codex-mini-latest 是 o4-mini 的微调版本，专门用于 Codex CLI。对于直接通过 API 使用，我们推荐从 gpt-4.1 开始。',
     displayName: 'Codex mini',
     id: 'codex-mini-latest',
     maxOutput: 100_000,
@@ -713,7 +715,8 @@ export const openaiChatModels: AIChatModelCard[] = [
       vision: true,
     },
     contextWindowTokens: 8192,
-    description: 'computer-use-preview 模型是专为“计算机使用工具”设计的专用模型，经过训练以理解并执行计算机相关任务。',
+    description:
+      'computer-use-preview 模型是专为“计算机使用工具”设计的专用模型，经过训练以理解并执行计算机相关任务。',
     displayName: 'Computer Use Preview',
     id: 'computer-use-preview',
     maxOutput: 1024,
@@ -804,7 +807,8 @@ export const openaiSTTModels: AISTTModelCard[] = [
   },
   {
     contextWindowTokens: 16_000,
-    description: 'GPT-4o Transcribe 是一种使用 GPT-4o 转录音频的语音转文本模型。与原始 Whisper 模型相比，它提高了单词错误率，并提高了语言识别和准确性。使用它来获得更准确的转录。',
+    description:
+      'GPT-4o Transcribe 是一种使用 GPT-4o 转录音频的语音转文本模型。与原始 Whisper 模型相比，它提高了单词错误率，并提高了语言识别和准确性。使用它来获得更准确的转录。',
     displayName: 'GPT-4o Transcribe',
     id: 'gpt-4o-transcribe',
     maxOutput: 2000,
@@ -816,7 +820,8 @@ export const openaiSTTModels: AISTTModelCard[] = [
   },
   {
     contextWindowTokens: 16_000,
-    description: 'GPT-4o Mini Transcribe 是一种使用 GPT-4o 转录音频的语音转文本模型。与原始 Whisper 模型相比，它提高了单词错误率，并提高了语言识别和准确性。使用它来获得更准确的转录。',
+    description:
+      'GPT-4o Mini Transcribe 是一种使用 GPT-4o 转录音频的语音转文本模型。与原始 Whisper 模型相比，它提高了单词错误率，并提高了语言识别和准确性。使用它来获得更准确的转录。',
     displayName: 'GPT-4o Mini Transcribe',
     id: 'gpt-4o-mini-transcribe',
     maxOutput: 2000,
@@ -829,7 +834,7 @@ export const openaiSTTModels: AISTTModelCard[] = [
 ];
 
 // 图像生成模型
-export const openaiImageModels: AIText2ImageModelCard[] = [
+export const openaiImageModels: AIImageModelCard[] = [
   {
     description:
       '最新的 DALL·E 模型，于2023年11月发布。支持更真实、准确的图像生成，具有更强的细节表现力',
@@ -852,6 +857,15 @@ export const openaiImageModels: AIText2ImageModelCard[] = [
     resolutions: ['256x256', '512x512', '1024x1024'],
     type: 'image',
   },
+  // https://platform.openai.com/docs/models/gpt-image-1
+  {
+    description: 'ChatGPT 原生多模态图片生成模型',
+    displayName: 'GPT Image 1',
+    enabled: true,
+    id: 'gpt-image-1',
+    parameters: gptImage1ParamsSchema,
+    type: 'image',
+  },
 ];
 
 // GPT-4o 和 GPT-4o-mini 实时模型

package/src/config/llm.ts

@@ -163,6 +163,9 @@ export const getLLMConfig = () => {
       ENABLED_INFINIAI: z.boolean(),
       INFINIAI_API_KEY: z.string().optional(),
 
+      ENABLED_FAL: z.boolean(),
+      FAL_API_KEY: z.string().optional(),
+
       ENABLED_MODELSCOPE: z.boolean(),
       MODELSCOPE_API_KEY: z.string().optional(),
 
@@ -329,6 +332,9 @@ export const getLLMConfig = () => {
       ENABLED_INFINIAI: !!process.env.INFINIAI_API_KEY,
       INFINIAI_API_KEY: process.env.INFINIAI_API_KEY,
 
+      ENABLED_FAL: process.env.ENABLED_FAL !== '0',
+      FAL_API_KEY: process.env.FAL_API_KEY,
+
       ENABLED_MODELSCOPE: !!process.env.MODELSCOPE_API_KEY,
       MODELSCOPE_API_KEY: process.env.MODELSCOPE_API_KEY,
 

package/src/config/modelProviders/fal.ts

@@ -0,0 +1,21 @@
+import { ModelProviderCard } from '@/types/llm';
+
+/**
+ * @see https://fal.ai/models/fal-ai/flux/schnell
+ */
+const Fal: ModelProviderCard = {
+  chatModels: [],
+  description: '面向开发者的生成式媒体平台',
+  enabled: true,
+  id: 'fal',
+  name: 'Fal',
+  settings: {
+    disableBrowserRequest: true,
+    showAddNewModel: false,
+    showChecker: false,
+    showModelFetcher: false,
+  },
+  url: 'https://fal.ai',
+};
+
+export default Fal;

package/src/config/modelProviders/index.ts

@@ -10,6 +10,7 @@ import BedrockProvider from './bedrock';
 import CloudflareProvider from './cloudflare';
 import CohereProvider from './cohere';
 import DeepSeekProvider from './deepseek';
+import FalProvider from './fal';
 import FireworksAIProvider from './fireworksai';
 import GiteeAIProvider from './giteeai';
 import GithubProvider from './github';
@@ -123,6 +124,7 @@ export const DEFAULT_MODEL_PROVIDER_LIST = [
   DeepSeekProvider,
   HuggingFaceProvider,
   OpenRouterProvider,
+  FalProvider,
   CloudflareProvider,
   GithubProvider,
   NovitaProvider,
@@ -185,6 +187,7 @@ export { default as BedrockProviderCard } from './bedrock';
 export { default as CloudflareProviderCard } from './cloudflare';
 export { default as CohereProviderCard } from './cohere';
 export { default as DeepSeekProviderCard } from './deepseek';
+export { default as FalProviderCard } from './fal';
 export { default as FireworksAIProviderCard } from './fireworksai';
 export { default as GiteeAIProviderCard } from './giteeai';
 export { default as GithubProviderCard } from './github';

package/src/config/paramsSchemas/fal/flux-kontext-dev.ts

@@ -0,0 +1,8 @@
+import { type ModelParamsSchema } from '@/libs/standard-parameters/meta-schema';
+
+export const fluxKontextDevParamsSchema: ModelParamsSchema = {
+  imageUrl: { default: null },
+  prompt: { default: '' },
+  seed: { default: null },
+  steps: { default: 28, max: 50, min: 10 },
+};

package/src/config/paramsSchemas/fal/flux-pro-kontext.ts

@@ -0,0 +1,11 @@
+import { type ModelParamsSchema } from '@/libs/standard-parameters/meta-schema';
+
+export const fluxProKontextParamsSchema: ModelParamsSchema = {
+  aspectRatio: {
+    default: '1:1',
+    enum: ['21:9', '16:9', '4:3', '3:2', '1:1', '2:3', '3:4', '9:16', '9:21'],
+  },
+  imageUrl: { default: null },
+  prompt: { default: '' },
+  seed: { default: null },
+};

package/src/config/paramsSchemas/fal/flux-schnell.ts

@@ -0,0 +1,9 @@
+import { type ModelParamsSchema } from '@/libs/standard-parameters/meta-schema';
+
+export const fluxSchnellParamsSchema: ModelParamsSchema = {
+  height: { default: 1024, max: 1536, min: 512, step: 1 },
+  prompt: { default: '' },
+  seed: { default: null },
+  steps: { default: 4, max: 12, min: 1 },
+  width: { default: 1024, max: 1536, min: 512, step: 1 },
+};

package/src/config/paramsSchemas/fal/imagen4.ts

@@ -0,0 +1,10 @@
+import { type ModelParamsSchema } from '@/libs/standard-parameters/meta-schema';
+
+export const imagen4ParamsSchema: ModelParamsSchema = {
+  aspectRatio: {
+    default: '1:1',
+    enum: ['1:1', '16:9', '9:16', '3:4', '4:3'],
+  },
+  prompt: { default: '' },
+  seed: { default: null },
+};

package/src/config/paramsSchemas/openai/gpt-image-1.ts

@@ -0,0 +1,10 @@
+import { type ModelParamsSchema } from '@/libs/standard-parameters/meta-schema';
+
+export const gptImage1ParamsSchema: ModelParamsSchema = {
+  imageUrls: { default: [] },
+  prompt: { default: '' },
+  size: {
+    default: 'auto',
+    enum: ['auto', '1024x1024', '1536x1024', '1024x1536'],
+  },
+};

package/src/const/hotkeys.ts

@@ -41,13 +41,13 @@ export const HOTKEYS_REGISTRATION: HotkeyRegistration = [
     group: HotkeyGroupEnum.Essential,
     id: HotkeyEnum.ToggleLeftPanel,
     keys: combineKeys([KeyEnum.Mod, KeyEnum.BracketLeft]),
-    scopes: [HotkeyScopeEnum.Chat, HotkeyScopeEnum.Files],
+    scopes: [HotkeyScopeEnum.Chat, HotkeyScopeEnum.Files, HotkeyScopeEnum.Image],
   },
   {
     group: HotkeyGroupEnum.Essential,
     id: HotkeyEnum.ToggleRightPanel,
     keys: combineKeys([KeyEnum.Mod, KeyEnum.BracketRight]),
-    scopes: [HotkeyScopeEnum.Chat],
+    scopes: [HotkeyScopeEnum.Chat, HotkeyScopeEnum.Image],
   },
   {
     group: HotkeyGroupEnum.Essential,

package/src/const/image.ts

@@ -0,0 +1,6 @@
+/**
+ * 默认宽高比，当模型不支持原生宽高比时使用
+ */
+export const DEFAULT_ASPECT_RATIO = '1:1';
+
+export const PRESET_ASPECT_RATIOS = [DEFAULT_ASPECT_RATIO, '16:9', '9:16', '4:3', '3:4'];

package/src/const/settings/systemAgent.ts

@@ -22,6 +22,7 @@ export const DEFAULT_QUERY_REWRITE_SYSTEM_AGENT_ITEM: QueryRewriteSystemAgent =
 
 export const DEFAULT_SYSTEM_AGENT_CONFIG: UserSystemAgentConfig = {
   agentMeta: DEFAULT_SYSTEM_AGENT_ITEM,
+  generationTopic: DEFAULT_SYSTEM_AGENT_ITEM,
   historyCompress: DEFAULT_SYSTEM_AGENT_ITEM,
   queryRewrite: DEFAULT_QUERY_REWRITE_SYSTEM_AGENT_ITEM,
   thread: DEFAULT_SYSTEM_AGENT_ITEM,

package/src/database/client/migrations.json

@@ -514,5 +514,32 @@
     "bps": true,
     "folderMillis": 1749309388370,
     "hash": "39cea379f08ee4cb944875c0b67f7791387b508c2d47958bb4cd501ed1ef33eb"
+  },
+  {
+    "sql": [
+      "-- Migration to apply specific autovacuum settings to high-traffic tables\n-- This is crucial to prevent table and TOAST bloat for 'embeddings' and 'chunks'\n-- https://github.com/lobehub/lobe-chat/issues/8316\n\n-- Tuning for the 'embeddings' table\n-- Default scale factor (0.2) is too high, leading to infrequent vacuuming.\n-- Lowering to 2% to ensure frequent cleanup.\nALTER TABLE \"embeddings\" SET (autovacuum_vacuum_scale_factor = 0.02, autovacuum_vacuum_threshold = 1000);\n\n",
+      "\n\n-- Tuning for the 'chunks' table\n-- This table also experiences many updates/deletes and requires similar tuning.\nALTER TABLE \"chunks\" SET (autovacuum_vacuum_scale_factor = 0.02, autovacuum_vacuum_threshold = 1000);\n"
+    ],
+    "bps": true,
+    "folderMillis": 1752212281564,
+    "hash": "e445db851f792aac49482e04689569a37939f7d0a3851cbcdf10dc24e8943b2c"
+  },
+  {
+    "sql": [
+      "CREATE TABLE \"generation_batches\" (\n\t\"id\" text PRIMARY KEY NOT NULL,\n\t\"user_id\" text NOT NULL,\n\t\"generation_topic_id\" text NOT NULL,\n\t\"provider\" text NOT NULL,\n\t\"model\" text NOT NULL,\n\t\"prompt\" text NOT NULL,\n\t\"width\" integer,\n\t\"height\" integer,\n\t\"ratio\" varchar(64),\n\t\"config\" jsonb,\n\t\"accessed_at\" timestamp with time zone DEFAULT now() NOT NULL,\n\t\"created_at\" timestamp with time zone DEFAULT now() NOT NULL,\n\t\"updated_at\" timestamp with time zone DEFAULT now() NOT NULL\n);\n",
+      "\nCREATE TABLE \"generation_topics\" (\n\t\"id\" text PRIMARY KEY NOT NULL,\n\t\"user_id\" text NOT NULL,\n\t\"title\" text,\n\t\"cover_url\" text,\n\t\"accessed_at\" timestamp with time zone DEFAULT now() NOT NULL,\n\t\"created_at\" timestamp with time zone DEFAULT now() NOT NULL,\n\t\"updated_at\" timestamp with time zone DEFAULT now() NOT NULL\n);\n",
+      "\nCREATE TABLE \"generations\" (\n\t\"id\" text PRIMARY KEY NOT NULL,\n\t\"user_id\" text NOT NULL,\n\t\"generation_batch_id\" varchar(64) NOT NULL,\n\t\"async_task_id\" uuid,\n\t\"file_id\" text,\n\t\"seed\" integer,\n\t\"asset\" jsonb,\n\t\"accessed_at\" timestamp with time zone DEFAULT now() NOT NULL,\n\t\"created_at\" timestamp with time zone DEFAULT now() NOT NULL,\n\t\"updated_at\" timestamp with time zone DEFAULT now() NOT NULL\n);\n",
+      "\nALTER TABLE \"files\" ADD COLUMN \"source\" text;",
+      "\nALTER TABLE \"generation_batches\" ADD CONSTRAINT \"generation_batches_user_id_users_id_fk\" FOREIGN KEY (\"user_id\") REFERENCES \"public\".\"users\"(\"id\") ON DELETE cascade ON UPDATE no action;",
+      "\nALTER TABLE \"generation_batches\" ADD CONSTRAINT \"generation_batches_generation_topic_id_generation_topics_id_fk\" FOREIGN KEY (\"generation_topic_id\") REFERENCES \"public\".\"generation_topics\"(\"id\") ON DELETE cascade ON UPDATE no action;",
+      "\nALTER TABLE \"generation_topics\" ADD CONSTRAINT \"generation_topics_user_id_users_id_fk\" FOREIGN KEY (\"user_id\") REFERENCES \"public\".\"users\"(\"id\") ON DELETE cascade ON UPDATE no action;",
+      "\nALTER TABLE \"generations\" ADD CONSTRAINT \"generations_user_id_users_id_fk\" FOREIGN KEY (\"user_id\") REFERENCES \"public\".\"users\"(\"id\") ON DELETE cascade ON UPDATE no action;",
+      "\nALTER TABLE \"generations\" ADD CONSTRAINT \"generations_generation_batch_id_generation_batches_id_fk\" FOREIGN KEY (\"generation_batch_id\") REFERENCES \"public\".\"generation_batches\"(\"id\") ON DELETE cascade ON UPDATE no action;",
+      "\nALTER TABLE \"generations\" ADD CONSTRAINT \"generations_async_task_id_async_tasks_id_fk\" FOREIGN KEY (\"async_task_id\") REFERENCES \"public\".\"async_tasks\"(\"id\") ON DELETE set null ON UPDATE no action;",
+      "\nALTER TABLE \"generations\" ADD CONSTRAINT \"generations_file_id_files_id_fk\" FOREIGN KEY (\"file_id\") REFERENCES \"public\".\"files\"(\"id\") ON DELETE cascade ON UPDATE no action;"
+    ],
+    "bps": true,
+    "folderMillis": 1752413805765,
+    "hash": "abed92b1356df6d7eb35c03f47fbbdcdaf25aefda750dc3e4963c1c2a0d38b54"
   }
 ]

package/src/database/migrations/0026_add_autovacuum_tuning.sql

@@ -7,6 +7,8 @@
 -- Lowering to 2% to ensure frequent cleanup.
 ALTER TABLE "embeddings" SET (autovacuum_vacuum_scale_factor = 0.02, autovacuum_vacuum_threshold = 1000);
 
+--> statement-breakpoint
+
 -- Tuning for the 'chunks' table
 -- This table also experiences many updates/deletes and requires similar tuning.
 ALTER TABLE "chunks" SET (autovacuum_vacuum_scale_factor = 0.02, autovacuum_vacuum_threshold = 1000);

package/src/database/migrations/0027_ai_image.sql

@@ -0,0 +1,47 @@
+CREATE TABLE "generation_batches" (
+	"id" text PRIMARY KEY NOT NULL,
+	"user_id" text NOT NULL,
+	"generation_topic_id" text NOT NULL,
+	"provider" text NOT NULL,
+	"model" text NOT NULL,
+	"prompt" text NOT NULL,
+	"width" integer,
+	"height" integer,
+	"ratio" varchar(64),
+	"config" jsonb,
+	"accessed_at" timestamp with time zone DEFAULT now() NOT NULL,
+	"created_at" timestamp with time zone DEFAULT now() NOT NULL,
+	"updated_at" timestamp with time zone DEFAULT now() NOT NULL
+);
+--> statement-breakpoint
+CREATE TABLE "generation_topics" (
+	"id" text PRIMARY KEY NOT NULL,
+	"user_id" text NOT NULL,
+	"title" text,
+	"cover_url" text,
+	"accessed_at" timestamp with time zone DEFAULT now() NOT NULL,
+	"created_at" timestamp with time zone DEFAULT now() NOT NULL,
+	"updated_at" timestamp with time zone DEFAULT now() NOT NULL
+);
+--> statement-breakpoint
+CREATE TABLE "generations" (
+	"id" text PRIMARY KEY NOT NULL,
+	"user_id" text NOT NULL,
+	"generation_batch_id" varchar(64) NOT NULL,
+	"async_task_id" uuid,
+	"file_id" text,
+	"seed" integer,
+	"asset" jsonb,
+	"accessed_at" timestamp with time zone DEFAULT now() NOT NULL,
+	"created_at" timestamp with time zone DEFAULT now() NOT NULL,
+	"updated_at" timestamp with time zone DEFAULT now() NOT NULL
+);
+--> statement-breakpoint
+ALTER TABLE "files" ADD COLUMN "source" text;--> statement-breakpoint
+ALTER TABLE "generation_batches" ADD CONSTRAINT "generation_batches_user_id_users_id_fk" FOREIGN KEY ("user_id") REFERENCES "public"."users"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "generation_batches" ADD CONSTRAINT "generation_batches_generation_topic_id_generation_topics_id_fk" FOREIGN KEY ("generation_topic_id") REFERENCES "public"."generation_topics"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "generation_topics" ADD CONSTRAINT "generation_topics_user_id_users_id_fk" FOREIGN KEY ("user_id") REFERENCES "public"."users"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "generations" ADD CONSTRAINT "generations_user_id_users_id_fk" FOREIGN KEY ("user_id") REFERENCES "public"."users"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "generations" ADD CONSTRAINT "generations_generation_batch_id_generation_batches_id_fk" FOREIGN KEY ("generation_batch_id") REFERENCES "public"."generation_batches"("id") ON DELETE cascade ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "generations" ADD CONSTRAINT "generations_async_task_id_async_tasks_id_fk" FOREIGN KEY ("async_task_id") REFERENCES "public"."async_tasks"("id") ON DELETE set null ON UPDATE no action;--> statement-breakpoint
+ALTER TABLE "generations" ADD CONSTRAINT "generations_file_id_files_id_fk" FOREIGN KEY ("file_id") REFERENCES "public"."files"("id") ON DELETE cascade ON UPDATE no action;