Kanium-PUBLIC/KaniumCogs
5
0
Fork 0
Code Issues 6 Pull Requests Packages Projects 1 Releases Wiki Activity

Updated summarize memory function

Browse Source
This commit is contained in:
AllfatherHatt 2025-02-21 19:12:22 +01:00
parent b6e6bd5bd7
commit bf61df30dd
1 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
reginaldCog/reginald.py
View File

@ -138,13 +138,15 @@ class ReginaldCog(commands.Cog):
async def summarize_memory(self, ctx, messages): async def summarize_memory(self, ctx, messages):
"""✅ Generates a summary of past conversations for mid-term storage.""" """✅ Generates a structured, compact summary of past conversations for mid-term storage."""
summary_prompt = ( summary_prompt = (
"Condense the following conversation into an efficient, structured summary that maximizes information density while minimizing token usage. " "Summarize the following conversation into a structured, concise format that retains key details while maximizing brevity. "
"Focus on key insights, decisions, disputes, and facts, ensuring clarity and conciseness without excessive detail. " "The summary should be **organized** into clear sections: "
"Categorize information into: (1) Agreed Conclusions, (2) Disputed Points, (3) Notable User Contributions, and (4) Miscellaneous Context. " "\n\n📌 **Key Takeaways:** Important facts or conclusions reached."
"Avoid unnecessary flavor text but ensure all critical meaning and context are retained. Summarize each topic distinctly, avoiding generalization. " "\n🔹 **Disputed Points:** Areas where opinions or facts conflicted."
"Prioritize compression while keeping essential nuances intact." "\n🗣️ **Notable User Contributions:** Key statements from users that shaped the discussion."
"\n📜 **Additional Context:** Any other relevant information."
"\n\nEnsure the summary is **dense but not overly verbose**. Avoid unnecessary repetition while keeping essential meaning intact."
) )
summary_text = "\n".join(f"{msg['user']}: {msg['content']}" for msg in messages) summary_text = "\n".join(f"{msg['user']}: {msg['content']}" for msg in messages)
@ -161,7 +163,10 @@ class ReginaldCog(commands.Cog):
client = openai.AsyncClient(api_key=api_key) client = openai.AsyncClient(api_key=api_key)
response = await client.chat.completions.create( response = await client.chat.completions.create(
model="gpt-4o-mini", model="gpt-4o-mini",
messages=[{"role": "system", "content": summary_prompt}, {"role": "user", "content": summary_text}], messages=[
{"role": "system", "content": summary_prompt},
{"role": "user", "content": summary_text}
],
max_tokens=2048 max_tokens=2048
) )
@ -190,6 +195,7 @@ class ReginaldCog(commands.Cog):
return random.choice(reginald_responses) return random.choice(reginald_responses)
def extract_topics_from_summary(self, summary): def extract_topics_from_summary(self, summary):
"""Dynamically extracts the most important topics from a summary.""" """Dynamically extracts the most important topics from a summary."""