example.py

import os
import argparse
from dotenv import load_dotenv

from src.agents.legal_agent import LegalAgent

# Load environment variables
load_dotenv()

def main():
    """Example script demonstrating the Agentic Defensor RAG system."""
    
    # Parse command line arguments
    parser = argparse.ArgumentParser(description="Agentic Defensor Example")
    parser.add_argument("--query", type=str, default="En qué tomo se encuentra Contrato Andrea Monsalve",
                      help="The legal query to process")
    parser.add_argument("--top-k", type=int, default=200, help="Number of chunks to retrieve")
    args = parser.parse_args()
    
    # Initialize the legal agent
    print("Initializing legal agent...")
    agent = LegalAgent()
    
    # Process the query
    print(f"\nProcessing query: {args.query}")
    result = agent.answer_query(args.query, args.top_k)
    
    # Display the result
    print("\n" + "="*80)
    print("QUERY:")
    print(args.query)
    print("\nANSWER:")
    print(result["answer"])
    print("="*80)
    
    # Display metadata
    print(f"\nRetrieved {len(result['retrieved_chunks'])} document chunks")
    print(f"Used model: {result['model_used']}")

if __name__ == "__main__":
    main()