from pas2 import PAS2
import os
def progress_callback(stage, **kwargs):
"""Custom progress tracking callback"""
if stage == "starting":
print(f"Starting analysis for: {kwargs.get('query')}")
elif stage == "generating_paraphrases":
print("Generating paraphrases...")
elif stage == "paraphrases_complete":
print(f"Generated {kwargs.get('count')} paraphrases")
elif stage == "responses_progress":
completed = kwargs.get('completed_responses', 0)
total = kwargs.get('total_responses', 0)
print(f"Getting responses: {completed}/{total}")
elif stage == "judging":
print("Analyzing for hallucinations...")
elif stage == "complete":
print("Analysis complete!")
# Initialize with progress callback
pas2 = PAS2(
mistral_api_key=os.environ.get("MISTRAL_API_KEY"),
openai_api_key=os.environ.get("OPENAI_API_KEY"),
progress_callback=progress_callback
)
results = pas2.detect_hallucination(
"Who wrote the novel 1984?",
n_paraphrases=2
)