mirror of https://github.com/kortix-ai/suna.git
38 lines
876 B
Python
38 lines
876 B
Python
"""
|
|
System prompt module for Suna.so AI agents.
|
|
|
|
Provides functions to generate complete or custom system prompts using a
|
|
modular YAML + JSON architecture.
|
|
"""
|
|
|
|
from .assembler import PromptAssembler
|
|
|
|
_assembler = PromptAssembler()
|
|
|
|
|
|
def get_system_prompt() -> str:
|
|
"""Get the complete system prompt with all capabilities."""
|
|
return _assembler.get_full_prompt()
|
|
|
|
|
|
def get_custom_prompt(
|
|
include_tools: list = None,
|
|
include_templates: list = None
|
|
) -> str:
|
|
"""
|
|
Get a custom system prompt with specific tools and templates.
|
|
|
|
Args:
|
|
include_tools: Tool schemas to include
|
|
include_templates: Templates to include
|
|
|
|
Returns:
|
|
Assembled system prompt string
|
|
"""
|
|
return _assembler.assemble_prompt(
|
|
include_tools=include_tools,
|
|
include_templates=include_templates
|
|
)
|
|
|
|
|
|
SYSTEM_PROMPT = get_system_prompt() |