Compare commits
2 Commits
18b0e79fb9
...
daa69445f1
Author | SHA1 | Date | |
---|---|---|---|
daa69445f1 | |||
358a423c6b |
3
bot.py
3
bot.py
@ -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",
|
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}
|
||||||
|
Loading…
Reference in New Issue
Block a user