Umar - Bootcamp

week8/community_contributions/w8d5/tests/test_components.py

@@ -0,0 +1,66 @@
+import os
+import sys
+from dotenv import load_dotenv
+
+project_root = os.path.join(os.path.dirname(__file__), '..')
+sys.path.insert(0, project_root)
+sys.path.insert(0, os.path.join(project_root, '..', '..'))
+
+from helpers.travel_deals import ScrapedTravelDeal
+from agents.travel_scanner_agent import TravelScannerAgent
+from agents.travel_estimator_agent import TravelEstimatorAgent
+
+load_dotenv()
+
+print("\nTesting Travel Deal Hunter Components\n")
+
+print("1. RSS Feed Scraping")
+deals = ScrapedTravelDeal.fetch(show_progress=False)
+print(f"Fetched {len(deals)} deals from RSS feeds")
+if deals:
+    print(f"Sample: {deals[0].title[:60]}...")
+
+
+print("\n2. OpenAI Connection")
+if os.getenv("OPENAI_API_KEY"):
+    print("OPENAI_API_KEY found")
+else:
+    print("OPENAI_API_KEY not found - set in .env file")
+
+print("\n3. Scanner Agent")
+scanner = TravelScannerAgent()
+print("Scanner agent initialized")
+
+print("\n4. Deal Scanning")
+try:
+    selection = scanner.scan(memory=[])
+    if selection and selection.deals:
+        print(f"Scanner found {len(selection.deals)} processed deals")
+        print(f"Sample: {selection.deals[0].destination} - ${selection.deals[0].price}")
+    else:
+        print("No deals returned")
+except Exception as e:
+    print(f"Error: {e}")
+
+print("\n5. ChromaDB Access")
+import chromadb
+try:
+    db_path = "travel_vectorstore"
+    client = chromadb.PersistentClient(path=db_path)
+    collection = client.get_or_create_collection('travel_deals')
+    count = collection.count()
+    print(f"ChromaDB connected - {count} travel items in collection")
+except Exception as e:
+    print(f"Error: {e}")
+
+print("\n6. Estimator Check using travel vectorstore")
+try:
+    estimator = TravelEstimatorAgent(collection)
+    sample = "Non-stop economy flight from New York to London, duration 7 hours"
+    estimate = estimator.estimate(sample)
+    print(f"Estimate: ${estimate:.2f}")
+except Exception as e:
+    print(f"Error: {e}")
+
+print("\nComponent tests complete")
+