giers10/Heimgeist
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Improve prompt generation by handling long messages and dynamic history context

Browse Source
This commit is contained in:
Victor Giers
2026-06-15 20:26:34 +02:00
parent bd2c003965
commit 983081eccd
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
backend/agent/router.py
View File

@@ -204,14 +204,18 @@ async def select_workflow(
if not model: if not model:
return fallback_result return fallback_result
message_excerpt = str(message or "")
if len(message_excerpt) > 1800:
message_excerpt = message_excerpt[:1800] + "…"
history_chars = 300 if len(message_excerpt) > 1000 else 500
compact_history = [ compact_history = [
{"role": item.get("role"), "content": str(item.get("content") or "")[:700]} {"role": item.get("role"), "content": str(item.get("content") or "")[:history_chars]}
for item in recent_messages[-4:] for item in recent_messages[-4:]
] ]
prompt = ( prompt = (
"Select exactly one enabled workflow for the current request. Do not invent workflows. " "Select exactly one enabled workflow for the current request. Do not invent workflows. "
"Prefer the lowest-cost workflow that can complete the task. Return JSON only.\n\n" "Prefer the lowest-cost workflow that can complete the task. Return JSON only.\n\n"
f"Current request: {message}\n" f"Current request: {message_excerpt}\n"
f"Recent messages: {json.dumps(compact_history, ensure_ascii=False)}\n" f"Recent messages: {json.dumps(compact_history, ensure_ascii=False)}\n"
f"Attachment count: {len(attachments)}\n" f"Attachment count: {len(attachments)}\n"
f"Selected knowledge database: {library_slug or 'none'}\n" f"Selected knowledge database: {library_slug or 'none'}\n"