from cat.mad_hatter.mad_hatter import MadHatter
from cat.memory.vector_memory import VectorMemoryCollection
from qdrant_client import QdrantClient

# Connect to Qdrant
client = QdrantClient(host="localhost", port=6333)

# Check if collections exist
collections = client.get_collections()
print("Collections:", [c.name for c in collections.collections])

# Try to query episodic directly
episodic_points = client.scroll(
    collection_name="episodic",
    limit=5,
    with_payload=True,
    with_vectors=False
)
print(f"\nEpisodic memories found: {len(episodic_points[0])}")
for point in episodic_points[0]:
    print(f"  - {point.payload.get('page_content', '')[:100]}")

# Try declarative
declarative_points = client.scroll(
    collection_name="declarative",
    limit=5,
    with_payload=True,
    with_vectors=False
)
print(f"\nDeclarative facts found: {len(declarative_points[0])}")
for point in declarative_points[0]:
    print(f"  - {point.payload.get('page_content', '')}")