mirrors/hermes-agent
1
0
Fork 0
mirror of https://github.com/NousResearch/hermes-agent.git synced 2026-04-25 00:51:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add skip_context_files option to AIAgent for batch processing

Browse source 
- Introduced a new parameter `skip_context_files` in the AIAgent class to control the inclusion of context files (SOUL.md, AGENTS.md, .cursorrules) in the system prompt.
- Updated the _process_single_prompt function to set `skip_context_files` to True, preventing pollution of trajectories during batch processing and data generation.
This commit is contained in:
teknium1 2026-02-16 22:40:31 -08:00
parent a7609c97be
commit 48b5cfd085
2 changed files with 11 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

1
batch_runner.py
View file

@ -276,6 +276,7 @@ def _process_single_prompt(
max_tokens=config.get("max_tokens"), max_tokens=config.get("max_tokens"),
reasoning_config=config.get("reasoning_config"), reasoning_config=config.get("reasoning_config"),
prefill_messages=config.get("prefill_messages"), prefill_messages=config.get("prefill_messages"),
skip_context_files=True, # Don't pollute trajectories with SOUL.md/AGENTS.md
) )
# Run the agent with task_id to ensure each task gets its own isolated VM # Run the agent with task_id to ensure each task gets its own isolated VM

13
run_agent.py
View file

@ -1013,6 +1013,7 @@ class AIAgent:
reasoning_config: Dict[str, Any] = None, reasoning_config: Dict[str, Any] = None,
prefill_messages: List[Dict[str, Any]] = None, prefill_messages: List[Dict[str, Any]] = None,
platform: str = None, platform: str = None,
skip_context_files: bool = False,
): ):
""" """
Initialize the AI Agent. Initialize the AI Agent.
@ -1045,6 +1046,9 @@ class AIAgent:
Example: [{"role": "user", "content": "Hi!"}, {"role": "assistant", "content": "Hello!"}] Example: [{"role": "user", "content": "Hi!"}, {"role": "assistant", "content": "Hello!"}]
platform (str): The interface platform the user is on (e.g. "cli", "telegram", "discord", "whatsapp"). platform (str): The interface platform the user is on (e.g. "cli", "telegram", "discord", "whatsapp").
Used to inject platform-specific formatting hints into the system prompt. Used to inject platform-specific formatting hints into the system prompt.
skip_context_files (bool): If True, skip auto-injection of SOUL.md, AGENTS.md, and .cursorrules
into the system prompt. Use this for batch processing and data generation to avoid
polluting trajectories with user-specific persona or project instructions.
""" """
self.model = model self.model = model
self.max_iterations = max_iterations self.max_iterations = max_iterations
@ -1054,6 +1058,7 @@ class AIAgent:
self.quiet_mode = quiet_mode self.quiet_mode = quiet_mode
self.ephemeral_system_prompt = ephemeral_system_prompt self.ephemeral_system_prompt = ephemeral_system_prompt
self.platform = platform # "cli", "telegram", "discord", "whatsapp", etc. self.platform = platform # "cli", "telegram", "discord", "whatsapp", etc.
self.skip_context_files = skip_context_files
self.log_prefix_chars = log_prefix_chars self.log_prefix_chars = log_prefix_chars
self.log_prefix = f"{log_prefix} " if log_prefix else "" self.log_prefix = f"{log_prefix} " if log_prefix else ""
# Store effective base URL for feature detection (prompt caching, reasoning, etc.) # Store effective base URL for feature detection (prompt caching, reasoning, etc.)
@ -2016,9 +2021,11 @@ class AIAgent:
prompt_parts.append(skills_prompt) prompt_parts.append(skills_prompt)
# Auto-include context files (SOUL.md, AGENTS.md, .cursorrules). # Auto-include context files (SOUL.md, AGENTS.md, .cursorrules).
context_files_prompt = build_context_files_prompt() # Skipped for batch processing / data generation to avoid polluting trajectories.
if context_files_prompt: if not self.skip_context_files:
prompt_parts.append(context_files_prompt) context_files_prompt = build_context_files_prompt()
if context_files_prompt:
prompt_parts.append(context_files_prompt)
# Current local date and time so the model is never confused about # Current local date and time so the model is never confused about
# what day/time it is (LLM training cutoffs can otherwise mislead it). # what day/time it is (LLM training cutoffs can otherwise mislead it).