import asyncio
import os
from dotenv import load_dotenv
load_dotenv("/root/Agent/.env")

import sys
sys.path.append("/root/Agent")
from knowhub.knowhub_db.pg_store import PostgreSQLStore

store = PostgreSQLStore()

try:
    print("--- Searching for CAP-001 (Uppercase) ---")
    results = store.query("id != ''", limit=10, relation_filters={"capability_id": "CAP-001"})
    print("CAP-001 count:", len(results))
    
    print("\n--- Searching for cap-001 (Lowercase) ---")
    results2 = store.query("id != ''", limit=10, relation_filters={"capability_id": "cap-001"})
    print("cap-001 count:", len(results2))
except Exception as e:
    print("Error:", e)