mirror of
https://github.com/AskDavis/nate-discord.git
synced 2025-12-31 20:55:58 -08:00
add logging
This commit is contained in:
@@ -2,18 +2,28 @@ from dotenv import load_dotenv
|
||||
import os
|
||||
import discord
|
||||
from app.mabel.connect_openai import chatgpt_response
|
||||
import logging
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
|
||||
#load_dotenv()
|
||||
discord_token=os.getenv('DISCORD_TOKEN')
|
||||
class MyClient(discord.Client):
|
||||
async def on_ready(self):
|
||||
print('Logged in as: ', self.user)
|
||||
logging.info('Logged in as: ', self.user)
|
||||
async def on_message(self, message):
|
||||
logging.info('Message from {0.author}: {0.content}'.format(message))
|
||||
if message.author == client.user:
|
||||
return
|
||||
if "mabel" in message.content.lower():
|
||||
response = await chatgpt_response(message.content)
|
||||
await message.channel.send(response)
|
||||
try:
|
||||
response = await chatgpt_response(message.content)
|
||||
except Exception as e:
|
||||
logging.info('CHATGPT Error: ', e)
|
||||
try:
|
||||
logging.info('Response: ', response)
|
||||
await message.channel.send(response)
|
||||
except Exception as e:
|
||||
logging.info('DISCORD Error: ', e)
|
||||
return
|
||||
|
||||
intents=discord.Intents.default()
|
||||
|
||||
@@ -28,15 +28,20 @@ async def chatgpt_response(prompt):
|
||||
|
||||
if "mabel" in prompt.lower():
|
||||
logger.info("PROMPT: " + prompt.lower())
|
||||
response = openai.ChatCompletion.create(
|
||||
model="gpt-3.5-turbo",
|
||||
messages=messages,
|
||||
max_tokens=1800,
|
||||
)
|
||||
response_content = response['choices'][0]['message']['content']
|
||||
logger.info("RESPONSE: " + response_content)
|
||||
DIALOGUE_STACK.append({"role": "assistant", "content": response_content })
|
||||
return response_content
|
||||
try:
|
||||
response = openai.ChatCompletion.create(
|
||||
model="gpt-3.5-turbo",
|
||||
messages=messages,
|
||||
max_tokens=1800,
|
||||
)
|
||||
response_content = response['choices'][0]['message']['content']
|
||||
logger.info("RESPONSE: " + response_content)
|
||||
DIALOGUE_STACK.append({"role": "assistant", "content": response_content })
|
||||
return response_content
|
||||
except Exception as e:
|
||||
logger.info("OPENAI Error: " + str(e))
|
||||
return "NO RESPONSE"
|
||||
|
||||
else:
|
||||
DIALOGUE_STACK.append({"role": "user", "content": prompt})
|
||||
return
|
||||
Reference in New Issue
Block a user