feat: allow custom seed (#6709)

2026-02-13 21:05:19 +00:00 · 2024-04-26 11:40:07 +00:00
parent 5d114ea965
commit b639e52dca
4 changed files with 59 additions and 79 deletions
--- a/packages/backend/server/src/plugins/copilot/controller.ts
+++ b/packages/backend/server/src/plugins/copilot/controller.ts
@@ -100,6 +100,17 @@ export class CopilotController {
    return controller.signal;
  }

+  private parseNumber(value: string | string[] | undefined) {
+    if (!value) {
+      return undefined;
+    }
+    const num = Number.parseInt(Array.isArray(value) ? value[0] : value, 10);
+    if (Number.isNaN(num)) {
+      return undefined;
+    }
+    return num;
+  }
+
  private handleError(err: any) {
    if (err instanceof Error) {
      const ret = {
@@ -256,6 +267,7 @@ export class CopilotController {

      return from(
        provider.generateImagesStream(session.finish(params), session.model, {
+          seed: this.parseNumber(params.seed),
          signal: this.getSignal(req),
          user: user.id,
        })