mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-23 11:34:18 +00:00
[0.2]
This commit is contained in:
parent
29acd1fae6
commit
a4c9af91a0
7 changed files with 113 additions and 37 deletions
|
@ -1,22 +1,28 @@
|
|||
import random
|
||||
import datetime
|
||||
import logging
|
||||
from typing import *
|
||||
from starlette.requests import Request
|
||||
from starlette.responses import *
|
||||
from royalnet.constellation import *
|
||||
from royalnet.utils import *
|
||||
from royalnet.commands import CommandInterface
|
||||
from ..tables import *
|
||||
from ..utils import *
|
||||
|
||||
|
||||
log = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class ApiKei(PageStar):
|
||||
path = "/api/kei"
|
||||
|
||||
methods = ["POST"]
|
||||
|
||||
def __init__(self, config: Dict[str, Any], constellation: "Constellation"):
|
||||
super().__init__(config, constellation)
|
||||
def __init__(self, interface: CommandInterface):
|
||||
super().__init__(interface)
|
||||
self._conversations: Dict[str, Conversation] = {}
|
||||
log.debug("Kei initialized.")
|
||||
|
||||
async def page(self, request: Request) -> JSONResponse:
|
||||
async with self.session_acm() as session:
|
||||
|
@ -38,18 +44,34 @@ class ApiKei(PageStar):
|
|||
while True:
|
||||
if convid not in self._conversations:
|
||||
# Create a new conversation
|
||||
self._conversations[convid] = await ExampleConversation.create()
|
||||
self._conversations[convid] = await ExampleConversation.create(self.interface)
|
||||
log.info(f"[{convid}] SYSTEM: New conversation created - {self._conversations[convid]}")
|
||||
conv: Conversation = self._conversations[convid]
|
||||
|
||||
try:
|
||||
emotion, text = await conv.next(session=session, person=person, message=message)
|
||||
log.info(f"[{convid}] {person}: '{message}'")
|
||||
except Exception:
|
||||
pass
|
||||
try:
|
||||
result = await conv.next(session=session,
|
||||
person=person,
|
||||
message=message)
|
||||
except StopAsyncIteration:
|
||||
del self._conversations[convid]
|
||||
continue
|
||||
except Exception as e:
|
||||
print(e)
|
||||
log.error(f"[{convid}] ERROR: {e}")
|
||||
emotion, text = Emotion.NEUTRAL, "...?"
|
||||
else:
|
||||
del self._conversations[convid]
|
||||
break
|
||||
else:
|
||||
if isinstance(result, Conversation):
|
||||
self._conversations[convid] = result
|
||||
log.info(f"[{convid}] SYSTEM: Switched conversation - {self._conversations[convid]}")
|
||||
else:
|
||||
emotion, text = result
|
||||
break
|
||||
log.info(f"[{convid}] Kei ({emotion.value}): '{text}'")
|
||||
return JSONResponse({
|
||||
"emotion": str(emotion),
|
||||
"text": text,
|
||||
|
|
|
@ -20,3 +20,6 @@ class KeiPerson:
|
|||
|
||||
def __repr__(self):
|
||||
return f"<{self.__class__.__qualname__} {self.kpid}{' ' + self.user.username if self.user is not None else ''}>"
|
||||
|
||||
def __str__(self):
|
||||
return self.user.username if self.user is not None else self.kpid
|
||||
|
|
|
@ -1,28 +1,43 @@
|
|||
from royalnet.commands import CommandInterface
|
||||
from .emotion import Emotion
|
||||
|
||||
|
||||
class Conversation:
|
||||
def __init__(self):
|
||||
def __init__(self, interface: CommandInterface):
|
||||
self.generator = self._generator()
|
||||
self.interface: CommandInterface = interface
|
||||
|
||||
self._person = None
|
||||
self._session = None
|
||||
self._message = None
|
||||
|
||||
async def _generator(self):
|
||||
yield
|
||||
raise NotImplementedError()
|
||||
|
||||
@classmethod
|
||||
async def create(cls):
|
||||
conv = cls()
|
||||
async def create(cls, interface: CommandInterface):
|
||||
conv = cls(interface=interface)
|
||||
await conv.generator.asend(None)
|
||||
return conv
|
||||
|
||||
async def next(self, session, person, message):
|
||||
reply = await self.generator.asend((session, person, message))
|
||||
self._session = session
|
||||
self._person = person
|
||||
self._message = message
|
||||
reply = await self.generator.asend(None)
|
||||
return reply
|
||||
|
||||
|
||||
# noinspection PyTupleAssignmentBalance
|
||||
class ExampleConversation(Conversation):
|
||||
async def _generator(self):
|
||||
session, person, message = yield
|
||||
session, person, message = yield Emotion.HAPPY, "Ciao!"
|
||||
session, person, message = yield Emotion.NEUTRAL, "Questa è una conversazione di prova."
|
||||
session, person, message = yield Emotion.X, "X_X"
|
||||
yield
|
||||
yield Emotion.HAPPY, "Ciao!"
|
||||
|
||||
response = await self.interface.call_herald_event("discord", "discord_cv")
|
||||
yield Emotion.SURPRISED, f"Ci sono {len(response['guild']['members'])} persone in RYG."
|
||||
|
||||
yield Emotion.NEUTRAL, "Questa è una conversazione di prova."
|
||||
yield await ExampleConversation.create(self.interface)
|
||||
yield Emotion.WORRIED, "Questo non dovrebbe mai venire fuori."
|
||||
|
|
|
@ -1 +1 @@
|
|||
semantic = "23.12"
|
||||
semantic = "0.2"
|
||||
|
|
65
poetry.lock
generated
65
poetry.lock
generated
|
@ -291,7 +291,7 @@ description = "Alternative regular expression module, to replace re."
|
|||
name = "regex"
|
||||
optional = false
|
||||
python-versions = "*"
|
||||
version = "2019.11.1"
|
||||
version = "2019.12.9"
|
||||
|
||||
[[package]]
|
||||
category = "main"
|
||||
|
@ -331,7 +331,7 @@ description = "A multipurpose bot and web framework"
|
|||
name = "royalnet"
|
||||
optional = false
|
||||
python-versions = ">=3.8,<4.0"
|
||||
version = "5.1.4"
|
||||
version = "5.1.5"
|
||||
|
||||
[package.dependencies]
|
||||
dateparser = ">=0.7.2,<0.8.0"
|
||||
|
@ -400,6 +400,30 @@ herald = ["websockets (>=8.1,<9.0)"]
|
|||
sentry = ["sentry_sdk (>=0.13.2,<0.14.0)"]
|
||||
telegram = ["python_telegram_bot (>=12.2.0,<13.0.0)"]
|
||||
|
||||
[[package]]
|
||||
category = "main"
|
||||
description = "A Royalnet command pack for the Royal Games community"
|
||||
name = "royalpack"
|
||||
optional = false
|
||||
python-versions = ">=3.8,<4.0"
|
||||
version = "5.1.6"
|
||||
|
||||
[package.dependencies]
|
||||
riotwatcher = ">=2.7.1,<3.0.0"
|
||||
royalspells = ">=3.2,<4.0"
|
||||
|
||||
[package.dependencies.royalnet]
|
||||
extras = ["sentry", "coloredlogs", "alchemy_easy", "discord", "constellation", "herald", "telegram", "bard"]
|
||||
version = ">=5.1.4,<6.0.0"
|
||||
|
||||
[[package]]
|
||||
category = "main"
|
||||
description = "A package to procedurally generate useless spells!"
|
||||
name = "royalspells"
|
||||
optional = false
|
||||
python-versions = "~=3.6"
|
||||
version = "3.2"
|
||||
|
||||
[[package]]
|
||||
category = "main"
|
||||
description = "Python client for Sentry (https://getsentry.com)"
|
||||
|
@ -574,7 +598,7 @@ python-versions = "*"
|
|||
version = "2019.11.28"
|
||||
|
||||
[metadata]
|
||||
content-hash = "7f30fe6e329457ef7d6a8b6dea1067131a22df3b35197fe0834d27ef3b5535d5"
|
||||
content-hash = "7b99fc704733ab0bc30b37829a5124f6d04a37e1d945b93abe0c231cae553a30"
|
||||
python-versions = "^3.8"
|
||||
|
||||
[metadata.files]
|
||||
|
@ -814,19 +838,17 @@ pytz = [
|
|||
{file = "pytz-2019.3.tar.gz", hash = "sha256:b02c06db6cf09c12dd25137e563b31700d3b80fcc4ad23abb7a315f2789819be"},
|
||||
]
|
||||
regex = [
|
||||
{file = "regex-2019.11.1-cp27-none-win32.whl", hash = "sha256:604dc563a02a74d70ae1f55208ddc9bfb6d9f470f6d1a5054c4bd5ae58744ab1"},
|
||||
{file = "regex-2019.11.1-cp27-none-win_amd64.whl", hash = "sha256:5e00f65cc507d13ab4dfa92c1232d004fa202c1d43a32a13940ab8a5afe2fb96"},
|
||||
{file = "regex-2019.11.1-cp35-none-win32.whl", hash = "sha256:15454b37c5a278f46f7aa2d9339bda450c300617ca2fca6558d05d870245edc7"},
|
||||
{file = "regex-2019.11.1-cp35-none-win_amd64.whl", hash = "sha256:d2b302f8cdd82c8f48e9de749d1d17f85ce9a0f082880b9a4859f66b07037dc6"},
|
||||
{file = "regex-2019.11.1-cp36-cp36m-manylinux1_i686.whl", hash = "sha256:b4e0406d822aa4993ac45072a584d57aa4931cf8288b5455bbf30c1d59dbad59"},
|
||||
{file = "regex-2019.11.1-cp36-cp36m-manylinux1_x86_64.whl", hash = "sha256:7faf534c1841c09d8fefa60ccde7b9903c9b528853ecf41628689793290ca143"},
|
||||
{file = "regex-2019.11.1-cp36-none-win32.whl", hash = "sha256:7caf47e4a9ac6ef08cabd3442cc4ca3386db141fb3c8b2a7e202d0470028e910"},
|
||||
{file = "regex-2019.11.1-cp36-none-win_amd64.whl", hash = "sha256:e3d8dd0ec0ea280cf89026b0898971f5750a7bd92cb62c51af5a52abd020054a"},
|
||||
{file = "regex-2019.11.1-cp37-none-win32.whl", hash = "sha256:c31eaf28c6fe75ea329add0022efeed249e37861c19681960f99bbc7db981fb2"},
|
||||
{file = "regex-2019.11.1-cp37-none-win_amd64.whl", hash = "sha256:1ad40708c255943a227e778b022c6497c129ad614bb7a2a2f916e12e8a359ee7"},
|
||||
{file = "regex-2019.11.1-cp38-none-win32.whl", hash = "sha256:ec032cbfed59bd5a4b8eab943c310acfaaa81394e14f44454ad5c9eba4f24a74"},
|
||||
{file = "regex-2019.11.1-cp38-none-win_amd64.whl", hash = "sha256:c7393597191fc2043c744db021643549061e12abe0b3ff5c429d806de7b93b66"},
|
||||
{file = "regex-2019.11.1.tar.gz", hash = "sha256:720e34a539a76a1fedcebe4397290604cc2bdf6f81eca44adb9fb2ea071c0c69"},
|
||||
{file = "regex-2019.12.9-cp27-none-win32.whl", hash = "sha256:40b7d1291a56897927e08bb973f8c186c2feb14c7f708bfe7aaee09483e85a20"},
|
||||
{file = "regex-2019.12.9-cp27-none-win_amd64.whl", hash = "sha256:c203c9ee755e9656d0af8fab82754d5a664ebaf707b3f883c7eff6a3dd5151cf"},
|
||||
{file = "regex-2019.12.9-cp35-none-win32.whl", hash = "sha256:719978a9145d59fc78509ea1d1bb74243f93583ef2a34dcc5623cf8118ae9726"},
|
||||
{file = "regex-2019.12.9-cp35-none-win_amd64.whl", hash = "sha256:75cf3796f89f75f83207a5c6a6e14eaf57e0369ef0ffff8e22bf36bbcfa0f1de"},
|
||||
{file = "regex-2019.12.9-cp36-none-win32.whl", hash = "sha256:3dbd8333fd2ebd50977ac8747385a73aa1f546eb6b16fcd83d274470fe11f243"},
|
||||
{file = "regex-2019.12.9-cp36-none-win_amd64.whl", hash = "sha256:ad9e3c7260809c0d1ded100269f78ea0217c0704f1eaaf40a382008461848b45"},
|
||||
{file = "regex-2019.12.9-cp37-none-win32.whl", hash = "sha256:91235c98283d2bddf1a588f0fbc2da8afa37959294bbd18b76297bdf316ba4d6"},
|
||||
{file = "regex-2019.12.9-cp37-none-win_amd64.whl", hash = "sha256:aaffd68c4c1ed891366d5c390081f4bf6337595e76a157baf453603d8e53fbcb"},
|
||||
{file = "regex-2019.12.9-cp38-none-win32.whl", hash = "sha256:e865bc508e316a3a09d36c8621596e6599a203bc54f1cd41020a127ccdac468a"},
|
||||
{file = "regex-2019.12.9-cp38-none-win_amd64.whl", hash = "sha256:77396cf80be8b2a35db863cca4c1a902d88ceeb183adab328b81184e71a5eafe"},
|
||||
{file = "regex-2019.12.9.tar.gz", hash = "sha256:77a3799152951d6d14ae5720ca162c97c64f85d4755da585418eac216b736cad"},
|
||||
]
|
||||
requests = [
|
||||
{file = "requests-2.22.0-py2.py3-none-any.whl", hash = "sha256:9cf5292fcd0f598c671cfc1e0d7d1a7f13bb8085e9a590f48c010551dc6c4b31"},
|
||||
|
@ -837,8 +859,15 @@ riotwatcher = [
|
|||
{file = "riotwatcher-2.7.1.tar.gz", hash = "sha256:5349271c7e00637b7619491a6070e66603705db60558ea2a690e7016f6e6d9a4"},
|
||||
]
|
||||
royalnet = [
|
||||
{file = "royalnet-5.1.4-py3-none-any.whl", hash = "sha256:6d2a9d244b9ac3f907a7f4b63e2aa192c560bb63f9abb1426647ddbabb2498a0"},
|
||||
{file = "royalnet-5.1.4.tar.gz", hash = "sha256:439f80b1efc12b6d7c38286d544c04aefb0d43b05a8fe7c7e17c84f7237382f4"},
|
||||
{file = "royalnet-5.1.5-py3-none-any.whl", hash = "sha256:2673c195566b6f65c45cd9feb3a6512c8de2cd07557f1434e9345cd345cc361c"},
|
||||
{file = "royalnet-5.1.5.tar.gz", hash = "sha256:d25603b39065a1af7e5e4e05ffccb85eb1fbd1bc3f8ab3873d723dd9ac51fe1f"},
|
||||
]
|
||||
royalpack = [
|
||||
{file = "royalpack-5.1.6-py3-none-any.whl", hash = "sha256:322aed877a634c60b7216b438ecd3bf0057a8a9b5d113ae390e975e1b6f46653"},
|
||||
{file = "royalpack-5.1.6.tar.gz", hash = "sha256:489e19c6ae66f25197abaaac2654d79a9c6496799b1c9aa741140a6517130c54"},
|
||||
]
|
||||
royalspells = [
|
||||
{file = "royalspells-3.2.tar.gz", hash = "sha256:2bd4a9a66514532e35c02c3907425af48c7cb292364c4843c795719a82b25dfe"},
|
||||
]
|
||||
sentry-sdk = [
|
||||
{file = "sentry-sdk-0.13.5.tar.gz", hash = "sha256:c6b919623e488134a728f16326c6f0bcdab7e3f59e7f4c472a90eea4d6d8fe82"},
|
||||
|
|
|
@ -2,7 +2,7 @@
|
|||
|
||||
[tool.poetry]
|
||||
name = "keipack"
|
||||
version = "23.12"
|
||||
version = "0.2"
|
||||
description = "A mysterious AI assistant"
|
||||
authors = ["Stefano Pigozzi <ste.pigozzi@gmail.com>"]
|
||||
license = "AGPL-3.0+"
|
||||
|
@ -19,10 +19,10 @@
|
|||
[tool.poetry.dependencies]
|
||||
python = "^3.8"
|
||||
|
||||
riotwatcher = "^2.7.1"
|
||||
royalpack = "^5.1.6"
|
||||
|
||||
[tool.poetry.dependencies.royalnet]
|
||||
version = "^5.1.4"
|
||||
version = "^5.1.5"
|
||||
# Maybe... there is a way to make these selectable?
|
||||
extras = [
|
||||
"telegram",
|
||||
|
|
|
@ -6,10 +6,17 @@
|
|||
# * 'ptr' and 'ptrp' create a POST request with a simple or parameter-like body;
|
||||
# * 'mptr' and 'fptr' create a POST request to submit a form with a text or file field (multipart/form-data);
|
||||
|
||||
### Is the server up?
|
||||
|
||||
GET http://localhost:44445/api/royalnet/version
|
||||
|
||||
### Basic conversation
|
||||
|
||||
POST http://localhost:44445/api/kei
|
||||
Accept: application/json
|
||||
Content-Type: application/x-www-form-urlencoded
|
||||
|
||||
kpid=test&convid=1&first=true&message=lul
|
||||
|
||||
###
|
||||
###
|
||||
|
||||
|
|
Loading…
Reference in a new issue