add noexb flag for model

1 files changed, 6 insertions, 3 deletions

diff --git a/supergenerator.py b/supergenerator.py
index 4131734..19c235d 100644
--- a/supergenerator.py
+++ b/supergenerator.py
@@ -358,15 +358,18 @@ async def openai_generate_text(
     message_id,
 ) -> str:
     try:
+        extra_body = {
+            "think": "false"
+        }
+        if settings.get("noexb"):
+            extra_body = None
         
         completion = await oai.chat.completions.create(
             model=model,
             messages=messages,
             stream=stream,
             max_tokens=MAX_TOKENS,
-            extra_body={
-                "think": "false"
-            }
+            extra_body=extra_body
         )
         if stream:
             message = ""