Spaces:
Running on Zero
Running on Zero
| """Test all 3 agents directly""" | |
| import asyncio | |
| import sys | |
| if hasattr(sys.stdout, 'reconfigure'): | |
| sys.stdout.reconfigure(encoding='utf-8', errors='replace') | |
| sys.path.insert(0, 'backend') | |
| async def test_agents(): | |
| # Test 1: Researcher - knowledge mode | |
| print('===== RESEARCHER (knowledge) =====') | |
| from agents.researcher import agent as researcher | |
| r = await researcher.run('What is Section 138 of NIA?', mode='knowledge') | |
| print('Mode:', r.get('mode_used')) | |
| print('Laws:', r.get('laws_retrieved')) | |
| resp = r.get('response', '') | |
| print('Response preview:', resp[:300]) | |
| print() | |
| # Test 2: Researcher - pathfinder mode | |
| print('===== RESEARCHER (pathfinder) =====') | |
| r2 = await researcher.run('How to file an FIR?', mode='pathfinder') | |
| print('Mode:', r2.get('mode_used')) | |
| print('Laws:', r2.get('laws_retrieved')) | |
| resp2 = r2.get('response', '') | |
| print('Response preview:', resp2[:300]) | |
| print() | |
| # Test 3: Drafter | |
| print('===== DRAFTER =====') | |
| from agents.drafter import agent as drafter | |
| d = await drafter.run('Draft a cheque bounce notice for bounced cheque of 5 lakhs', mode='document') | |
| print('Mode:', d.get('mode_used')) | |
| print('Doc type:', d.get('document_type')) | |
| resp3 = d.get('response', '') | |
| print('Response preview:', resp3[:300]) | |
| print() | |
| # Test 4: Reviewer - scrutiny | |
| print('===== REVIEWER (scrutiny) =====') | |
| from agents.reviewer import agent as reviewer | |
| v = await reviewer.run('I had a motor accident 2 years ago and want to file a claim now', mode='scrutiny') | |
| print('Mode:', v.get('mode_used')) | |
| resp4 = v.get('response', '') | |
| print('Response preview:', resp4[:300]) | |
| asyncio.run(test_agents()) | |