Spaces:
Runtime error
Runtime error
| from typing import Dict, Any | |
| from .graph import build_graph | |
| class DevAgent: | |
| def __init__(self): | |
| self.graph = build_graph() | |
| def process(self, session_id: str, user_message: str, history: list, state_init: dict = None) -> Dict[str, Any]: | |
| state = { | |
| 'session_id': session_id, | |
| 'user_message': user_message, | |
| 'conversation_history': history, | |
| 'extracted_requirements': (state_init or {}).get('extracted_requirements', {}) or {}, | |
| 'missing_information': (state_init or {}).get('missing_information', []) or [], | |
| 'clarification_questions': (state_init or {}).get('clarification_questions', []) or [], | |
| 'draft_spec': (state_init or {}).get('draft_spec', None), | |
| 'validation_result': (state_init or {}).get('validation_result', None), | |
| 'status': (state_init or {}).get('status', 'collecting_requirements') or 'collecting_requirements' | |
| } | |
| result = self.graph.invoke(state, config={'recursion_limit': 50}) | |
| return result | |