Compare commits
3 Commits
40c1f9dcb4
...
d1b6085ee8
| Author | SHA1 | Date | |
|---|---|---|---|
| d1b6085ee8 | |||
| 78e9cc11a1 | |||
| 18137c054b |
@ -25,7 +25,6 @@ CALLABLE_FUNCTIONS = {
|
||||
|
||||
DEFAULT_MODEL = "gpt-5-mini-2025-08-07"
|
||||
DEFAULT_MAX_COMPLETION_TOKENS = 2000
|
||||
TYPING_HEARTBEAT_INTERVAL_SECONDS = 8
|
||||
STATUS_UPDATE_MIN_INTERVAL_SECONDS = 1.5
|
||||
|
||||
|
||||
@ -100,12 +99,6 @@ class ReginaldCog(PermissionsMixin, BlacklistMixin, MemoryMixin, commands.Cog):
|
||||
}
|
||||
return tool_statuses.get(tool_name, "Reginald is consulting an external source...")
|
||||
|
||||
async def typing_heartbeat(self, channel: discord.TextChannel):
|
||||
while True:
|
||||
with suppress(discord.HTTPException):
|
||||
await channel.trigger_typing()
|
||||
await asyncio.sleep(TYPING_HEARTBEAT_INTERVAL_SECONDS)
|
||||
|
||||
def make_status_updater(
|
||||
self, status_message: discord.Message
|
||||
) -> Callable[[str, bool], Awaitable[None]]:
|
||||
@ -215,18 +208,37 @@ class ReginaldCog(PermissionsMixin, BlacklistMixin, MemoryMixin, commands.Cog):
|
||||
status_message = await message.channel.send(self.get_thinking_status_message())
|
||||
status_update = self.make_status_updater(status_message)
|
||||
|
||||
typing_task = asyncio.create_task(self.typing_heartbeat(message.channel))
|
||||
response_text = None
|
||||
if hasattr(message.channel, "typing"):
|
||||
try:
|
||||
async with message.channel.typing():
|
||||
response_text = await self.generate_response(
|
||||
api_key,
|
||||
formatted_messages,
|
||||
status_update=status_update,
|
||||
)
|
||||
except (discord.HTTPException, AttributeError):
|
||||
# Fall back to normal processing if typing indicator isn't available.
|
||||
response_text = await self.generate_response(
|
||||
api_key,
|
||||
formatted_messages,
|
||||
status_update=status_update,
|
||||
)
|
||||
else:
|
||||
response_text = await self.generate_response(
|
||||
api_key,
|
||||
formatted_messages,
|
||||
status_update=status_update,
|
||||
)
|
||||
finally:
|
||||
typing_task.cancel()
|
||||
with suppress(asyncio.CancelledError):
|
||||
await typing_task
|
||||
|
||||
try:
|
||||
await self.send_split_message(message.channel, response_text)
|
||||
finally:
|
||||
if status_message is not None:
|
||||
with suppress(discord.HTTPException):
|
||||
await status_message.delete()
|
||||
|
||||
try:
|
||||
memory.append({"user": user_name, "content": prompt})
|
||||
memory.append({"user": "Reginald", "content": response_text})
|
||||
|
||||
@ -250,11 +262,8 @@ class ReginaldCog(PermissionsMixin, BlacklistMixin, MemoryMixin, commands.Cog):
|
||||
).mid_term_memory() as mid_memory:
|
||||
short_memory[channel_id] = memory
|
||||
mid_memory[channel_id] = mid_term_summaries[-self.summary_retention_limit :]
|
||||
|
||||
await self.send_split_message(message.channel, response_text)
|
||||
if status_message is not None:
|
||||
with suppress(discord.HTTPException):
|
||||
await status_message.delete()
|
||||
except Exception as error:
|
||||
print(f"DEBUG: Memory persistence failed after response delivery: {error}")
|
||||
|
||||
def should_reginald_interject(self, message_content: str) -> bool:
|
||||
direct_invocation = {"reginald,"}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user