# 1) 수집 및 추출
texts = load_documents("./docs")
triplets = extract_triplets_with_llm(texts) # (head, relation, tail)
# 2) 그래프 구축
import networkx as nx
g = nx.DiGraph
for h, r, t in triplets:
g.add_node(h)
g.add_node(t)
g.add_edge(h, t, relation=r)
# 3) 하이브리드 검색
query = "2023년에 예산을 초과한 프로젝트에 참여한 공급업체는 누구입니까?"
vector_hits = vector_search(texts, query, top_k=8)
seed_nodes = entities_from_query(query)
# neighborhood 확장
subgraph = expand_neighborhood(g, seed_nodes, depth=2)
# 4) 합성 프롬프트
context = render(vector_hits) + render_paths(subgraph)
answer = llm("""
당신은 정확한 분석가입니다. 컨텍스트의 사실만 사용하여 답변하세요.
관련된 경우 그래프 노드/에지를 인용하세요.
질문: {query}
컨텍스트: {context}
""")
# 5) 평가
assert grounded(answer)