LocalFoodAI_FerRo988/test_rag.py

import sys
import os
sys.path.insert(0, '/home/roni/LocalFoodAI')
from main import extract_food_context

messages = [{"role": "user", "content": "how many calories in tuna?"}]
context = extract_food_context(messages)
if context:
    print("SUCCESS: Context extracted")
    print(context)
else:
    print("FAILURE: No context extracted")