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()
print("Total knowledge:", store.count())

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