chatgpt-irc - chatgpt via irc
git clone https://github.com/rj1/chatgpt-irc
commit 69fc3a9e12ef3690dc2e241112f878d1cca49d56 author: rj1 <[email protected]> date: Mon, 5 Dec 2022 18:37:41 -0600 first Diffstat:
| A | LICENSE | | | 21 | +++++++++++++++++++++ |
| A | README.md | | | 1 | + |
| A | chatgpt-irc.py | | | 204 | +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ |
3 files changed, 226 insertions(+), 0 deletions(-)
diff --git a/LICENSE b/LICENSE @@ -0,0 +1,21 @@ +MIT License + +Copyright (c) 2022 rj1 <[email protected]> + +Permission is hereby granted, free of charge, to any person obtaining a copy +of this software and associated documentation files (the "Software"), to deal +in the Software without restriction, including without limitation the rights +to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +copies of the Software, and to permit persons to whom the Software is +furnished to do so, subject to the following conditions: + +The above copyright notice and this permission notice shall be included in all +copies or substantial portions of the Software. + +THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE +SOFTWARE. diff --git a/README.md b/README.md @@ -0,0 +1 @@ +# chatgpt-irc diff --git a/chatgpt-irc.py b/chatgpt-irc.py @@ -0,0 +1,204 @@ +import asyncio +from collections import namedtuple +import functools +import json +import requests +import uuid + + +class ChatGPT: + def __init__(self): + self.access_token = "" + self.conversation_id = "" + self.parent_message_id = "" + self.message_id = self.parent_message_id + + def reset(self): + self.message_id = self.parent_message_id + + def prompt(self, message): + self.parent_message_id = self.message_id + self.message_id = str(uuid.uuid4()) + + url = "https://chat.openai.com/backend-api/conversation" + + payload = json.dumps( + { + "action": "next", + "messages": [ + { + "id": self.message_id, + "role": "user", + "content": {"content_type": "text", "parts": [message]}, + } + ], + "conversation_id": self.conversation_id, + "parent_message_id": self.parent_message_id, + "model": "text-davinci-002-render", + } + ) + headers = { + "Authorization": f"Bearer {self.access_token}", + } + + query = requests.request("POST", url, headers=headers, data=payload) + response = query.text + + try: + last = response.split(("data:"))[-2] + items = json.loads(last)["message"]["content"]["parts"][0] + lines = items.split("\n") + return lines + except IndexError: + # raise IndexError() + return ["We couldn't get a response for you, please try again"] + + +# irc +Message = namedtuple("Message", "prefix command params") +Prefix = namedtuple("Prefix", "nick ident host") + + +def parse_line(line): + prefix = None + + if line.startswith(":"): + prefix, line = line.split(None, 1) + name = prefix[1:] + ident = None + host = None + if "!" in name: + name, ident = name.split("!", 1) + if "@" in ident: + ident, host = ident.split("@", 1) + elif "@" in name: + name, host = name.split("@", 1) + prefix = Prefix(name, ident, host) + + command, *line = line.split(None, 1) + command = command.upper() + + params = [] + if line: + line = line[0] + while line: + if line.startswith(":"): + params.append(line[1:]) + line = "" + else: + param, *line = line.split(None, 1) + params.append(param) + if line: + line = line[0] + + return Message(prefix, command, params) + + +def send_line_to_writer(writer: asyncio.StreamWriter, line): + print("->", line) + writer.write(line.encode("utf-8") + b"\r\n") + + +def send_cmd_to_writer(writer: asyncio.StreamWriter, cmd, *params): + params = list(params) + if params: + if " " in params[-1]: + params[-1] = ":" + params[-1] + params = [cmd] + params + send_line_to_writer(writer, " ".join(params)) + + +def send_msg(writer: asyncio.StreamWriter, target, msg): + send_cmd_to_writer(writer, "PRIVMSG", target, msg) + + +async def main_loop(host, port, **options): + reader, writer = await asyncio.open_connection( + host, port, ssl=options.get("ssl", True) + ) + + sendline = functools.partial(send_line_to_writer, writer) + sendcmd = functools.partial(send_cmd_to_writer, writer) + + sendline("NICK {nickname}".format(**options)) + sendline("USER {ident} * * :{realname}".format(**options)) + + chatgpt = ChatGPT() + + while not reader.at_eof(): + line = await reader.readline() + try: + line = line.decode("utf-8") + except UnicodeDecodeError: + line = line.decode("latin1") + + line = line.strip() + print(line) + if line: + message = parse_line(line) + if message.command.isdigit() and int(message.command) >= 400: + # error? + print(message) + + if message.command == "PING": + sendcmd("PONG", *message.params) + elif message.command == "001": + sendcmd("JOIN", options["channel"]) + + elif message.command == "PRIVMSG": + target = str(message.params[0]) # channel or nick + text = str(message.params[1]) # msg text + host = str(message.prefix.host) # user's hostname + source = str(message.prefix.nick) # nick + print(f"[{target}] <{source}> {text}") + + parts = text.split() + + if len(parts) == 0: + continue + + if parts[0] == "!reset": + sendcmd( + "PRIVMSG", options["channel"], f"{source}: Let's start fresh" + ) + chatgpt.reset() + continue + + if len(parts) <= 1: + continue + + if parts[0] == f"{options['nickname']}:": + sendcmd( + "PRIVMSG", + options["channel"], + f"{source}: hold on, I'm thinking..", + ) + prompt = parts[1:] + prompt = " ".join(prompt) + + lines = chatgpt.prompt(prompt) + messages = [] + + for line in lines: + if len(line) > 350: + messages.append(line[:350]) + messages.append(line[350:]) + else: + messages.append(line) + + for i, message in enumerate(messages): + if i == 0: + message = f"{source}: {message}" + sendcmd("PRIVMSG", options["channel"], f"{message}") + + +options = { + "nickname": "chatgpt", + "ident": "chatgpt", + "realname": "chatgpt", + "channel": "#h4x", + "ssl": True, +} + +loop = asyncio.get_event_loop() +loop.run_until_complete(main_loop("internetrelaychat.net", 6697, **options))