49 lines
1.8 KiB
Python
49 lines
1.8 KiB
Python
"""Builds the system prompt that is sent to the AI at the start of each chat session."""
|
|
|
|
# Prevents very large files from flooding the context window with tokens.
|
|
MAX_FILE_CHARS = 4000
|
|
|
|
|
|
class SystemPrompter:
|
|
"""Generates system prompts for the chat assistant.
|
|
|
|
When a file is open in the editor it can be embedded in the prompt so the
|
|
AI has direct context of the code the user is currently working on.
|
|
"""
|
|
|
|
@staticmethod
|
|
def generate_prompt(file_context: dict | None = None) -> str:
|
|
"""Build a system prompt, optionally embedding a file's content.
|
|
|
|
Args:
|
|
file_context: dict with keys 'name' (filename) and 'content' (raw text),
|
|
or None if no file should be included.
|
|
|
|
Returns:
|
|
A ready-to-use system prompt string.
|
|
"""
|
|
base = (
|
|
"You are an expert code assistant integrated into a lightweight code editor. "
|
|
"Help the user with code suggestions, debugging, explanations, and improvements. "
|
|
"Be concise and precise. Use markdown and fenced code blocks where appropriate."
|
|
)
|
|
|
|
if file_context:
|
|
name = file_context.get("name", "unknown")
|
|
content = file_context.get("content", "")
|
|
|
|
# Truncate large files to avoid exceeding the model's token limit
|
|
if len(content) > MAX_FILE_CHARS:
|
|
content = content[:MAX_FILE_CHARS] + "\n... [truncated]"
|
|
|
|
file_section = (
|
|
f"\n\nThe user currently has the following file open in the editor:\n"
|
|
f"<file name=\"{name}\">\n"
|
|
f"<code>\n{content}\n</code>\n"
|
|
f"</file>\n"
|
|
f"Refer to this file when answering questions about the code."
|
|
)
|
|
return base + file_section
|
|
|
|
return base
|