mirror of
https://github.com/open-jarvis/OpenJarvis.git
synced 2026-07-28 14:07:55 +00:00
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
"""Backward-compat tests: ensure old import paths still work.
|
|
|
|
The canonical tests are in test_native_react.py. This file verifies
|
|
that ``from openjarvis.agents.react import ReActAgent`` still works
|
|
and produces a working agent.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
from openjarvis.agents.native_react import NativeReActAgent
|
|
from openjarvis.agents.react import ReActAgent
|
|
|
|
|
|
def _engine_response(content):
|
|
return {
|
|
"content": content,
|
|
"usage": {"prompt_tokens": 10, "completion_tokens": 5, "total_tokens": 15},
|
|
"model": "test-model",
|
|
"finish_reason": "stop",
|
|
}
|
|
|
|
|
|
class TestReActShim:
|
|
def test_is_native_react(self):
|
|
"""ReActAgent imported from old path is NativeReActAgent."""
|
|
assert ReActAgent is NativeReActAgent
|
|
|
|
def test_can_instantiate(self):
|
|
engine = MagicMock()
|
|
engine.engine_id = "mock"
|
|
agent = ReActAgent(engine, "test-model")
|
|
assert agent.agent_id == "native_react"
|
|
|
|
def test_can_run(self):
|
|
engine = MagicMock()
|
|
engine.engine_id = "mock"
|
|
engine.generate.return_value = _engine_response(
|
|
"Thought: Simple.\nFinal Answer: Hello!"
|
|
)
|
|
agent = ReActAgent(engine, "test-model")
|
|
result = agent.run("Hello")
|
|
assert result.content == "Hello!"
|