added checks to conversation history

pull/1240/head
Hugh 2 days ago
parent 8b21367125
commit b56b9eeaca

@ -20,6 +20,7 @@ networkx
aiofiles aiofiles
httpx httpx
requests requests
litellm
# vllm>=0.2.0 # vllm>=0.2.0
aiohttp aiohttp
mcp mcp

@ -123,11 +123,32 @@ def test_llm_council_run():
query = "What is 2 + 2? Provide a brief answer." query = "What is 2 + 2? Provide a brief answer."
result = council.run(query) result = council.run(query)
# Basic assertions
assert result is not None, "Result should not be None" assert result is not None, "Result should not be None"
assert council.conversation is not None, "Conversation should exist" assert council.conversation is not None, "Conversation should exist"
assert len(council.conversation.messages) > 0, "Conversation should have messages" assert len(council.conversation.conversation_history) > 0, "Conversation should have messages"
# Enhanced assertions to verify workflow steps
messages = council.conversation.conversation_history
# Step 1: Verify User query was added
user_messages = [msg for msg in messages if msg.get("role") == "User"]
assert len(user_messages) > 0, "User query should be in conversation"
# Step 2: Verify all council members responded
member_responses = [msg for msg in messages if msg.get("role") in ["TestAgent1", "TestAgent2"]]
assert len(member_responses) == len(custom_members), f"All {len(custom_members)} council members should have responded"
# Step 3: Verify evaluations were performed
evaluation_messages = [msg for msg in messages if "-Evaluation" in msg.get("role", "")]
assert len(evaluation_messages) == len(custom_members), f"All {len(custom_members)} members should have evaluated"
# Step 4: Verify Chairman synthesis occurred
chairman_messages = [msg for msg in messages if msg.get("role") == "Chairman"]
assert len(chairman_messages) > 0, "Chairman should have synthesized final response"
logger.info("✓ Run method test passed") logger.info("✓ Run method test passed")
logger.info(f"✓ Verified {len(member_responses)} member responses, {len(evaluation_messages)} evaluations, and {len(chairman_messages)} chairman synthesis")
except Exception as e: except Exception as e:
logger.error(f"✗ Run method test failed: {e}") logger.error(f"✗ Run method test failed: {e}")

Loading…
Cancel
Save