mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-10-06 19:22:13 +00:00
feat: inject AGENTS.md into system prompt
This commit is contained in:
parent
3d9ea5b916
commit
2d5da34b3d
3 changed files with 49 additions and 2 deletions
|
|
@ -4,6 +4,7 @@ You are operating in a **remote Linux sandbox** at `{working_dir}`.
|
||||||
|
|
||||||
All code execution and file operations happen in this sandbox environment.
|
All code execution and file operations happen in this sandbox environment.
|
||||||
|
|
||||||
|
{agents_md_section}
|
||||||
|
|
||||||
**Important:**
|
**Important:**
|
||||||
- Use `{working_dir}` as your working directory for all operations
|
- Use `{working_dir}` as your working directory for all operations
|
||||||
|
|
@ -72,9 +73,20 @@ def construct_system_prompt(
|
||||||
working_dir: str,
|
working_dir: str,
|
||||||
linear_project_id: str = "",
|
linear_project_id: str = "",
|
||||||
linear_issue_number: str = "",
|
linear_issue_number: str = "",
|
||||||
|
agents_md: str | None = None,
|
||||||
) -> str:
|
) -> str:
|
||||||
|
agents_md_section = ""
|
||||||
|
if agents_md:
|
||||||
|
agents_md_section = (
|
||||||
|
"\nThe following text is pulled from the repository's AGENTS.md file. "
|
||||||
|
"It may contain specific instructions and guidelines for the agent.\n"
|
||||||
|
"<agents_md>\n"
|
||||||
|
f"{agents_md}\n"
|
||||||
|
"</agents_md>\n"
|
||||||
|
)
|
||||||
return SYSTEM_PROMPT.format(
|
return SYSTEM_PROMPT.format(
|
||||||
working_dir=working_dir,
|
working_dir=working_dir,
|
||||||
linear_project_id=linear_project_id or "<PROJECT_ID>",
|
linear_project_id=linear_project_id or "<PROJECT_ID>",
|
||||||
linear_issue_number=linear_issue_number or "<ISSUE_NUMBER>",
|
linear_issue_number=linear_issue_number or "<ISSUE_NUMBER>",
|
||||||
|
agents_md_section=agents_md_section,
|
||||||
)
|
)
|
||||||
|
|
|
||||||
|
|
@ -42,6 +42,7 @@ SANDBOX_CREATING = "__creating__"
|
||||||
SANDBOX_CREATION_TIMEOUT = 180
|
SANDBOX_CREATION_TIMEOUT = 180
|
||||||
SANDBOX_POLL_INTERVAL = 1.0
|
SANDBOX_POLL_INTERVAL = 1.0
|
||||||
|
|
||||||
|
from .utils.agents_md import read_agents_md_in_sandbox
|
||||||
from .utils.github import (
|
from .utils.github import (
|
||||||
git_has_uncommitted_changes,
|
git_has_uncommitted_changes,
|
||||||
is_valid_git_repo,
|
is_valid_git_repo,
|
||||||
|
|
@ -136,8 +137,10 @@ async def _clone_or_pull_repo_in_sandbox( # noqa: PLR0915
|
||||||
logger.exception("Failed to restore clean remote URL")
|
logger.exception("Failed to restore clean remote URL")
|
||||||
raise
|
raise
|
||||||
|
|
||||||
logger.info("Repo updated at %s", repo_dir)
|
logger.info("Repo updated at %s", repo_dir)
|
||||||
return repo_dir
|
return repo_dir
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
logger.info("Cloning repo %s/%s to %s", owner, repo, repo_dir)
|
logger.info("Cloning repo %s/%s to %s", owner, repo, repo_dir)
|
||||||
try:
|
try:
|
||||||
|
|
@ -327,6 +330,7 @@ async def get_agent(config: RunnableConfig) -> Pregel: # noqa: PLR0915
|
||||||
linear_issue = config["configurable"].get("linear_issue", {})
|
linear_issue = config["configurable"].get("linear_issue", {})
|
||||||
linear_project_id = linear_issue.get("linear_project_id", "")
|
linear_project_id = linear_issue.get("linear_project_id", "")
|
||||||
linear_issue_number = linear_issue.get("linear_issue_number", "")
|
linear_issue_number = linear_issue.get("linear_issue_number", "")
|
||||||
|
agents_md = await read_agents_md_in_sandbox(sandbox_backend, repo_dir)
|
||||||
|
|
||||||
logger.info("Returning agent with sandbox for thread %s", thread_id)
|
logger.info("Returning agent with sandbox for thread %s", thread_id)
|
||||||
return create_deep_agent(
|
return create_deep_agent(
|
||||||
|
|
@ -335,6 +339,7 @@ async def get_agent(config: RunnableConfig) -> Pregel: # noqa: PLR0915
|
||||||
repo_dir,
|
repo_dir,
|
||||||
linear_project_id=linear_project_id,
|
linear_project_id=linear_project_id,
|
||||||
linear_issue_number=linear_issue_number,
|
linear_issue_number=linear_issue_number,
|
||||||
|
agents_md=agents_md,
|
||||||
),
|
),
|
||||||
tools=[http_request, fetch_url, commit_and_open_pr],
|
tools=[http_request, fetch_url, commit_and_open_pr],
|
||||||
backend=sandbox_backend,
|
backend=sandbox_backend,
|
||||||
|
|
|
||||||
30
apps/agent/agent/utils/agents_md.py
Normal file
30
apps/agent/agent/utils/agents_md.py
Normal file
|
|
@ -0,0 +1,30 @@
|
||||||
|
"""Helpers for reading agent instructions from AGENTS.md."""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import asyncio
|
||||||
|
import shlex
|
||||||
|
|
||||||
|
from deepagents.backends.protocol import SandboxBackendProtocol
|
||||||
|
|
||||||
|
|
||||||
|
async def read_agents_md_in_sandbox(
|
||||||
|
sandbox_backend: SandboxBackendProtocol,
|
||||||
|
repo_dir: str | None,
|
||||||
|
) -> str | None:
|
||||||
|
"""Read AGENTS.md from the repo root if it exists."""
|
||||||
|
if not repo_dir:
|
||||||
|
return None
|
||||||
|
|
||||||
|
safe_agents_path = shlex.quote(f"{repo_dir}/AGENTS.md")
|
||||||
|
loop = asyncio.get_event_loop()
|
||||||
|
result = await loop.run_in_executor(
|
||||||
|
None,
|
||||||
|
sandbox_backend.execute,
|
||||||
|
f"test -f {safe_agents_path} && cat {safe_agents_path}",
|
||||||
|
)
|
||||||
|
if result.exit_code != 0:
|
||||||
|
return None
|
||||||
|
content = result.output or ""
|
||||||
|
content = content.strip()
|
||||||
|
return content or None
|
||||||
Loading…
Add table
Reference in a new issue