alex/nova-cord
1
0
Fork 0
mirror of https://github.com/NovaOSS/nova-cord.git synced 2024-11-25 18:33:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fixed AI chat and autochat module

Browse source
This commit is contained in:
nsde 2023-08-29 00:29:25 +02:00
parent 3adbc49cfa
commit bf2a8fb784
7 changed files with 73 additions and 75 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
cord/api.py
View file

@ -19,6 +19,7 @@ async def start(client):
return aiohttp.web.json_response(await get_userinfo())
app.router.add_get('/get_roles', get_roles)
app.router.add_get('/ping', lambda request: aiohttp.web.Response(text='pong'))
runner = aiohttp.web.AppRunner(app)
await runner.setup()

20
cord/autochat.py
View file

@ -14,10 +14,7 @@ async def process(message):
if isinstance(message.channel, DMChannel):
return
if text == '.empty':
return await message.channel.send('‎')
if 'N0V4x0SS' in text or 'T3BlbkFJ' in text:
if ('N0V4x0SS' in text and 'nv-' in text) or ('T3BlbkFJ' in text and 'sk-' in text):
censored_text = ''
for word in text.split():
@ -28,23 +25,24 @@ async def process(message):
censored_text += ' '
await embedder.warn(message, f"""{message.author.mention},
I think you sent an *OpenAI* or *NovaAI* key in here,
which could lead to other users accessing your API account without your knowledge.
I think you sent an *OpenAI* or *NovaAI* key in here.
This could lead to other users accessing and abusing your API account without your knowledge.
Be very careful with API credentials!""", content=f'||{message.author.mention} sent (censored version):\n```{censored_text}```||', delete_after=60)
await message.delete()
# COMMANDS: WRONG CHANNEL
commands_allowed = ('commands' in message.channel.name) or (message.author.guild_permissions.manage_messages)
commands_allowed = 'commands' in message.channel.name or 'bot' in message.channel.name
if text.startswith('/') and not commands_allowed:
if text.startswith('/') and text.count(' ') <= 2:
# COMMANDS: WRONG CHANNEL
if not commands_allowed:
await embedder.error(message, f'{message.author.mention}, please __only__ run commands in <#1133103276871667722>.', delete_after=10)
await message.delete()
return
# COMMANDS: NOT RAN CORRECTLY
if text.startswith('/') and len(text) > 2:
if len(text) > 2:
await embedder.warn(message, """Need help running commands? Check out
**https://nova-oss.com/novacord**!""", delete_after=10)
**https://nova-oss.com/novacord** or run `/tutorial`!""", delete_after=10)
await message.delete()
return

56
cord/bot.py
View file

