orbitalo/homelab-brain
1
0
Fork 0
Code Issues 41 Pull requests Projects Releases Packages Wiki Activity Actions

Memory: Duplikat-Handling mit statusabhaengiger Rueckmeldung + Buttons bei bestehendem Kandidat

Browse source
This commit is contained in:
root 2026-03-15 13:18:20 +07:00
parent 49141a4da4
commit 22404bdccc
2 changed files with 59 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
homelab-ai-bot/context.py
View file

@ -294,8 +294,15 @@ def _tool_memory_read(scope=""):
return "\n".join(lines) return "\n".join(lines)
last_suggest_result = {"type": None, "candidate_id": None}
def _tool_memory_suggest(scope, kind, content, memory_type="temporary", expires_at=None): def _tool_memory_suggest(scope, kind, content, memory_type="temporary", expires_at=None):
import memory_client import memory_client
from datetime import datetime
global last_suggest_result
last_suggest_result = {"type": None, "candidate_id": None}
if memory_type not in ("temporary", "permanent"): if memory_type not in ("temporary", "permanent"):
memory_type = "temporary" memory_type = "temporary"
@ -321,8 +328,34 @@ def _tool_memory_suggest(scope, kind, content, memory_type="temporary", expires_
result = memory_client._post("/memory", data) result = memory_client._post("/memory", data)
if result and result.get("duplicate"): if result and result.get("duplicate"):
return f"Bereits gespeichert (ID {result.get('existing_id')})." ex_status = result.get("existing_status", "?")
ex_type = result.get("existing_memory_type", "")
ex_exp = result.get("existing_expires_at")
ex_id = result.get("existing_id")
if ex_status == "candidate":
last_suggest_result = {"type": "existing_candidate", "candidate_id": ex_id}
return "Noch nicht bestaetigt — zeige Auswahl erneut."
elif ex_status == "active":
if ex_type == "temporary" and ex_exp:
exp_str = datetime.fromtimestamp(ex_exp).strftime("%d.%m.%Y")
last_suggest_result = {"type": "active_temporary", "candidate_id": None}
return f"Schon temporaer gespeichert bis {exp_str}."
elif ex_type == "permanent":
last_suggest_result = {"type": "active_permanent", "candidate_id": None}
return "Schon dauerhaft gespeichert."
else:
last_suggest_result = {"type": "active_other", "candidate_id": None}
return "Bereits aktiv gespeichert."
elif ex_status == "archived":
last_suggest_result = {"type": "existing_candidate", "candidate_id": ex_id}
memory_client._patch(f"/memory/{ex_id}", {"status": "candidate"})
return "War archiviert — erneut als Kandidat vorgeschlagen."
else:
return "Bereits vorhanden."
if result and result.get("ok"): if result and result.get("ok"):
last_suggest_result = {"type": "new_candidate", "candidate_id": None}
type_label = "temporaer" if memory_type == "temporary" else "dauerhaft" type_label = "temporaer" if memory_type == "temporary" else "dauerhaft"
return f"Vorschlag gespeichert als Kandidat ({type_label})." return f"Vorschlag gespeichert als Kandidat ({type_label})."
return "Konnte Vorschlag nicht speichern." return "Konnte Vorschlag nicht speichern."

34
homelab-ai-bot/telegram_bot.py
View file

@ -384,6 +384,7 @@ async def handle_message(update: Update, ctx: ContextTypes.DEFAULT_TYPE):
await update.message.reply_text("🤔 Denke nach...") await update.message.reply_text("🤔 Denke nach...")
try: try:
context.last_suggest_result = {"type": None, "candidate_id": None}
candidates_before = {c["id"] for c in memory_client.get_candidates()} candidates_before = {c["id"] for c in memory_client.get_candidates()}
handlers = context.get_tool_handlers(session_id=session_id) handlers = context.get_tool_handlers(session_id=session_id)
answer = llm.ask_with_tools(text, handlers, session_id=session_id) answer = llm.ask_with_tools(text, handlers, session_id=session_id)
@ -391,16 +392,31 @@ async def handle_message(update: Update, ctx: ContextTypes.DEFAULT_TYPE):
memory_client.log_message(session_id, "user", text) memory_client.log_message(session_id, "user", text)
memory_client.log_message(session_id, "assistant", answer) memory_client.log_message(session_id, "assistant", answer)
candidates_after = memory_client.get_candidates() suggest = context.last_suggest_result
new_candidates = [c for c in candidates_after if c["id"] not in candidates_before]
if new_candidates: if suggest["type"] == "existing_candidate" and suggest["candidate_id"]:
c = new_candidates[0] candidates = memory_client.get_candidates()
type_icon = "🕒" if c.get("memory_type") == "temporary" else "📌" if c.get("memory_type") == "permanent" else "📝" item = next((c for c in candidates if c["id"] == suggest["candidate_id"]), None)
await update.message.reply_text( if item:
answer[:4000] + "\n\n" + type_icon + " " + c["content"], await update.message.reply_text(
reply_markup=_memory_buttons(c["id"]), answer[:4000] + "\n\n" + _format_candidate(item),
) reply_markup=_memory_buttons(item["id"]),
)
else:
await update.message.reply_text(answer[:4000], reply_markup=KEYBOARD)
elif suggest["type"] == "new_candidate":
candidates_after = memory_client.get_candidates()
new_candidates = [c for c in candidates_after if c["id"] not in candidates_before]
if new_candidates:
c = new_candidates[0]
type_icon = "🕒" if c.get("memory_type") == "temporary" else "📌" if c.get("memory_type") == "permanent" else "📝"
await update.message.reply_text(
answer[:4000] + "\n\n" + type_icon + " " + c["content"],
reply_markup=_memory_buttons(c["id"]),
)
else:
await update.message.reply_text(answer[:4000], reply_markup=KEYBOARD)
else: else:
await update.message.reply_text(answer[:4000], reply_markup=KEYBOARD) await update.message.reply_text(answer[:4000], reply_markup=KEYBOARD)
except Exception as e: except Exception as e: