This commit is contained in:
codey 2024-11-17 13:41:20 -05:00
parent b33470a412
commit 4e782de528
6 changed files with 198 additions and 10 deletions

1
cah/__init__.py Normal file
View File

@ -0,0 +1 @@
#!/usr/bin/env python3.12

3
cah/__main__.py Normal file
View File

@ -0,0 +1,3 @@
#!/usr/bin/env python3.12
import constructors

View File

@ -169,13 +169,23 @@ class AI(FastAPI):
# } # }
# ai_req_data = { # ai_req_data = {
# 'max_context_length': 8192, # 'max_context_length': 8192,
# 'max_length': 256, # 'max_length': 512,
# 'temperature': 0.3, # 'temperature': 0,
# 'n': 1,
# 'top_k': 30,
# 'top_a': 0,
# 'top_p': 0,
# 'typical': 0,
# 'mirostat': 0,
# 'use_default_badwordsids': False,
# 'rep_pen': 1.0,
# 'rep_pen_range': 320,
# 'rep_pen_slope': 0.05,
# 'quiet': 1, # 'quiet': 1,
# 'bypass_eos': False, # 'bypass_eos': False,
# 'trim_stop': True, # # 'trim_stop': True,
# 'sampler_order': [6,0,1,3,4,2,5], # 'sampler_order': [6,0,1,3,4,2,5],
# 'memory': "You are a helpful assistant who will provide only totally accurate tidbits of info on songs the user may listen to. You do not include information about which album a song was released on, or when it was released, and do not mention that you are not including this information in your response. If the input provided is not a song you are aware of, simply state that. Begin your output at your own response.", # 'memory': "You are a helpful assistant who will provide ONLY TOTALLY ACCURATE tidbits of info on songs the user may listen to. You do not include information about which album a song was released on, or when it was released, and do not mention that you are not including this information in your response. If the input provided is not a song you are aware of, simply state that. Begin your output at your own response.",
# 'stop': ['### Inst', '### Resp'], # 'stop': ['### Inst', '### Resp'],
# 'prompt': ai_question # 'prompt': ai_question
# } # }

View File

@ -41,7 +41,7 @@ class KarmaDB:
async def get_karma(self, keyword: str) -> int | dict: async def get_karma(self, keyword: str) -> int | dict:
async with sqlite3.connect(self.db_path, timeout=2) as db_conn: async with sqlite3.connect(self.db_path, timeout=2) as db_conn:
async with db_conn.execute("SELECT score FROM karma WHERE keyword = ? LIMIT 1", (keyword,)) as db_cursor: async with db_conn.execute("SELECT score FROM karma WHERE keyword LIKE ? LIMIT 1", (keyword,)) as db_cursor:
try: try:
(score,) = await db_cursor.fetchone() (score,) = await db_cursor.fetchone()
return score return score
@ -65,7 +65,7 @@ class KarmaDB:
return return
modifier = "score + 1" if not flag else "score - 1" modifier = "score + 1" if not flag else "score - 1"
query = f"UPDATE karma SET score = {modifier}, last_change = ? WHERE keyword = ?" query = f"UPDATE karma SET score = {modifier}, last_change = ? WHERE keyword LIKE ?"
new_keyword_query = "INSERT INTO karma(keyword, score, last_change) VALUES(?, ?, ?)" new_keyword_query = "INSERT INTO karma(keyword, score, last_change) VALUES(?, ?, ?)"
friendly_flag = "++" if not flag else "--" friendly_flag = "++" if not flag else "--"
audit_message = f"{granter} adjusted karma for {keyword} @ {datetime.datetime.now().isoformat()}: {friendly_flag}" audit_message = f"{granter} adjusted karma for {keyword} @ {datetime.datetime.now().isoformat()}: {friendly_flag}"
@ -119,7 +119,7 @@ class Karma(FastAPI):
""" """
/karma/top/ /karma/top/
Get top keywords for karma Get top keywords for karma
Requires key (Requires key)
""" """
if not self.util.check_key(request.url.path, request.headers.get('X-Authd-With')): if not self.util.check_key(request.url.path, request.headers.get('X-Authd-With')):
@ -140,13 +140,16 @@ class Karma(FastAPI):
'errorText': 'Exception occurred.', 'errorText': 'Exception occurred.',
} }
async def get_karma_handler(self, data: ValidKarmaRetrievalRequest): async def get_karma_handler(self, data: ValidKarmaRetrievalRequest, request: Request):
""" """
/karma/get/ /karma/get/
Get current karma value Get current karma value
Requires key (Requires key)
""" """
if not self.util.check_key(request.url.path, request.headers.get('X-Authd-With')):
raise HTTPException(status_code=403, detail="Unauthorized")
keyword = data.keyword keyword = data.keyword
try: try:
count = await self.db.get_karma(keyword) count = await self.db.get_karma(keyword)
@ -165,7 +168,7 @@ class Karma(FastAPI):
""" """
/karma/update/ /karma/update/
Update karma count Update karma count
Requires key (Requires key)
""" """
if not self.util.check_key(request.url.path, request.headers.get('X-Authd-With'), 2): if not self.util.check_key(request.url.path, request.headers.get('X-Authd-With'), 2):

