diff --git a/main.py b/main.py index 5fe515f..6ff814c 100644 --- a/main.py +++ b/main.py @@ -12,6 +12,7 @@ DISCORD_TOKEN = os.getenv("DISCORD_TOKEN", "placeholder") OPENAI_API_ENDPOINT = os.getenv("OPENAI_API_ENDPOINT") IMAGE_GEN_ENDPOINT = os.getenv("IMAGE_GEN_ENDPOINT") IMAGE_EDIT_ENDPOINT = os.getenv("IMAGE_EDIT_ENDPOINT") +MAX_COMPLETION_TOKENS = int(os.getenv("MAX_COMPLETION_TOKENS", "1000")) if not OPENAI_API_ENDPOINT: raise Exception("OPENAI_API_ENDPOINT required.") @@ -148,6 +149,7 @@ async def on_message(message): }, {"role": "user", "content": user_message}, ], + "max_completion_tokens": MAX_COMPLETION_TOKENS, } response_prefix = f"**{bot_name} response**"