add logging

This commit is contained in:
William Lane
2023-04-04 12:53:40 -07:00
parent edc79ec184
commit 7e0eeaa7ee
2 changed files with 27 additions and 12 deletions

View File

@@ -2,18 +2,28 @@ from dotenv import load_dotenv
import os import os
import discord import discord
from app.mabel.connect_openai import chatgpt_response from app.mabel.connect_openai import chatgpt_response
import logging
logging.basicConfig(level=logging.INFO)
#load_dotenv() #load_dotenv()
discord_token=os.getenv('DISCORD_TOKEN') discord_token=os.getenv('DISCORD_TOKEN')
class MyClient(discord.Client): class MyClient(discord.Client):
async def on_ready(self): async def on_ready(self):
print('Logged in as: ', self.user) logging.info('Logged in as: ', self.user)
async def on_message(self, message): async def on_message(self, message):
logging.info('Message from {0.author}: {0.content}'.format(message))
if message.author == client.user: if message.author == client.user:
return return
if "mabel" in message.content.lower(): if "mabel" in message.content.lower():
try:
response = await chatgpt_response(message.content) 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) await message.channel.send(response)
except Exception as e:
logging.info('DISCORD Error: ', e)
return return
intents=discord.Intents.default() intents=discord.Intents.default()

View File

@@ -28,6 +28,7 @@ async def chatgpt_response(prompt):
if "mabel" in prompt.lower(): if "mabel" in prompt.lower():
logger.info("PROMPT: " + prompt.lower()) logger.info("PROMPT: " + prompt.lower())
try:
response = openai.ChatCompletion.create( response = openai.ChatCompletion.create(
model="gpt-3.5-turbo", model="gpt-3.5-turbo",
messages=messages, messages=messages,
@@ -37,6 +38,10 @@ async def chatgpt_response(prompt):
logger.info("RESPONSE: " + response_content) logger.info("RESPONSE: " + response_content)
DIALOGUE_STACK.append({"role": "assistant", "content": response_content }) DIALOGUE_STACK.append({"role": "assistant", "content": response_content })
return response_content return response_content
except Exception as e:
logger.info("OPENAI Error: " + str(e))
return "NO RESPONSE"
else: else:
DIALOGUE_STACK.append({"role": "user", "content": prompt}) DIALOGUE_STACK.append({"role": "user", "content": prompt})
return return