services/vibe-bot
1
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

limit max tokens

Browse Source
This commit is contained in:
Reese Wells
2026-03-03 11:14:05 -05:00
parent 493b6b8425
commit c547edc44b
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@@ -12,6 +12,7 @@ DISCORD_TOKEN = os.getenv("DISCORD_TOKEN", "placeholder")
OPENAI_API_ENDPOINT = os.getenv("OPENAI_API_ENDPOINT") OPENAI_API_ENDPOINT = os.getenv("OPENAI_API_ENDPOINT")
IMAGE_GEN_ENDPOINT = os.getenv("IMAGE_GEN_ENDPOINT") IMAGE_GEN_ENDPOINT = os.getenv("IMAGE_GEN_ENDPOINT")
IMAGE_EDIT_ENDPOINT = os.getenv("IMAGE_EDIT_ENDPOINT") IMAGE_EDIT_ENDPOINT = os.getenv("IMAGE_EDIT_ENDPOINT")
MAX_COMPLETION_TOKENS = int(os.getenv("MAX_COMPLETION_TOKENS", "1000"))
if not OPENAI_API_ENDPOINT: if not OPENAI_API_ENDPOINT:
raise Exception("OPENAI_API_ENDPOINT required.") raise Exception("OPENAI_API_ENDPOINT required.")
@@ -148,6 +149,7 @@ async def on_message(message):
}, },
{"role": "user", "content": user_message}, {"role": "user", "content": user_message},
], ],
"max_completion_tokens": MAX_COMPLETION_TOKENS,
} }
response_prefix = f"**{bot_name} response**" response_prefix = f"**{bot_name} response**"