mirror of
https://github.com/All-Hands-AI/OpenHands.git
synced 2026-01-14 09:18:04 -05:00
d9a8b53bc2
* update microagent name and update template.toml * substitute actual micro_agent_name for prompt manager * add python-frontmatter * support micro agent in codeact * add test cases * add instruction from require env var * add draft gh micro agent * update poetry lock * update poetry lock
64 lines
2.3 KiB
Python
64 lines
2.3 KiB
Python
import os
|
|
|
|
from jinja2 import Template
|
|
|
|
from openhands.utils.microagent import MicroAgent
|
|
|
|
|
|
class PromptManager:
|
|
"""
|
|
Manages prompt templates and micro-agents for AI interactions.
|
|
|
|
This class handles loading and rendering of system and user prompt templates,
|
|
as well as loading micro-agent specifications. It provides methods to access
|
|
rendered system and initial user messages for AI interactions.
|
|
|
|
Attributes:
|
|
prompt_dir (str): Directory containing prompt templates.
|
|
agent_skills_docs (str): Documentation of agent skills.
|
|
micro_agent (MicroAgent | None): Micro-agent, if specified.
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
prompt_dir: str,
|
|
agent_skills_docs: str,
|
|
micro_agent: MicroAgent | None = None,
|
|
):
|
|
self.prompt_dir: str = prompt_dir
|
|
self.agent_skills_docs: str = agent_skills_docs
|
|
|
|
self.system_template: Template = self._load_template('system_prompt')
|
|
self.user_template: Template = self._load_template('user_prompt')
|
|
self.micro_agent: MicroAgent | None = micro_agent
|
|
|
|
def _load_template(self, template_name: str) -> Template:
|
|
template_path = os.path.join(self.prompt_dir, f'{template_name}.j2')
|
|
if not os.path.exists(template_path):
|
|
raise FileNotFoundError(f'Prompt file {template_path} not found')
|
|
with open(template_path, 'r') as file:
|
|
return Template(file.read())
|
|
|
|
@property
|
|
def system_message(self) -> str:
|
|
rendered = self.system_template.render(
|
|
agent_skills_docs=self.agent_skills_docs,
|
|
).strip()
|
|
return rendered
|
|
|
|
@property
|
|
def initial_user_message(self) -> str:
|
|
"""This is the initial user message provided to the agent
|
|
before *actual* user instructions are provided.
|
|
|
|
It is used to provide a demonstration of how the agent
|
|
should behave in order to solve the user's task. And it may
|
|
optionally contain some additional context about the user's task.
|
|
These additional context will convert the current generic agent
|
|
into a more specialized agent that is tailored to the user's task.
|
|
"""
|
|
rendered = self.user_template.render(
|
|
micro_agent=self.micro_agent.content if self.micro_agent else None
|
|
)
|
|
return rendered.strip()
|