Compare commits

...

2 Commits

Author SHA1 Message Date
daa69445f1 Added variable for max
tokens
2024-06-01 00:52:57 +00:00
358a423c6b Added max_tokens. Actually, I should make that an environment variable. Fuck 2024-06-01 00:45:26 +00:00

3
bot.py
View File

@@ -8,6 +8,7 @@ from dotenv import load_dotenv
load_dotenv()
DISCORD_TOKEN = os.getenv('DISCORD_TOKEN')
OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
MAX_TOKENS = int(os.getenv('MAX_TOKENS'))
aclient = AsyncOpenAI(api_key=OPENAI_API_KEY)
# Initialize OpenAI API
@@ -34,7 +35,7 @@ async def on_message(message):
async def ask_gpt(ctx, *, question: str):
try:
# Send the question to GPT-4o using the new OpenAI API
response = await aclient.chat.completions.create(model="gpt-4o",
response = await aclient.chat.completions.create(model="gpt-4o", max_tokens=MAX_TOKENS,
messages=[
{"role": "system", "content": "You are a helpful assistant. You exist inside a discord server. There is a character limit so please keep responses brief if possible."},
{"role": "user", "content": question}