Added variable for max

tokens
This commit is contained in:
Brad Ganley 2024-06-01 00:52:57 +00:00
parent 358a423c6b
commit daa69445f1

3
bot.py
View File

@ -8,6 +8,7 @@ from dotenv import load_dotenv
load_dotenv() load_dotenv()
DISCORD_TOKEN = os.getenv('DISCORD_TOKEN') DISCORD_TOKEN = os.getenv('DISCORD_TOKEN')
OPENAI_API_KEY = os.getenv('OPENAI_API_KEY') OPENAI_API_KEY = os.getenv('OPENAI_API_KEY')
MAX_TOKENS = int(os.getenv('MAX_TOKENS'))
aclient = AsyncOpenAI(api_key=OPENAI_API_KEY) aclient = AsyncOpenAI(api_key=OPENAI_API_KEY)
# Initialize OpenAI API # Initialize OpenAI API
@ -34,7 +35,7 @@ async def on_message(message):
async def ask_gpt(ctx, *, question: str): async def ask_gpt(ctx, *, question: str):
try: try:
# Send the question to GPT-4o using the new OpenAI API # Send the question to GPT-4o using the new OpenAI API
response = await aclient.chat.completions.create(model="gpt-4o", max_tokens=2000, response = await aclient.chat.completions.create(model="gpt-4o", max_tokens=MAX_TOKENS,
messages=[ 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": "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} {"role": "user", "content": question}