mirror of
https://github.com/lrnselfreliance/wrolpi.git
synced 2026-10-26 10:51:36 -04:00
d612b03577
Live testing surfaced three ways small models went wrong summarizing a library: they gave up after one empty search (misspelled channel names found nothing), so zero-result searches now carry a retry hint and the Research prompt requires list_collections before channel/site questions and search_videos(channel_id) before claiming a channel is summarized; they presented the familiar YouTube source URL instead of the WROLPi link, so listings omit the source URL (archive detail keeps it); and they invented links for channels, so channel collections now carry their real /videos/channel page link. Search descriptions state that search_str is optional for browsing the newest items. Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
91 lines
4.2 KiB
Python
91 lines
4.2 KiB
Python
"""Chat modes: a mode is a system prompt plus a tool allowlist, enforced server-side in the agent
|
|
loop — the model cannot opt out. Tool names are the /api/ai blueprint's operationIds."""
|
|
|
|
COMMON_PROMPT = (
|
|
'You are the WROLPi assistant, running locally on the user\'s own offline WROLPi server.'
|
|
' WROLPi is an off-grid digital library: videos, archived web pages, ebooks, Zim encyclopedias,'
|
|
' maps, and inventories. You have no internet access.'
|
|
' Answer briefly and plainly. Use the provided tools to find real information before answering;'
|
|
' never invent library content or system state.'
|
|
)
|
|
|
|
MODES = {
|
|
'help': dict(
|
|
system_prompt=(
|
|
f'{COMMON_PROMPT}'
|
|
' Your job is to teach the user how to use WROLPi.'
|
|
' You do NOT know how WROLPi works from memory: before answering ANY question, you MUST'
|
|
' first call search_help, then read the matching page with get_help_doc, and answer only'
|
|
' from what the docs actually say. Always include the link of every help page you used.'
|
|
' If the docs do not cover the question, say so.'
|
|
),
|
|
tools=(
|
|
'search_help',
|
|
'get_help_doc',
|
|
'get_system_status',
|
|
),
|
|
),
|
|
'research': dict(
|
|
system_prompt=(
|
|
f'{COMMON_PROMPT}'
|
|
' Your job is to be the librarian: find and summarize content from the user\'s library.'
|
|
' Search first, then read captions, page text, or entries when the user wants detail.'
|
|
' When the user asks about a channel, website, or their whole library, call'
|
|
' list_collections first to learn the exact names (spellings often differ).'
|
|
' list_collections only gives names — it never tells you what content exists. You have'
|
|
' not summarized a channel until you have called search_videos with its channel_id (and'
|
|
' NO search_str) and read the resulting titles.'
|
|
' To browse everything of one kind (e.g. "summarize my archives"), call the search tool'
|
|
' WITHOUT search_str — that lists the newest items.'
|
|
' When a search finds nothing, do not give up: retry with fewer or different words, or'
|
|
' browse with list_collections.'
|
|
' ALWAYS include the WROLPi link for every item you mention, exactly as the tools return it.'
|
|
' Never give external URLs. When a search returns a large total, narrow the query instead'
|
|
' of paging.'
|
|
),
|
|
tools=(
|
|
'search_all',
|
|
'search_videos',
|
|
'get_video',
|
|
'get_video_captions',
|
|
'search_archives',
|
|
'get_archive_text',
|
|
'search_docs',
|
|
'get_doc',
|
|
'list_zims',
|
|
'search_zims',
|
|
'get_zim_entry',
|
|
'list_collections',
|
|
'list_inventories',
|
|
'get_inventory',
|
|
'read_file',
|
|
),
|
|
),
|
|
'system': dict(
|
|
system_prompt=(
|
|
f'{COMMON_PROMPT}'
|
|
' Your job is to troubleshoot the user\'s WROLPi.'
|
|
' Start every diagnosis with get_system_status; read a failing service\'s logs with'
|
|
' get_service_logs; check drives with list_disks.'
|
|
' You CANNOT execute commands or change anything — you can only read.'
|
|
' Only suggest commands that appear in the help documentation: find them with search_help,'
|
|
' quote them exactly, and include the help page link.'
|
|
),
|
|
tools=(
|
|
'get_system_status',
|
|
'list_services',
|
|
'get_service_logs',
|
|
'list_disks',
|
|
'search_help',
|
|
'get_help_doc',
|
|
),
|
|
),
|
|
}
|
|
|
|
# Suggestion chips shown by the Chat tab; small models do much better when the opening message
|
|
# matches what the mode's prompt was tuned for.
|
|
MODE_SUGGESTIONS = {
|
|
'help': ['How do I download videos?', 'How do I add my own files?', 'What are Tags?'],
|
|
'research': ['Find videos about canning', 'What do I have about gardening?', 'Summarize my newest archives'],
|
|
'system': ['Why is my drive full?', 'Is everything running?', 'Why do downloads fail?'],
|
|
}
|