diff options
| -rw-r--r-- | supergenerator.py | 9 |
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 = "" |