PsychoticNinja/oddluck-limnoria-plugins
1
0
Fork 0
mirror of https://github.com/oddluck/limnoria-plugins.git synced 2025-04-25 20:41:21 -05:00
Code Issues Projects Releases Wiki Activity

Update plugin.py

Browse Source 
Simplify/modernize/fix?
This commit is contained in:
Gordon Shumway 2024-05-29 04:39:14 -04:00 committed by GitHub
parent 5d1ca4bf86
commit 73dffac290
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ChatGPT/plugin.py
View File

@ -31,7 +31,7 @@
from supybot import utils, plugins, ircutils, callbacks from supybot import utils, plugins, ircutils, callbacks
from supybot.commands import * from supybot.commands import *
from supybot.i18n import PluginInternationalization from supybot.i18n import PluginInternationalization
from openai import OpenAI import openai
_ = PluginInternationalization("ChatGPT") _ = PluginInternationalization("ChatGPT")
@ -44,9 +44,9 @@ class ChatGPT(callbacks.Plugin):
def chat(self, irc, msg, args, text): def chat(self, irc, msg, args, text):
"""Manual Call to the ChatGPT API""" """Manual Call to the ChatGPT API"""
client = OpenAI(api_key=self.registryValue("api_key")) openai.api_key = self.registryValue("api_key")
prompt = self.registryValue("prompt", msg.channel).replace("$botnick", irc.nick) prompt = self.registryValue("prompt", msg.channel).replace("$botnick", irc.nick)
completion = client.chat.completions.create( completion = openai.chat.completions.create(
model=self.registryValue("model", msg.channel), model=self.registryValue("model", msg.channel),
messages=[ messages=[
{"role": "system", "content": prompt}, {"role": "system", "content": prompt},
@ -70,8 +70,6 @@ class ChatGPT(callbacks.Plugin):
chat = wrap(chat, ["text"]) chat = wrap(chat, ["text"])
Class = ChatGPT Class = ChatGPT
# vim:set shiftwidth=4 softtabstop=4 expandtab textwidth=79: # vim:set shiftwidth=4 softtabstop=4 expandtab textwidth=79: