nex/college-bot-v2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Activity Actions

Fix JSON serilisation error

Browse source
This commit is contained in:
Nexus 2024-01-12 16:59:29 +00:00
parent cd1b4411d7
commit 68868a5cf2
1 changed files with 4 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/cogs/ollama.py
View file

@ -18,9 +18,6 @@ from discord.ext import commands
from conf import CONFIG from conf import CONFIG
_M = typing.TypeVar("_M", discord.Member, discord.User)
def get_time_spent(nanoseconds: int) -> str: def get_time_spent(nanoseconds: int) -> str:
hours, minutes, seconds = 0, 0, 0 hours, minutes, seconds = 0, 0, 0
seconds = nanoseconds / 1e9 seconds = nanoseconds / 1e9
@ -94,7 +91,7 @@ class ChatHistory:
""" """
key = os.urandom(3).hex() key = os.urandom(3).hex()
self._internal[key] = { self._internal[key] = {
"member": member, "member": member.id,
"seed": round(time.time()), "seed": round(time.time()),
"messages": [] "messages": []
} }
@ -124,15 +121,15 @@ class ChatHistory:
instance = cog.history instance = cog.history
return discord.utils.basic_autocomplete(list(instance.threads_for(ctx.interaction.user).keys()))(ctx) return discord.utils.basic_autocomplete(list(instance.threads_for(ctx.interaction.user).keys()))(ctx)
def all_threads(self) -> dict[str, dict[str, list[dict[str, str]] | discord.Member | int]]: def all_threads(self) -> dict[str, dict[str, list[dict[str, str]] | int]]:
"""Returns all saved threads.""" """Returns all saved threads."""
return self._internal.copy() return self._internal.copy()
def threads_for(self, user: _M) -> dict[str, list[dict[str, str]] | _M | int]: def threads_for(self, user: discord.Member) -> dict[str, list[dict[str, str]] | int]:
"""Returns all saved threads for a specific user""" """Returns all saved threads for a specific user"""
t = self.all_threads() t = self.all_threads()
for k, v in t.copy().items(): for k, v in t.copy().items():
if v["member"] != user: if v["member"] != user.id:
t.pop(k) t.pop(k)
return t return t