Compare commits
No commits in common. "10fdf5f33c5228e4b01311cc2426aab26abe6efc" and "8f0ae10fd44c52eb82041edf7cd9ed5ca82fdc82" have entirely different histories.
10fdf5f33c
...
8f0ae10fd4
3
.gitignore
vendored
3
.gitignore
vendored
@ -1,4 +1,3 @@
|
||||
.idea/
|
||||
*.env
|
||||
*.log
|
||||
*.pyc
|
||||
.idea/
|
||||
|
@ -43,6 +43,3 @@ logging.config.dictConfig({
|
||||
},
|
||||
}
|
||||
})
|
||||
|
||||
# module specific configs
|
||||
MARKOV_CONTEXT = os.environ.get('HITOMI_MARKOV_CONTEXT', 'hitomi')
|
||||
|
@ -23,6 +23,7 @@ class Dice(commands.Cog):
|
||||
logger.info("rolling dice: %s", dice)
|
||||
try:
|
||||
response = dr_botzo.post('/dice/rpc/roll/', json={'dice': dice})
|
||||
response.raise_for_status()
|
||||
logger.debug("result of rolling dice: HTTP %s, %s", response.status_code, response.text)
|
||||
output = re.sub(r'(\d+)(.*?\s+)(\(.*?\))', r'**\1**\2\3', response.json()['result'])
|
||||
output = '\n'.join(output.split('; '))
|
||||
|
@ -1,38 +0,0 @@
|
||||
"""Commands for populating and retrieving markov chains, for silly chatter."""
|
||||
import logging
|
||||
from urllib.parse import quote
|
||||
|
||||
from hitomi import bot, config
|
||||
from hitomi.backends import dr_botzo, DrBotzoError
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
@bot.listen(name='on_message')
|
||||
async def on_message_learn(message):
|
||||
"""Learn lines, potentially also respond to them."""
|
||||
logger.info("learning to %s: '%s'", config.MARKOV_CONTEXT, message.clean_content)
|
||||
try:
|
||||
response = dr_botzo.post('/markov/rpc/context/{context}/learn/'.format(context=quote(config.MARKOV_CONTEXT)),
|
||||
json={'line': message.clean_content})
|
||||
logger.debug("succeeded to learn '%s'", response.json()['line'])
|
||||
except DrBotzoError as drex:
|
||||
logger.exception("received an error from dr.botzo attempting to learn to %s: %s", config.MARKOV_CONTEXT, drex)
|
||||
|
||||
|
||||
@bot.listen(name='on_message')
|
||||
async def on_message_reply(message):
|
||||
"""Learn lines, potentially also respond to them."""
|
||||
if bot.user in message.mentions:
|
||||
logger.info("I was addressed, going to reply with chatter")
|
||||
await message.channel.trigger_typing()
|
||||
try:
|
||||
response = dr_botzo.post('/markov/rpc/context/{context}/generate/'
|
||||
''.format(context=quote(config.MARKOV_CONTEXT)),
|
||||
json={'topics': message.clean_content.split(' ')})
|
||||
reply = response.json()['generated_line']
|
||||
logger.info("replying to %s with '%s'", message.channel, reply)
|
||||
await message.channel.send(reply)
|
||||
except DrBotzoError as drex:
|
||||
logger.exception("received an error from dr.botzo attempting to reply to %s: %s",
|
||||
message.clean_content, drex)
|
Loading…
x
Reference in New Issue
Block a user