33 lines
984 B
Python
33 lines
984 B
Python
|
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', '')}")
|