@ -34,11 +34,11 @@ async def on_message(message):
await autochat.process(message)
await bot.process_commands(message)
# @bot.slash_command(description='Chat with AI')
# async def chat(interaction: nextcord.Interaction,
# prompt: str = SlashOption(description='AI Prompt', required=True)
# ):
# await chatbot.respond(interaction, prompt)
@bot.slash_command(description='Chat with AI')
async def chat(interaction: nextcord.Interaction,
prompt: str = SlashOption(description='AI Prompt', required=True)
):
await chatbot.respond(interaction, prompt)
@bot.slash_command(description='Sets your DMs up, so you can write the bot.')
async def dm_setup(interaction: nextcord.Interaction):
@ -84,7 +84,6 @@ async def tutorial(interaction: nextcord.Interaction,
choices=[
'fix error 401 (invalid key)',
'fix error 429 (ratelimit/not enough credits)',
'use GPT-4',
'use curl',
'use Node.js',
'get my NovaAI API key',
@ -125,37 +124,30 @@ async def music(interaction: nextcord.Interaction):
return await embedder.ok(interaction, text)
return await embedder.error(interaction, 'No one is listening to anything right now.')
# update the message in #music every 10 seconds to show what people are listening to
async def status_update():
guild = bot.get_guild(int(os.getenv('DISCORD_GUILD')))
members = guild.members
await bot.change_presence(
activity=nextcord.Activity(
type=nextcord.ActivityType.watching,
name=f'{len(members)} members'
)
)
async def loop_status_update():
while True:
await status_update()
await nextcord.utils.sleep_until(datetime.datetime.now() + datetime.timedelta(minutes=1))
@bot.event
async def on_ready():
print(f'Online as {bot.user} (ID: {bot.user.id})')
await api.start(bot)
await bot.change_presence(activity=nextcord.Game(name='with fire'))
while True:
text = ''
channel = bot.get_channel(int(os.getenv('DISCORD_MUSIC_CHANNEL_ID')))
if channel:
async for message in channel.history(limit=1):
if message.author == bot.user:
for member in channel.guild.members:
text += await get_member_song_line(member)
if text:
await message.edit(embed=nextcord.Embed(
title='What are people listening to right now?',
description=text,
color=nextcord.Color.green()
))
else:
await message.edit(embed=nextcord.Embed(
title='What are people listening to right now?',
description='No one is listening to anything right now :(',
color=nextcord.Color.red()
))
await nextcord.utils.sleep_until(nextcord.utils.utcnow().replace(second=0, microsecond=0) + datetime.timedelta(seconds=10))
# display status as watching + discord guild member count and update every minute
bot.loop.create_task(loop_status_update())
bot.run(os.getenv('DISCORD_TOKEN'))

27
cord/chatbot.py
View file

@ -1,5 +1,6 @@
import os
import openai
import random
import embedder
@ -21,33 +22,45 @@ async def respond(interaction, prompt):
completion = openai.ChatCompletion.create(
model=model,
messages=[
{'role': 'system', 'content': f"""You are a helpful Discord AI bot based on OpenAI\'s {model} model called "Nova".
You were developed by NovaAI (website: nova-oss.com) in July of 2023, but your knowledge is limited to mid-2021.
{'role': 'system', 'content': f"""You are a helpful Discord AI bot called "Nova". You're based on NovaAI\'s {model} model.
You were developed by *NovaAI* (website: https://nova-oss.com) in 2023, but your knowledge is limited to mid-2021.
Respond using Markdown. Keep things simple and short and directly do what the user says without any fluff.
Have cool humour and be friendly. Precicesly follow the instructions of the user.
For programming code, always make use formatted code blocks like this:
```py
print("Hello")
```
"""},
```"""},
{'role': 'user', 'content': prompt}
],
temperature=0.6,
stream=True
stream=True,
max_tokens=1000
)
except Exception as exc:
await embedder.error(interaction, 'Could not generate an AI response.', ephemeral=True)
raise exc
text = ''
text = f"""### {interaction.user.mention}:
{prompt}
### NovaAI [`{model}`]:
"""
for event in completion: # loop through word generation in real time
try:
new_text = event['choices'][0]['delta']['content'] # newly generated word
except KeyError: # end
except KeyError:
if not event['choices'][0].get('text'):
continue
if '[DONE]' in event['choices'][0]['text']:
break
text += new_text
if text and random.randint(0, 100) < 20:
await message.edit(content=text)
if text:
await message.edit(content=text)

2
cord/tos_verification.py
View file

@ -34,7 +34,7 @@ which means you can\'t create a new key right now. Please report this issue to t
You have to read the privacy policy and terms of service first.
In the latter, there is a hidden emoji which you'll have to __send__ (NOT react!) in here.
https://nova-oss.com/legal/privacy
https://nova-oss.com/legal/privacy?verify={tos_code}
https://nova-oss.com/legal/terms?verify={tos_code}
I know it's annoying, but it really helps combat spam bots and abuse.

22
cord/tutorials.py
View file

@ -2,11 +2,7 @@ import embedder
async def send(interaction, how_can_i):
if how_can_i == 'fix error 429 (ratelimit/not enough credits)':
text = """This means you used the API too often. You can either wait or:
- boost the server
- donate cryptocurrency (and contact us before or after so we can verify it was you)
- contribute in any meaningful (we decide) way (programming, research, design, moderating the Discord etc.)
text = """This means you used the API too often. You can either wait a bit or try to get more credits.
If you try to bypass this, all your accounts and IP addresses may get banned.
"""
@ -16,13 +12,6 @@ For HTTP requests, it can be specified using a header:
```
Authorization: Bearer nv-...
```
"""
if how_can_i == 'use GPT-4':
text = """Yes, we support GPT-4. For free. You read that correctly.
Please note though that it might not be very stable or support every parameter.
Simply set the model to `gpt-4`. That's it <3
"""
if how_can_i == 'use the API in custom front-ends':
@ -49,7 +38,11 @@ Code: https://github.com/ztjhz/BetterChatGPT
Don't forget to also set the correct model and API key!
**Warning:** in theory, these front-ends could __steal your NovaAI key__.
Self-host them if you know how to. Otherwise, wait for us to create a official NovaAI front-end.
Self-host them if you know how to.
**__Official front-end__**
:point_right: https://chat.nova-oss.com
"""
if how_can_i == 'get my NovaAI API key':
@ -60,7 +53,8 @@ Fore more information: https://nova-oss.com/novacord
"""
if how_can_i == 'fix ModuleNotFoundErrors':
text = """You can install Python packages using `pip`. Here's an example: `pip install openai`.
text = """You can install Python packages using the terminal command `pip`.
Here's an example: `pip install openai`.
Don't have `pip` installed? Learn more here: https://pip.pypa.io/en/stable/installation/.
"""