feat: allow custom seed (#6709)

packages/backend/server/src/plugins/copilot/providers/fal.ts

@@ -2,6 +2,7 @@ import assert from 'node:assert';
 
 import {
   CopilotCapability,
+  CopilotImageOptions,
   CopilotImageToImageProvider,
   CopilotProviderType,
   CopilotTextToImageProvider,
@@ -57,10 +58,7 @@ export class FalProvider
   async generateImages(
     messages: PromptMessage[],
     model: string = this.availableModels[0],
-    options: {
-      signal?: AbortSignal;
-      user?: string;
-    } = {}
+    options: CopilotImageOptions = {}
   ): Promise<Array<string>> {
     const { content, attachments } = messages.pop() || {};
     if (!this.availableModels.includes(model)) {
@@ -82,7 +80,7 @@ export class FalProvider
         image_url: attachments?.[0],
         prompt: content,
         sync_mode: true,
-        seed: 42,
+        seed: options.seed || 42,
         enable_safety_checks: false,
       }),
       signal: options.signal,
@@ -100,10 +98,7 @@ export class FalProvider
   async *generateImagesStream(
     messages: PromptMessage[],
     model: string = this.availableModels[0],
-    options: {
-      signal?: AbortSignal;
-      user?: string;
-    } = {}
+    options: CopilotImageOptions = {}
   ): AsyncIterable<string> {
     const ret = await this.generateImages(messages, model, options);
     for (const url of ret) {

packages/backend/server/src/plugins/copilot/providers/openai.ts

@@ -5,6 +5,9 @@ import { ClientOptions, OpenAI } from 'openai';
 import {
   ChatMessageRole,
   CopilotCapability,
+  CopilotChatOptions,
+  CopilotEmbeddingOptions,
+  CopilotImageOptions,
   CopilotImageToTextProvider,
   CopilotProviderType,
   CopilotTextToEmbeddingProvider,
@@ -147,12 +150,7 @@ export class OpenAIProvider
   async generateText(
     messages: PromptMessage[],
     model: string = 'gpt-3.5-turbo',
-    options: {
-      temperature?: number;
-      maxTokens?: number;
-      signal?: AbortSignal;
-      user?: string;
-    } = {}
+    options: CopilotChatOptions = {}
   ): Promise<string> {
     this.checkParams({ messages, model });
     const result = await this.instance.chat.completions.create(
@@ -175,12 +173,7 @@ export class OpenAIProvider
   async *generateTextStream(
     messages: PromptMessage[],
     model: string = 'gpt-3.5-turbo',
-    options: {
-      temperature?: number;
-      maxTokens?: number;
-      signal?: AbortSignal;
-      user?: string;
-    } = {}
+    options: CopilotChatOptions = {}
   ): AsyncIterable<string> {
     this.checkParams({ messages, model });
     const result = await this.instance.chat.completions.create(
@@ -214,11 +207,7 @@ export class OpenAIProvider
   async generateEmbedding(
     messages: string | string[],
     model: string,
-    options: {
-      dimensions: number;
-      signal?: AbortSignal;
-      user?: string;
-    } = { dimensions: DEFAULT_DIMENSIONS }
+    options: CopilotEmbeddingOptions = { dimensions: DEFAULT_DIMENSIONS }
   ): Promise<number[][]> {
     messages = Array.isArray(messages) ? messages : [messages];
     this.checkParams({ embeddings: messages, model });
@@ -236,10 +225,7 @@ export class OpenAIProvider
   async generateImages(
     messages: PromptMessage[],
     model: string = 'dall-e-3',
-    options: {
-      signal?: AbortSignal;
-      user?: string;
-    } = {}
+    options: CopilotImageOptions = {}
   ): Promise<Array<string>> {
     const { content: prompt } = messages.pop() || {};
     if (!prompt) {
@@ -261,10 +247,7 @@ export class OpenAIProvider
   async *generateImagesStream(
     messages: PromptMessage[],
     model: string = 'dall-e-3',
-    options: {
-      signal?: AbortSignal;
-      user?: string;
-    } = {}
+    options: CopilotImageOptions = {}
   ): AsyncIterable<string> {
     const ret = await this.generateImages(messages, model, options);
     for (const url of ret) {