docker-api / app.py
dasomaru's picture
Upload folder using huggingface_hub
06696b5 verified
raw
history blame
570 Bytes
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", "[응닡 μ—†μŒ]"))