170
tmp/Lyrics/plugin.py Normal file
View File

@ -0,0 +1,170 @@
###
# Copyright (c) 2024, codey
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice,
# this list of conditions, and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright notice,
# this list of conditions, and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
# * Neither the name of the author of this software nor the name of
# contributors to this software may be used to endorse or promote products
# derived from this software without specific prior written consent.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
###
import textwrap
import time
import re
import requests
from supybot import utils, plugins, ircutils, callbacks, conf, ircdb
from supybot.commands import *
from supybot.i18n import PluginInternationalization
from supybot.i18n import internationalizeDocstring
_ = PluginInternationalization('Lyrics')
class SearchException(Exception):
pass
class Lyrics(callbacks.Plugin):
"""Song Lyrics Search"""
threaded = True
@wrap(['text'])
def sing(self, irc, msg, args, query):
"""<query>
Sing <artist> : <song>
"""
"""Check if Lyrics Search is enabled in this context/channel"""
if not(self.registryValue('enable', network=irc.network, channel=msg.channel)):
return irc.noReply()
"""Check for botwide ignores"""
if ircdb.checkIgnored(msg.prefix, msg.args[0]):
return irc.noReply()
use_mores = self.registryValue('mores', network=irc.network, channel=msg.channel)
truncate = self.registryValue('truncate', network=irc.network, channel=msg.channel)
lines_to_send = self.registryValue('lines', network=irc.network, channel=msg.channel)
quiet = self.registryValue('quiet', network=irc.network, channel=msg.channel)
verbose = self.registryValue('verbose', network=irc.network, channel=msg.channel)
can_flood = self.registryValue('floodexempt')
subsearch = None
trailer = " [...]"
query = "".join(query.strip())
colons = len(re.findall(r':', query))
dashes = len(re.findall(r'\s-\s', query))
has_subsearch = (colons > 1)
"""Check for valid query ( must contain : or - )"""
if not(colons) and not(dashes):
irc.reply("\002\00304Invalid query! " \
"Query format should be either <artist> : <song>, <artist> - <song>, or optionally " \
"<artist> : <song> : <subsearch>", prefixNick=False)
return
if colons:
artist = query.split(":")[0].strip()
song = query.split(":")[1].strip()
else:
artist = query.split(" - ", maxsplit=1)[0].strip()
song = query.split(" - ", maxsplit=1)[1].strip()
searchObj = {
'a': artist,
's': song,
'src': 'IRC-SHARED',
'extras': True
}
headers = {
'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64; rv:127.0) Gecko/20100101 Firefox/127.0',
'Content-Type': 'application/json; charset=utf-8'
}
if has_subsearch:
subsearch = query.split(":")[2].strip()
searchObj['sub'] = subsearch
try:
if not(quiet):
irc.reply("\002\00307Searching...",
private=not(verbose), notice=not(verbose), prefixNick=False)
request = requests.post(self.registryValue('host'),
json=searchObj,
timeout=(10, 20), verify=False, headers=headers)
request.raise_for_status()
response = request.json()
if response.get('err'):
raise SearchException(response.get('errorText'))
returned_artist = response.get('artist')
returned_song = response.get('song')
if not(quiet):
irc.reply("\002\00309Found: %s - %s"
% (returned_artist, returned_song),
notice=not(verbose), private=not(verbose),
prefixNick=False)
lyrics = response.get('lyrics')
lyrics = re.sub(r'<br>', ' / ', lyrics)
if truncate and use_mores:
irc.reply(lyrics, prefixNick=False)
return
wrapped_lyrics = textwrap.wrap(text=lyrics,
width=360,
break_on_hyphens=True,
break_long_words=True)
if not(truncate):
lines_to_send = len(wrapped_lyrics)
for idx, line in enumerate(wrapped_lyrics):
line = re.sub(r'/(\s{0,})$', '', line).strip()
if idx >= lines_to_send:
break
truncated = (idx+1 == lines_to_send and truncate)
irc.reply(line.strip() + (trailer if truncated else ''), prefixNick=False)
if idx >= 5 and not(can_flood):
time.sleep(0.5) # Avoid RecvQ by sleeping for 0.5s starting with the 5th line of output
return
except Exception as e:
irc.reply(f"\002\00304Error: {str(e)}", prefixNick=False)
return
Class = Lyrics

1
tmp/README.md Normal file
View File

@ -0,0 +1 @@
# This stuff does not belong here and will be removed