mirror of
https://github.com/All-Hands-AI/OpenHands.git
synced 2026-04-29 03:00:45 -04: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
74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
import os
|
|
|
|
import pytest
|
|
from pytest import MonkeyPatch
|
|
|
|
import agenthub # noqa: F401
|
|
from openhands.core.exceptions import (
|
|
AgentNotRegisteredError,
|
|
MicroAgentValidationError,
|
|
)
|
|
from openhands.utils.microagent import MicroAgent
|
|
|
|
CONTENT = (
|
|
'# dummy header\n' 'dummy content\n' '## dummy subheader\n' 'dummy subcontent\n'
|
|
)
|
|
|
|
|
|
def test_micro_agent_load(tmp_path, monkeypatch: MonkeyPatch):
|
|
with open(os.path.join(tmp_path, 'dummy.md'), 'w') as f:
|
|
f.write(
|
|
(
|
|
'---\n'
|
|
'name: dummy\n'
|
|
'agent: CodeActAgent\n'
|
|
'require_env_var:\n'
|
|
' SANDBOX_OPENHANDS_TEST_ENV_VAR: "Set this environment variable for testing purposes"\n'
|
|
'---\n' + CONTENT
|
|
)
|
|
)
|
|
|
|
# Patch the required environment variable
|
|
monkeypatch.setenv('SANDBOX_OPENHANDS_TEST_ENV_VAR', 'dummy_value')
|
|
|
|
micro_agent = MicroAgent(os.path.join(tmp_path, 'dummy.md'))
|
|
assert micro_agent is not None
|
|
assert micro_agent.content == CONTENT.strip()
|
|
|
|
|
|
def test_not_existing_agent(tmp_path, monkeypatch: MonkeyPatch):
|
|
with open(os.path.join(tmp_path, 'dummy.md'), 'w') as f:
|
|
f.write(
|
|
(
|
|
'---\n'
|
|
'name: dummy\n'
|
|
'agent: NotExistingAgent\n'
|
|
'require_env_var:\n'
|
|
' SANDBOX_OPENHANDS_TEST_ENV_VAR: "Set this environment variable for testing purposes"\n'
|
|
'---\n' + CONTENT
|
|
)
|
|
)
|
|
monkeypatch.setenv('SANDBOX_OPENHANDS_TEST_ENV_VAR', 'dummy_value')
|
|
|
|
with pytest.raises(AgentNotRegisteredError):
|
|
MicroAgent(os.path.join(tmp_path, 'dummy.md'))
|
|
|
|
|
|
def test_not_existing_env_var(tmp_path):
|
|
with open(os.path.join(tmp_path, 'dummy.md'), 'w') as f:
|
|
f.write(
|
|
(
|
|
'---\n'
|
|
'name: dummy\n'
|
|
'agent: CodeActAgent\n'
|
|
'require_env_var:\n'
|
|
' SANDBOX_OPENHANDS_TEST_ENV_VAR: "Set this environment variable for testing purposes"\n'
|
|
'---\n' + CONTENT
|
|
)
|
|
)
|
|
|
|
with pytest.raises(MicroAgentValidationError) as excinfo:
|
|
MicroAgent(os.path.join(tmp_path, 'dummy.md'))
|
|
|
|
assert 'Set this environment variable for testing purposes' in str(excinfo.value)
|