File size: 570 Bytes
06696b5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from graph.rag_pipeline_graph import get_rag_pipeline_graph
from type.state_schema import RAGState

if __name__ == "__main__":
    graph = get_rag_pipeline_graph()

    input_data = RAGState(
        query="μœ μš©ν•œ μž¬λ¬΄μ •λ³΄μ˜ μ§ˆμ νŠΉμ„±μ— κ΄€ν•œ μ„€λͺ…",
        top_k=5
    )

    final_state = graph.invoke(input_data)
    # print("\n🧠 μ΅œμ’… 응닡:", final_state.final_response)
    # print("\n🧠 μ΅œμ’… 응닡:", final_state["final_response"])
    print("\n🧠 μ΅œμ’… 응닡:", final_state.get("final_response", "[응닡 μ—†μŒ]"))