addressing multiple issues; error handling

This commit is contained in:
konrad 2023-03-21 18:21:12 +00:00
parent 669cc5aa59
commit 75197d214f
2 changed files with 39 additions and 20 deletions

1
.gitignore vendored
View File

@ -1,3 +1,2 @@
# ignore test app
chatgpt_new.py
chat.conf

View File

@ -2,6 +2,7 @@ import openai
import socket
import time
import configparser
from typing import Union, Tuple
# Read configuration from file
config = configparser.ConfigParser()
@ -17,11 +18,19 @@ channel = config.get('irc', 'channel')
nickname = config.get('irc', 'nickname')
# Connect to IRC server
irc = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
irc.connect((server, port))
irc.send(bytes("USER " + nickname + " " + nickname + " " + nickname + " :" + nickname + "\n", "UTF-8"))
irc.send(bytes("NICK " + nickname + "\n", "UTF-8"))
irc.send(bytes("JOIN " + channel + "\n", "UTF-8"))
while True:
try:
print ("connecting to:" + server)
irc = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
irc.connect((server, port))
irc.send(bytes("USER " + nickname + " " + nickname + " " + nickname + " :" + nickname + "\n", "UTF-8"))
irc.send(bytes("NICK " + nickname + "\n", "UTF-8"))
irc.send(bytes("JOIN " + channel + "\n", "UTF-8"))
print ("connected to:" + server)
break
except:
print("Connection failed. Retrying in 5 seconds...")
time.sleep(5)
# Listen for messages from users
while True:
@ -29,20 +38,31 @@ while True:
if message.find("PING") != -1:
# Respond to server PING requests
irc.send(bytes("PONG " + message.split()[1] + "\n", "UTF-8"))
elif message.find("KICK " + channel + " " + nickname) != -1:
irc.send(bytes("JOIN " + channel + "\n", "UTF-8"))
print("Kicked from channel. Rejoining...")
elif message.find("PRIVMSG " + channel + " :" + nickname + ":") != -1:
question = message.split(nickname + ":")[1].strip()
response = openai.Completion.create(
model="text-davinci-003",
prompt="Q: " + question + "\nA:",
temperature=0.8,
max_tokens=300,
top_p=1,
frequency_penalty=0,
presence_penalty=0
)
answers = [x.strip() for x in response.choices[0].text.strip().split('\n')]
for answer in answers:
while len(answer) > 0:
irc.send(bytes("PRIVMSG " + channel + " :" + answer[:400] + "\n", "UTF-8"))
answer = answer[400:]
try:
response = openai.Completion.create(
model="text-davinci-003",
prompt="Q: " + question + "\nA:",
temperature=0.8,
max_tokens=1000,
top_p=1,
frequency_penalty=0,
presence_penalty=0,
request_timeout=5 # Set request timeout to 3 minutes
)
answers = [x.strip() for x in response.choices[0].text.strip().split('\n')]
for answer in answers:
while len(answer) > 0:
irc.send(bytes("PRIVMSG " + channel + " :" + answer[:400] + "\n", "UTF-8"))
answer = answer[400:]
except openai.error.Timeout as e: # Catch the timeout error
print("Error: " + str(e))
irc.send(bytes("PRIVMSG " + channel + " :API call timed out. Try again later.\n", "UTF-8"))
except Exception as e:
print("Error: " + str(e))
irc.send(bytes("PRIVMSG " + channel + " :API call failed. Try again later.\n", "UTF-8"))
time.sleep(1)