Laurent/nio-llm
1
0
Fork 0
mirror of https://github.com/Laurent2916/nio-llm.git synced 2024-11-27 00:28:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

✨ setup rich logging handler

Browse source
This commit is contained in:
Laureηt 2023-05-24 21:46:42 +02:00
parent 2893239bb3
commit aa6e8e79c2
Signed by: Laurent
SSH key fingerprint: SHA256:kZEpW8cMJ54PDeCvOhzreNr4FSh6R13CMGH/POoO8DI
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
nio-llm/client.py
View file

@ -9,6 +9,7 @@ import click
from huggingface_hub import hf_hub_download from huggingface_hub import hf_hub_download
from llama_cpp import Llama from llama_cpp import Llama
from nio import AsyncClient, MatrixRoom, RoomMessageText from nio import AsyncClient, MatrixRoom, RoomMessageText
from rich.logging import RichHandler
logger = logging.getLogger("nio-llm") logger = logging.getLogger("nio-llm")
@ -188,10 +189,10 @@ async def _main(
) )
# Login to the homeserver # Login to the homeserver
print(await client.login(password)) logger.debug(await client.login(password))
# Join the room, if not already joined # Join the room, if not already joined
print(await client.join(room)) logger.debug(await client.join(room))
# Sync with the server forever # Sync with the server forever
await client.sync_forever(timeout=30000) await client.sync_forever(timeout=30000)
@ -199,7 +200,12 @@ async def _main(
if __name__ == "__main__": if __name__ == "__main__":
# set up logging # set up logging
logging.basicConfig(level=logging.DEBUG) logging.captureWarnings(True)
logging.basicConfig(
level="DEBUG",
format="%(name)s: %(message)s",
handlers=[RichHandler(markup=True)],
)
# run the main program (with environment variables) # run the main program (with environment variables)
main(auto_envvar_prefix="NIOLLM") main(auto_envvar_prefix="NIOLLM")