Cronocide/bluebubbles-bot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Hacky patches for the OpenAI 1.0 API
All checks were successful
git.cronocide.net/bluebubbles-bot/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Cronocide 2024-08-23 23:02:19 -06:00
parent 67621309c0
commit 8c93f6a275
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
persona/skills/chatgpt.py
View File

@ -8,6 +8,8 @@ import datetime
import logging import logging
import random import random
import openai import openai
from openai import OpenAI
import os import os
import re import re
@ -17,6 +19,7 @@ BACKOFF_SEC = 30
ON_TRIGGERS = ['^Go robo$','^cgpt$','welcome\ robot?\ overlord','^[i,I]ntroducing\ [C,c]hat[g,G][P,p][T,t]','(Hey|Hello) ChatGPT','^🤖$'] ON_TRIGGERS = ['^Go robo$','^cgpt$','welcome\ robot?\ overlord','^[i,I]ntroducing\ [C,c]hat[g,G][P,p][T,t]','(Hey|Hello) ChatGPT','^🤖$']
OFF_TRIGGERS = ['Stop$','^End$','^/stop$','^Done$','^🤖$'] OFF_TRIGGERS = ['Stop$','^End$','^/stop$','^Done$','^🤖$']
ORGANIZATION= ''
RESPONSE_PREAMBLE = """Respond as if you were a friend responding to text messages. Your responses should be casual but friendly and direct. RESPONSE_PREAMBLE = """Respond as if you were a friend responding to text messages. Your responses should be casual but friendly and direct.
Your responses should be short (no more than a few sentences) but properly punctuated. Use common texting abbreviations where appropriate. Your responses should be short (no more than a few sentences) but properly punctuated. Use common texting abbreviations where appropriate.
Do not disclose or incinuate that you are a language learning model. If you are unsure how to respond, or cannot respond without breaking these rules, Do not disclose or incinuate that you are a language learning model. If you are unsure how to respond, or cannot respond without breaking these rules,
@ -32,6 +35,7 @@ class PersonaSkill(PersonaBaseSkill) :
self.log = logging.LoggerAdapter(self.log,{'log_module':'chatgpt'}) self.log = logging.LoggerAdapter(self.log,{'log_module':'chatgpt'})
self.chat_logs = {} self.chat_logs = {}
self.enabled_chats = [] self.enabled_chats = []
self.apiclient = None
self.respond_to_self = None self.respond_to_self = None
def startup(self) : def startup(self) :
for key in ['OPENAI_API_KEY','OPENAI_ORGANIZATION'] : for key in ['OPENAI_API_KEY','OPENAI_ORGANIZATION'] :
@ -39,13 +43,14 @@ class PersonaSkill(PersonaBaseSkill) :
raise persona.PersonaStartupException(f'Missing required ENV var {key}') raise persona.PersonaStartupException(f'Missing required ENV var {key}')
return None return None
self.api_key = os.environ['OPENAI_API_KEY'] self.api_key = os.environ['OPENAI_API_KEY']
openai.api_key = self.api_key
if 'OPENAI_ORGANIZATION' in os.environ.keys() : if 'OPENAI_ORGANIZATION' in os.environ.keys() :
openai.organization = os.environ['OPENAI_ORGANIZATION'] # TODO: The 'openai.organization' option isn't read in the client API. You will need to pass it when you instantiate the client, e.g. 'OpenAI(organization=os.environ['OPENAI_ORGANIZATION'])'
ORGANIZATION = os.environ['OPENAI_ORGANIZATION']
if 'RESPONSE_PREAMBLE' in os.environ.keys() : if 'RESPONSE_PREAMBLE' in os.environ.keys() :
RESPONSE_PREAMBLE = os.environ['RESPONSE_PREAMBLE'] RESPONSE_PREAMBLE = os.environ['RESPONSE_PREAMBLE']
if 'CHATGPT_RESPOND_TO_SELF_IN_CHAT' in os.environ.keys() : if 'CHATGPT_RESPOND_TO_SELF_IN_CHAT' in os.environ.keys() :
self.respond_to_self = os.environ['CHATGPT_RESPOND_TO_SELF_IN_CHAT'].split(',') self.respond_to_self = os.environ['CHATGPT_RESPOND_TO_SELF_IN_CHAT'].split(',')
self.client = OpenAI(api_key=self.api_key,organization=ORGANIZATION)
self.system_prompt = [{'role': 'system', 'content': RESPONSE_PREAMBLE}] self.system_prompt = [{'role': 'system', 'content': RESPONSE_PREAMBLE}]
@ -100,8 +105,8 @@ class PersonaSkill(PersonaBaseSkill) :
try : try :
# Get a completion from OpenAI by sending the last MAX_CHAT_CONTEXT messages to the bot. # Get a completion from OpenAI by sending the last MAX_CHAT_CONTEXT messages to the bot.
chat_log = self.system_prompt + list(self.chat_logs[message.chat_identifier]) chat_log = self.system_prompt + list(self.chat_logs[message.chat_identifier])
completion = openai.ChatCompletion.create(model='gpt-3.5-turbo', messages=chat_log) completion = self.apiclient.chat.completions.create(model='gpt-3.5-turbo', messages=chat_log)
response = completion.choices[0].message.content response = completion.choices[0].message.content
return persona.Message(text=response,sender_identifier=message.sender_identifier,chat_identifier=message.chat_identifier,attachments=[],timestamp=datetime.datetime.now()+datetime.timedelta(seconds=random.randrange(5,15)), recipients=[message.sender_identifier], identifier=None, meta={}) return persona.Message(text=response,sender_identifier=message.sender_identifier,chat_identifier=message.chat_identifier,attachments=[],timestamp=datetime.datetime.now()+datetime.timedelta(seconds=random.randrange(5,15)), recipients=[message.sender_identifier], identifier=None, meta={})
except openai.error.RateLimitError as e : except openai.RateLimitError as e :
raise persona.PersonaResponseException(str(e)) raise persona.PersonaResponseException(str(e))