"""
Enhanced Pension Rules document with specific focus on pension regulations
"""

ENHANCED_PENSION_RULES = {
    "content": """GOVERNMENT PENSION RULES - COMPREHENSIVE GUIDE

Section 1: PENSION ELIGIBILITY RULES
1.1 Minimum Service Requirement:
   - Qualifying service: 10 years minimum for pension eligibility
   - Short service: Less than 10 years - gratuity only
   - Voluntary retirement: After 20 years of service

1.2 Age-based Retirement Rules:
   - Superannuation age: 60 years (general employees)
   - Extended service: Up to 65 years for specialists (with approval)
   - Early retirement: Allowed after 30 years service or age 50

Section 2: PENSION CALCULATION RULES
2.1 Basic Pension Formula:
   - Standard Formula: (Last drawn basic pay + DA) × service years ÷ 70
   - Minimum pension: ₹9,000 per month (as per 7th Pay Commission)
   - Maximum pension: No upper limit

2.2 Service Counting Rules:
   - Military service: Counts fully
   - Temporary service: Subject to regularization
   - Break in service: May affect pension calculation
   - Foreign service: Counts with specific conditions

2.3 Dearness Relief:
   - Pension DA: Same percentage as serving employees
   - Automatic revision: Every 6 months based on inflation
   - Arrears payment: Applicable from effective date

Section 3: COMMUTATION OF PENSION RULES
3.1 Commutation Eligibility:
   - Maximum commutation: 50% of monthly pension
   - Commutation value: Based on age at retirement
   - One-time payment: Lump sum at retirement

3.2 Restoration Rules:
   - Restoration period: After 15 years from retirement
   - Full restoration: Original pension amount restored
   - Medical benefits: Continue throughout

Section 4: FAMILY PENSION RULES
4.1 Eligibility Conditions:
   - Death during service: Family gets pension
   - Death after retirement: Family continues to get pension
   - Widow/widower: Entitled to family pension
   - Children: Until age 25 or marriage (whichever earlier)
   - Parents: If no spouse/children eligible

4.2 Family Pension Rates:
   - Enhanced rate: 50% of last pay for first 10 years
   - Normal rate: 30% of last pay thereafter
   - Minimum family pension: ₹9,000 per month

Section 5: PENSION PROCESSING RULES
5.1 Application Timeline:
   - Advance application: 6 months before retirement
   - Document submission: All clearances required
   - PPO issuance: Pension Payment Order within 30 days
   - First payment: Within 45 days of retirement

5.2 Required Documents:
   - Service records verification
   - Medical fitness certificate
   - Nomination forms (for family pension)
   - Bank account details
   - Property return
   - No dues certificates

Section 6: PENSION REVISION RULES
6.1 Pay Commission Benefits:
   - Pension revision: As per pay commission recommendations
   - Effective date: Same as pay revision for serving employees
   - Arrears calculation: From effective date
   - Automatic updation: Through pension disbursing agencies

6.2 Court Order Compliance:
   - Legal modifications: As per court directions
   - Appeal provisions: Available for pension disputes
   - Tribunal jurisdiction: Armed Forces Tribunal, CAT

Section 7: SPECIAL PENSION PROVISIONS
7.1 Disability Pension:
   - Service-related disability: Enhanced pension rates
   - Medical invalidation: Special provisions
   - Constant attendance allowance: For severely disabled

7.2 Ex-gratia Payments:
   - Extraordinary circumstances: Compassionate allowance
   - Natural calamities: Special relief measures
   - Hardship cases: Additional support provisions

Section 8: MEDICAL BENEFITS RULES
8.1 Retired Employee Benefits:
   - CGHS continuation: Lifetime medical facility
   - Reimbursement rules: As per government norms
   - Emergency treatment: Immediate approval provisions

8.2 Family Coverage:
   - Dependent coverage: Spouse and unmarried children
   - Age limits: Children covered till 25 years
   - Disabled dependents: Lifetime coverage

Section 9: PENSION PAYMENT RULES
9.1 Payment Schedule:
   - Monthly payment: Last working day of month
   - Electronic transfer: Mandatory bank payment
   - Life certificate: Annual submission required

9.2 Arrears and Adjustments:
   - Arrears payment: Within 60 days of order
   - Recovery procedures: For excess payments
   - Interest on delays: As per government rules

Section 10: APPEAL AND GRIEVANCE RULES
10.1 Grievance Mechanism:
   - First appeal: To Head of Department
   - Second appeal: To Secretary level
   - Final appeal: To Pension Appellate Authority

10.2 Time Limits:
   - Appeal period: 3 months from date of order
   - Extension: Possible with valid reasons
   - Review provisions: For new evidence

These pension rules are based on Central Civil Services (Pension) Rules, 2021 and subsequent amendments. State governments may have similar rules with local variations.""",
    "filename": "comprehensive_pension_rules.txt",
    "source": "CCS Pension Rules 2021 - Updated Guide"
}

# Function to add enhanced pension document
async def add_enhanced_pension_rules():
    """Add comprehensive pension rules to the knowledge base"""
    try:
        from lancedb_service import lancedb_service
        
        # Add the enhanced pension document
        result = await lancedb_service.add_document(
            content=ENHANCED_PENSION_RULES["content"],
            filename=ENHANCED_PENSION_RULES["filename"],
            source=ENHANCED_PENSION_RULES["source"]
        )
        
        print(f"✅ Enhanced pension rules added to knowledge base")
        return result
    except Exception as e:
        print(f"❌ Error adding pension rules: {e}")
        return None

if __name__ == "__main__":
    import asyncio
    asyncio.run(add_enhanced_pension_rules())