From 56807e1f6b3cbf6a7dd687d88eebb20bcebb2bcb Mon Sep 17 00:00:00 2001
From: Max Dingsda <goumet.michel@gmail.com>
Date: Tue, 5 Aug 2025 20:12:26 +0200
Subject: [PATCH] added gpt-oss:20b model to prompt enhancer node

---
 ollama_prompt_booster.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/ollama_prompt_booster.py b/ollama_prompt_booster.py
index eebaa29..38961d3 100644
--- a/ollama_prompt_booster.py
+++ b/ollama_prompt_booster.py
@@ -15,12 +15,12 @@ class OllamaPromptBooster:
                     "tooltip": "If disabled, the prompt will pass through unchanged."
                 }),
                 "model": (
-                    ["zephyr:7b-beta", "deepseek-r1:8b", "llama3.2:latest", "mistral:latest"],
-                    {
-                        "default": "zephyr:7b-beta",
-                        "tooltip": "Choose which LLM to use for enhancement. No API key needed."
-                    }
-                ),
+                    ["zephyr:7b-beta", "deepseek-r1:8b", "llama3.2:latest", "mistral:latest", "gpt-oss:20b", "gpt-oss:120b"],
+                        {
+                            "default": "zephyr:7b-beta",
+                            "tooltip": "Choose which LLM to use for enhancement. No API key needed."
+                        }
+                        ),
                 "cleanup_output": ("BOOLEAN", {
                     "default": True,
                     "tooltip": "Remove extra formatting, <think> tags, or LLM commentary from output."