b52318eeae
- Add intelligent-router.sh hook for automatic agent routing - Add AUTO-TRIGGER-SUMMARY.md documentation - Add FINAL-INTEGRATION-SUMMARY.md documentation - Complete Prometheus integration (6 commands + 4 tools) - Complete Dexto integration (12 commands + 5 tools) - Enhanced Ralph with access to all agents - Fix /clawd command (removed disable-model-invocation) - Update hooks.json to v5 with intelligent routing - 291 total skills now available - All 21 commands with automatic routing 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
from unittest.mock import Mock
|
|
|
|
import pytest
|
|
from langchain_core.messages import HumanMessage, SystemMessage
|
|
|
|
from prometheus.graph.knowledge_graph import KnowledgeGraph
|
|
from prometheus.lang_graph.nodes.edit_node import EditNode
|
|
from tests.test_utils.fixtures import temp_test_dir # noqa: F401
|
|
from tests.test_utils.util import FakeListChatWithToolsModel
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_kg():
|
|
kg = Mock(spec=KnowledgeGraph)
|
|
return kg
|
|
|
|
|
|
@pytest.fixture
|
|
def fake_llm():
|
|
return FakeListChatWithToolsModel(responses=["File edit completed successfully"])
|
|
|
|
|
|
def test_init_edit_node(mock_kg, fake_llm, temp_test_dir): # noqa: F811
|
|
"""Test EditNode initialization."""
|
|
node = EditNode(fake_llm, temp_test_dir, mock_kg)
|
|
|
|
assert isinstance(node.system_prompt, SystemMessage)
|
|
assert len(node.tools) == 5 # Should have 5 file operation tools
|
|
assert node.model_with_tools is not None
|
|
|
|
|
|
def test_call_method_basic(mock_kg, fake_llm, temp_test_dir): # noqa: F811
|
|
"""Test basic call functionality without tool execution."""
|
|
node = EditNode(fake_llm, temp_test_dir, mock_kg)
|
|
state = {"edit_messages": [HumanMessage(content="Make the following changes: ...")]}
|
|
|
|
result = node(state)
|
|
|
|
assert "edit_messages" in result
|
|
assert len(result["edit_messages"]) == 1
|
|
assert result["edit_messages"][0].content == "File edit completed successfully"
|