src/test_heading_parsing.py

"""
Quick test to check heading parsing in API client
"""
import logging
from api_client import APIClient

# Set up logging to see debug info
logging.basicConfig(level=logging.INFO)

# Create API client
client = APIClient("http://localhost:8000")

# Test with a simple query
print("Testing heading parsing...")
response = client.send_query("Show me all employees")

print(f"\nResponse keys: {list(response.keys())}")
print(f"Message: {response.get('message')}")
print(f"Heading: {response.get('heading')}")
print(f"Summary: {response.get('summary')}")
print(f"Original heading: {response.get('original_heading')}")
print(f"Rows count: {len(response.get('rows', []))}")