From 05ef694cc3ab9605daf2b5699f3fa8773752a00d Mon Sep 17 00:00:00 2001
From: yeeyou <corner52@gmail.com>
Date: Tue, 28 Oct 2025 15:30:42 +0800
Subject: [PATCH] Add seed and seed_control options to LLM node

Introduces 'seed' and 'seed_control' parameters to allow deterministic or controlled randomization of the generation process. The seed is now included in the API request payload, and the class tracks the last used seed to support increment and decrement modes.
---
 openai_llm_node.py | 31 ++++++++++++++++++++++++++++---
 1 file changed, 28 insertions(+), 3 deletions(-)

diff --git a/openai_llm_node.py b/openai_llm_node.py
index d24f3f0..5172ed2 100644
--- a/openai_llm_node.py
+++ b/openai_llm_node.py
@@ -19,7 +19,7 @@ class OpenAICompatibleLLM:
     """
     
     def __init__(self):
-        pass
+        self.last_seed = 42
     
     @classmethod
     def INPUT_TYPES(cls):
@@ -47,6 +47,15 @@ class OpenAICompatibleLLM:
                     "max": 2.0,
                     "step": 0.01
                 }),
+                "seed": ("INT", {
+                    "default": 42,
+                    "min": 0,
+                    "max": 0xffffffffffffffff,
+                    "step": 1
+                }),
+                "seed_control": (["random", "fixed", "increment", "decrement"], {
+                    "default": "random"
+                }),
             },
             "optional": {
                 "image": ("IMAGE",),
@@ -86,12 +95,27 @@ class OpenAICompatibleLLM:
         
         return f"data:image/png;base64,{img_base64}"
     
-    def generate(self, prompt, endpoint, model, max_tokens, temperature, 
+    def generate(self, prompt, endpoint, model, max_tokens, temperature, seed, seed_control,
                  image=None, api_key=None, image_detail="auto"):
         """
         调用 OpenAI 兼容的 API 生成响应
         """
         
+        # Handle seed control
+        if seed_control == "random":
+            import random
+            actual_seed = random.randint(0, 0xffffffffffffffff)
+        elif seed_control == "fixed":
+            actual_seed = seed
+        elif seed_control == "increment":
+            actual_seed = self.last_seed + 1
+        elif seed_control == "decrement":
+            actual_seed = max(0, self.last_seed - 1)
+        else:
+            actual_seed = seed
+        
+        self.last_seed = actual_seed
+        
         # 构建消息内容
         if image is not None:
             # 如果有图片输入，使用视觉模型的格式
@@ -124,7 +148,8 @@ class OpenAICompatibleLLM:
                 }
             ],
             "max_tokens": max_tokens,
-            "temperature": temperature
+            "temperature": temperature,
+            "seed": actual_seed
         }
         
         # 设置请求头