#!/usr/bin/env python3
import os, sys, json
from qdrant_client import QdrantClient
from neo4j import GraphDatabase
ROOT = os.path.dirname(os.path.dirname(__file__))
CFG_PATH = os.path.join(ROOT, "config", "memory.config.json")
sys.path.append(os.path.join(ROOT, "scripts"))
from config_loader import load_config # type: ignore
CFG = load_config(CFG_PATH)
def main():
ok = True
msgs = []
try:
q = QdrantClient(url=CFG["qdrant"]["url"])
_ = q.get_collections()
msgs.append("Qdrant: OK")
except Exception as e:
ok = False; msgs.append(f"Qdrant: FAIL - {e}")
try:
d = GraphDatabase.driver(CFG["kg"]["uri"], auth=(CFG["kg"]["user"], CFG["kg"]["pass"]))
with d.session() as s:
s.run("RETURN 1 as one").single()
msgs.append("Neo4j: OK")
except Exception as e:
ok = False; msgs.append(f"Neo4j: FAIL - {e}")
print("\n".join(msgs))
if not ok: sys.exit(1)
if __name__ == "__main__":
main()
