Spaces:
Runtime error
Runtime error
| import pytest | |
| from app.agents.dev_agent import DevAgent | |
| def test_dev_agent_clarification_flow(): | |
| agent = DevAgent() | |
| history = [] | |
| # 1. First prompt (generic requirement description) | |
| result = agent.process( | |
| session_id="test-session", | |
| user_message="I want to create an agent that analyzes incoming support requests and alerts.", | |
| history=history, | |
| state_init=None | |
| ) | |
| assert result["status"] == "awaiting_clarification" | |
| assert len(result["clarification_questions"]) > 0 | |
| assert result["draft_spec"] is None | |
| def test_dev_agent_complete_generation_flow(): | |
| agent = DevAgent() | |
| # Simulate collected requirements state | |
| state_init = { | |
| "extracted_requirements": { | |
| "name": "Support Alert Agent", | |
| "description": "Analyzes alerts and posts to REST API", | |
| "trigger": "api", | |
| "connection": "api", | |
| "connection_base_url": "https://api.pagerduty.com", | |
| "error_on_failure": "retry", | |
| "retry_max_attempts": 3, | |
| "mode": "workflow" | |
| }, | |
| "missing_information": [], | |
| "clarification_questions": [], | |
| "draft_spec": None, | |
| "validation_result": None, | |
| "status": "collecting_requirements" | |
| } | |
| result = agent.process( | |
| session_id="test-session", | |
| user_message="Make the API connection base URL https://api.pagerduty.com and trigger via API call. Failures should retry 3 times.", | |
| history=[], | |
| state_init=state_init | |
| ) | |
| assert result["status"] == "completed" | |
| assert result["draft_spec"] is not None | |
| assert result["draft_spec"]["name"] == "Support Alert Agent" | |
| assert result["draft_spec"]["connection"]["config"]["base_url"] == "https://api.pagerduty.com" | |
| assert result["draft_spec"]["error_handling"]["retry"]["max_attempts"] == 3 | |