| import sys | |
| import os | |
| sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) | |
| from retrieval.graph_retriever import GraphRetriever | |
| gr = GraphRetriever() | |
| print("Graph loaded successfully.") | |
| print("Entities:", gr._extract_entities("find heart doctors in silver plan")) | |
| try: | |
| docs = gr.invoke("find heart doctors in silver plan") | |
| print(f"Success! Retrieved {len(docs)} documents.") | |
| for d in docs: | |
| print("DOCUMENT:") | |
| print(d.page_content) | |
| print("-" * 50) | |
| except Exception as e: | |
| print("CRASHED:", e) | |
| import traceback | |
| traceback.print_exc() | |