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>
261 lines
9.0 KiB
Python
261 lines
9.0 KiB
Python
from prometheus.models.context import Context
|
|
from prometheus.utils.knowledge_graph_utils import (
|
|
knowledge_graph_data_for_context_generator,
|
|
sort_contexts,
|
|
)
|
|
|
|
|
|
def test_empty_data():
|
|
"""Test with None or empty data"""
|
|
assert knowledge_graph_data_for_context_generator(None) == []
|
|
assert knowledge_graph_data_for_context_generator([]) == []
|
|
|
|
|
|
def test_skip_empty_content():
|
|
"""Test that empty or whitespace-only content is skipped"""
|
|
data = [
|
|
{
|
|
"FileNode": {"relative_path": "test.py"},
|
|
"ASTNode": {"text": "", "start_line": 1, "end_line": 1},
|
|
},
|
|
{
|
|
"FileNode": {"relative_path": "test.py"},
|
|
"TextNode": {"text": " \n\t ", "start_line": 2, "end_line": 2},
|
|
},
|
|
{
|
|
"FileNode": {"relative_path": "test.py"},
|
|
"ASTNode": {"text": "valid content", "start_line": 3, "end_line": 3},
|
|
},
|
|
]
|
|
result = knowledge_graph_data_for_context_generator(data)
|
|
assert len(result) == 1
|
|
assert result[0].content == "valid content"
|
|
|
|
|
|
def test_deduplication_identical_content():
|
|
"""Test deduplication of identical content"""
|
|
data = [
|
|
{
|
|
"FileNode": {"relative_path": "test.py"},
|
|
"ASTNode": {"text": "def hello():", "start_line": 1, "end_line": 1},
|
|
},
|
|
{
|
|
"FileNode": {"relative_path": "test.py"},
|
|
"TextNode": {"text": "def hello():", "start_line": 1, "end_line": 1},
|
|
},
|
|
]
|
|
result = knowledge_graph_data_for_context_generator(data)
|
|
assert len(result) == 1
|
|
assert result[0].content == "def hello():"
|
|
|
|
|
|
def test_deduplication_content_containment():
|
|
"""Test deduplication when one content contains another"""
|
|
data = [
|
|
{
|
|
"FileNode": {"relative_path": "test.py"},
|
|
"ASTNode": {"text": "def hello():\n print('world')", "start_line": 1, "end_line": 2},
|
|
},
|
|
{
|
|
"FileNode": {"relative_path": "test.py"},
|
|
"TextNode": {"text": "print('world')", "start_line": 2, "end_line": 2},
|
|
},
|
|
]
|
|
result = knowledge_graph_data_for_context_generator(data)
|
|
assert len(result) == 1
|
|
assert result[0].content == "def hello():\n print('world')"
|
|
|
|
|
|
def test_deduplication_line_containment():
|
|
"""Test deduplication based on line number containment"""
|
|
data = [
|
|
{
|
|
"FileNode": {"relative_path": "test.py"},
|
|
"ASTNode": {"text": "function body", "start_line": 1, "end_line": 5},
|
|
},
|
|
{
|
|
"FileNode": {"relative_path": "test.py"},
|
|
"TextNode": {"text": "inner content", "start_line": 2, "end_line": 3},
|
|
},
|
|
]
|
|
result = knowledge_graph_data_for_context_generator(data)
|
|
assert len(result) == 1
|
|
assert result[0].content == "function body"
|
|
assert result[0].start_line_number == 1
|
|
assert result[0].end_line_number == 5
|
|
|
|
|
|
def test_different_files_no_deduplication():
|
|
"""Test that identical content in different files is not deduplicated"""
|
|
data = [
|
|
{
|
|
"FileNode": {"relative_path": "file1.py"},
|
|
"ASTNode": {"text": "def hello():", "start_line": 1, "end_line": 1},
|
|
},
|
|
{
|
|
"FileNode": {"relative_path": "file2.py"},
|
|
"ASTNode": {"text": "def hello():", "start_line": 1, "end_line": 1},
|
|
},
|
|
]
|
|
result = knowledge_graph_data_for_context_generator(data)
|
|
assert len(result) == 2
|
|
assert result[0].relative_path == "file1.py"
|
|
assert result[1].relative_path == "file2.py"
|
|
|
|
|
|
def test_content_stripping():
|
|
"""Test that content is properly stripped of whitespace"""
|
|
data = [
|
|
{
|
|
"FileNode": {"relative_path": "test.py"},
|
|
"ASTNode": {"text": " \n def hello(): \n ", "start_line": 1, "end_line": 1},
|
|
}
|
|
]
|
|
result = knowledge_graph_data_for_context_generator(data)
|
|
assert len(result) == 1
|
|
assert result[0].content == "def hello():"
|
|
|
|
|
|
def test_complex_deduplication_scenario():
|
|
"""Test complex scenario with multiple overlapping contexts"""
|
|
data = [
|
|
# Large context containing everything
|
|
{
|
|
"FileNode": {"relative_path": "test.py"},
|
|
"ASTNode": {
|
|
"text": "class MyClass:\n def method1(self):\n pass\n def method2(self):\n pass",
|
|
"start_line": 1,
|
|
"end_line": 5,
|
|
},
|
|
},
|
|
# Smaller context contained within the large one
|
|
{
|
|
"FileNode": {"relative_path": "test.py"},
|
|
"TextNode": {
|
|
"text": "def method1(self):\n pass",
|
|
"start_line": 2,
|
|
"end_line": 3,
|
|
},
|
|
},
|
|
# Separate context in same file
|
|
{
|
|
"FileNode": {"relative_path": "test.py"},
|
|
"SelectedLines": {"text": "# Comment at end", "start_line": 10, "end_line": 10},
|
|
},
|
|
]
|
|
result = knowledge_graph_data_for_context_generator(data)
|
|
assert len(result) == 2 # Large context + separate comment
|
|
assert "class MyClass:" in result[0].content
|
|
assert result[1].content == "# Comment at end"
|
|
|
|
|
|
def test_sort_contexts_empty_list():
|
|
"""Test sorting an empty list"""
|
|
result = sort_contexts([])
|
|
assert result == []
|
|
|
|
|
|
def test_sort_contexts_by_relative_path():
|
|
"""Test sorting by relative path"""
|
|
contexts = [
|
|
Context(
|
|
relative_path="src/z.py", content="content z", start_line_number=1, end_line_number=5
|
|
),
|
|
Context(
|
|
relative_path="src/a.py", content="content a", start_line_number=1, end_line_number=5
|
|
),
|
|
Context(
|
|
relative_path="src/m.py", content="content m", start_line_number=1, end_line_number=5
|
|
),
|
|
]
|
|
result = sort_contexts(contexts)
|
|
assert result[0].relative_path == "src/a.py"
|
|
assert result[1].relative_path == "src/m.py"
|
|
assert result[2].relative_path == "src/z.py"
|
|
|
|
|
|
def test_sort_contexts_by_line_numbers():
|
|
"""Test sorting by line numbers within same file"""
|
|
contexts = [
|
|
Context(
|
|
relative_path="test.py", content="content 3", start_line_number=20, end_line_number=25
|
|
),
|
|
Context(
|
|
relative_path="test.py", content="content 1", start_line_number=1, end_line_number=5
|
|
),
|
|
Context(
|
|
relative_path="test.py", content="content 2", start_line_number=10, end_line_number=15
|
|
),
|
|
]
|
|
result = sort_contexts(contexts)
|
|
assert result[0].start_line_number == 1
|
|
assert result[1].start_line_number == 10
|
|
assert result[2].start_line_number == 20
|
|
|
|
|
|
def test_sort_contexts_none_line_numbers():
|
|
"""Test sorting when line numbers are None (should appear last)"""
|
|
contexts = [
|
|
Context(
|
|
relative_path="test.py",
|
|
content="content with lines",
|
|
start_line_number=10,
|
|
end_line_number=15,
|
|
),
|
|
Context(
|
|
relative_path="test.py",
|
|
content="content no lines",
|
|
start_line_number=None,
|
|
end_line_number=None,
|
|
),
|
|
Context(
|
|
relative_path="test.py", content="content first", start_line_number=1, end_line_number=5
|
|
),
|
|
]
|
|
result = sort_contexts(contexts)
|
|
assert result[0].start_line_number == 1
|
|
assert result[1].start_line_number == 10
|
|
assert result[2].start_line_number is None
|
|
|
|
|
|
def test_sort_contexts_mixed_files_and_lines():
|
|
"""Test sorting with multiple files and different line numbers"""
|
|
contexts = [
|
|
Context(
|
|
relative_path="b.py", content="b content 2", start_line_number=20, end_line_number=25
|
|
),
|
|
Context(
|
|
relative_path="a.py", content="a content 2", start_line_number=10, end_line_number=15
|
|
),
|
|
Context(
|
|
relative_path="b.py", content="b content 1", start_line_number=5, end_line_number=10
|
|
),
|
|
Context(
|
|
relative_path="a.py", content="a content 1", start_line_number=1, end_line_number=5
|
|
),
|
|
]
|
|
result = sort_contexts(contexts)
|
|
assert result[0].relative_path == "a.py" and result[0].start_line_number == 1
|
|
assert result[1].relative_path == "a.py" and result[1].start_line_number == 10
|
|
assert result[2].relative_path == "b.py" and result[2].start_line_number == 5
|
|
assert result[3].relative_path == "b.py" and result[3].start_line_number == 20
|
|
|
|
|
|
def test_sort_contexts_end_line_number_tiebreaker():
|
|
"""Test sorting uses end_line_number as tiebreaker when start_line_number is same"""
|
|
contexts = [
|
|
Context(
|
|
relative_path="test.py", content="content 3", start_line_number=10, end_line_number=30
|
|
),
|
|
Context(
|
|
relative_path="test.py", content="content 1", start_line_number=10, end_line_number=15
|
|
),
|
|
Context(
|
|
relative_path="test.py", content="content 2", start_line_number=10, end_line_number=20
|
|
),
|
|
]
|
|
result = sort_contexts(contexts)
|
|
assert result[0].end_line_number == 15
|
|
assert result[1].end_line_number == 20
|
|
assert result[2].end_line_number == 30
|