Spaces:
Sleeping
Sleeping
File size: 2,322 Bytes
07ea386 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 |
import requests
import json
import time
BASE_URL = "http://localhost:7860"
print("=" * 60)
print("LEGAL RAG BACKEND - API TEST")
print("=" * 60)
print("\n1. Testing health endpoint...")
try:
response = requests.get(f"{BASE_URL}/health", timeout=5)
if response.status_code == 200:
print(f" ✓ Health check passed: {response.json()}")
else:
print(f" ✗ Health check failed: {response.status_code}")
except Exception as e:
print(f" ✗ Cannot connect to server: {e}")
print("\n Make sure the server is running:")
print(" ./start.sh")
exit(1)
print("\n2. Testing /predict endpoint...")
test_case = {
"text": "The accused was found in possession of stolen goods worth Rs. 50,000. The police recovered the items from his residence during a search operation."
}
try:
start = time.time()
response = requests.post(f"{BASE_URL}/predict", json=test_case, timeout=30)
duration = time.time() - start
if response.status_code == 200:
result = response.json()
print(f" ✓ Prediction successful ({duration:.2f}s)")
print(f" Verdict: {result['verdict']}")
print(f" Confidence: {result['confidence']:.4f}")
else:
print(f" ✗ Prediction failed: {response.status_code}")
print(f" Error: {response.text}")
except Exception as e:
print(f" ✗ Error: {e}")
print("\n3. Testing /explain endpoint...")
try:
start = time.time()
response = requests.post(f"{BASE_URL}/explain", json=test_case, timeout=60)
duration = time.time() - start
if response.status_code == 200:
result = response.json()
print(f" ✓ Explanation successful ({duration:.2f}s)")
print(f" Verdict: {result['verdict']}")
print(f" Confidence: {result['confidence']:.4f}")
print(f" Explanation length: {len(result['explanation'])} chars")
print(f" Retrieved chunks:")
for category, chunks in result['retrievedChunks'].items():
print(f" - {category}: {len(chunks)} chunks")
else:
print(f" ✗ Explanation failed: {response.status_code}")
print(f" Error: {response.text}")
except Exception as e:
print(f" ✗ Error: {e}")
print("\n" + "=" * 60)
print("✅ API TESTING COMPLETE")
print("=" * 60)
|