torbjorn/my-pal-mcp-server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Advertise prompts, fixes https://github.com/BeehiveInnovations/zen-mcp-server/issues/63

Browse Source
This commit is contained in:
Fahad
2025-06-17 11:17:19 +04:00
parent 95556ba9ea
commit be7d80d7aa
6 changed files with 41 additions and 316 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
tests/test_consensus.py
View File

@@ -198,49 +198,6 @@ class TestConsensusTool(unittest.TestCase):
self.assertIn("pro:against", models_used) # critical -> against
self.assertIn("grok", models_used) # neutral (no suffix)
def test_parse_structured_prompt_models_comprehensive(self):
"""Test the structured prompt parsing method"""
# Test basic parsing
result = ConsensusTool.parse_structured_prompt_models("flash:for,o3:against,pro:neutral")
expected = [
{"model": "flash", "stance": "for"},
{"model": "o3", "stance": "against"},
{"model": "pro", "stance": "neutral"},
]
self.assertEqual(result, expected)
# Test with defaults
result = ConsensusTool.parse_structured_prompt_models("flash:for,o3:against,pro")
expected = [
{"model": "flash", "stance": "for"},
{"model": "o3", "stance": "against"},
{"model": "pro", "stance": "neutral"}, # Defaults to neutral
]
self.assertEqual(result, expected)
# Test all neutral
result = ConsensusTool.parse_structured_prompt_models("flash,o3,pro")
expected = [
{"model": "flash", "stance": "neutral"},
{"model": "o3", "stance": "neutral"},
{"model": "pro", "stance": "neutral"},
]
self.assertEqual(result, expected)
# Test with whitespace
result = ConsensusTool.parse_structured_prompt_models(" flash:for , o3:against , pro ")
expected = [
{"model": "flash", "stance": "for"},
{"model": "o3", "stance": "against"},
{"model": "pro", "stance": "neutral"},
]
self.assertEqual(result, expected)
# Test single model
result = ConsensusTool.parse_structured_prompt_models("flash:for")
expected = [{"model": "flash", "stance": "for"}]
self.assertEqual(result, expected)
if __name__ == "__main__":
unittest.main()