Added variable for max
tokens
This commit is contained in:
parent
358a423c6b
commit
daa69445f1
3
bot.py
3
bot.py
@ -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", max_tokens=2000,
|
||||
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}
|
||||
|
Loading…
Reference in New Issue
Block a user