Piteball
/
chatgpt
mirror of https://github.com/knrd1/chatgpt.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fixes and updates

This commit is contained in:
konrad 2023-09-18 23:00:55 +01:00
parent 48d3cbca64
commit 3dd7456696
2 changed files with 42 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
README.md
View File

@ -38,7 +38,7 @@ request_timeout = 60
server = open.ircnet.net
port = 6667
ssl = false
channels = #knrd1,#github
channels = #linux,#github
nickname = MyBot
ident = mybot
realname = My Bot
@ -48,6 +48,15 @@ password =
```
$ python3 chatgpt.py
```
Use screen to run bot in the background and keep it running even after you log out of your session:
```
$ screen python3 chatgpt.py
```
To detach from the screen session (leaving your ChatGPT IRC Bot running in the background), press Ctrl + A followed by d (for "detach").
If you need to reattach to the screen session later, use the following command:
```
screen -r
```
### Interaction:
ChatGPT IRC Bot will interact only if you mention its nickname:
```
@ -59,7 +68,7 @@ ChatGPT IRC Bot will interact only if you mention its nickname:
```
If you set the model to "dalle", the ChatGPT IRC Bot will return a shortened URL to the generated image:
```
17:33:16 <@knrd1> ChatGPT: two horses dancing on the street
17:33:16 <@knrd1> ChatGPT: impressionist style painting: two horses dancing on the street
17:33:23 < ChatGPT> https://tinyurl.com/2hr5uf4w
```
### Model endpoint compatibility
@ -67,28 +76,31 @@ If you set the model to "dalle", the ChatGPT IRC Bot will return a shortened URL
ChatGPT IRC Bot can use three API endpoints.
Following models support endpoint /v1/chat/completions:
```
gpt-4, gpt-4-0613, gpt-4-32k, gpt-4-32k-0613, gpt-3.5-turbo, gpt-3.5-turbo-0613, gpt-3.5-turbo-16k, gpt-3.5-turbo-16k-0613
```
> gpt-4, gpt-4-0613, gpt-4-32k, gpt-4-32k-0613, gpt-3.5-turbo, gpt-3.5-turbo-0613, gpt-3.5-turbo-16k, gpt-3.5-turbo-16k-0613
Models that support /v1/completions (Legacy):
```
text-davinci-003, text-davinci-002, text-davinci-001, text-curie-001, text-babbage-001, text-ada-001, davinci, curie, babbage, ada
```
> gpt-3.5-turbo-instruct, text-davinci-003, text-davinci-002, text-davinci-001, text-curie-001, text-babbage-001, text-ada-001, davinci, curie, babbage, ada, babbage-002, davinci-002
Create an image using endpoint /v1/images/generations:
```
dalle
```
> dalle
More details about models: https://platform.openai.com/docs/models
### Docker
To build the Docker image, you can use the following command:
```docker build -t my-chatgpt-app .```
```
docker build -t my-chatgpt-app .
```
To run the Docker container, you can use the following command:
```docker run -it my-chatgpt-app```
```
docker run -it my-chatgpt-app
```
To detach from a running Docker, press Ctrl + P. While holding down Ctrl, press Q.
To reattach to the container later, you can use the following command:
```docker attach container_id```
To reattach to the container later, use the following command:
```
docker attach <container_id>
```

20
chatgpt.py
View File

@ -33,6 +33,10 @@ ident = config.get('irc', 'ident')
realname = config.get('irc', 'realname')
password = config.get('irc', 'password')
# Define the list of models
completion_models = ["gpt-3.5-turbo-instruct", "text-davinci-003", "text-davinci-002", "text-davinci-001", "text-curie-001", "text-babbage-001", "text-ada-001", "davinci", "curie", "babbage", "ada", "babbage-002", "davinci-002"]
chatcompletion_models = ["gpt-4", "gpt-4-0613", "gpt-4-32k", "gpt-4-32k-0613", "gpt-3.5-turbo", "gpt-3.5-turbo-0613", "gpt-3.5-turbo-16k", "gpt-3.5-turbo-16k-0613"]
# Connect to IRC server
def connect(server, port, usessl, password, ident, realname, nickname, channels):
while True:
@ -57,7 +61,7 @@ def connect(server, port, usessl, password, ident, realname, nickname, channels)
irc = connect(server, port, usessl, password, ident, realname, nickname, channels)
# Listen for messages from users
# Listen for messages from users and answer questions
while True:
try:
data = irc.recv(4096).decode("UTF-8")
@ -83,7 +87,7 @@ while True:
time.sleep(5)
irc = connect(server, port, usessl, password, ident, realname, nickname, channels)
elif command == "471" or command == "473" or command == "474" or command == "475":
print("Unable to join " + chunk[3] + ": it can be full, invite only, bot is banned or need a key.")
print("Unable to join " + chunk[3] + ": Channel can be full, invite only, bot is banned or needs a key.")
elif command == "KICK" and chunk[3] == nickname:
irc.send(bytes("JOIN " + chunk[2] + "\n", "UTF-8"))
print("Kicked from channel " + chunk[2] + ". Rejoining...")
@ -94,7 +98,7 @@ while True:
elif command == "PRIVMSG" and chunk[2].startswith("#") and chunk[3] == ":" + nickname + ":":
channel = chunk[2].strip()
question = data.split(nickname + ":")[1].strip()
if model in ["gpt-4", "gpt-4-0613", "gpt-4-32k", "gpt-4-32k-0613", "gpt-3.5-turbo", "gpt-3.5-turbo-0613", "gpt-3.5-turbo-16k", "gpt-3.5-turbo-16k-0613"]:
if model in chatcompletion_models:
try:
response = openai.ChatCompletion.create(
model=model,
@ -123,8 +127,8 @@ while True:
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"))
elif model in ["text-davinci-003", "text-davinci-002", "text-davinci-001", "text-curie-001", "text-babbage-001", "text-ada-001", "davinci", "curie", "babbage", "ada"]:
irc.send(bytes("PRIVMSG " + channel + " :API call failed. Check console for error message.\n", "UTF-8"))
elif model in completion_models:
try:
response = openai.Completion.create(
model=model,
@ -153,8 +157,8 @@ while True:
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"))
elif model in ["dalle"]:
irc.send(bytes("PRIVMSG " + channel + " :API call failed. Check console for error message.\n", "UTF-8"))
elif model == "dalle":
try:
response = openai.Image.create(
prompt="Q: " + question + "\nA:",
@ -167,7 +171,7 @@ while True:
irc.send(bytes("PRIVMSG " + channel + " :" + short_url + "\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"))
irc.send(bytes("PRIVMSG " + channel + " :API call failed. Check console for error message..\n", "UTF-8"))
else:
print("Invalid model.")
irc.send(bytes("PRIVMSG " + channel + " :Invalid model.\n", "UTF-8"))