dasomaru commited on
Commit
f5d128f
Β·
verified Β·
1 Parent(s): db6a6c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -16
app.py CHANGED
@@ -1,16 +1,24 @@
1
- from graph.rag_pipeline_graph import get_rag_pipeline_graph
2
- from type.state_schema import RAGState
3
-
4
- if __name__ == "__main__":
5
- graph = get_rag_pipeline_graph()
6
-
7
- input_data = RAGState(
8
- query="μœ μš©ν•œ μž¬λ¬΄μ •λ³΄μ˜ μ§ˆμ νŠΉμ„±μ— κ΄€ν•œ μ„€λͺ…",
9
- top_k=5
10
- )
11
-
12
- final_state = graph.invoke(input_data)
13
- # print("\n🧠 μ΅œμ’… 응닡:", final_state.final_response)
14
- # print("\n🧠 μ΅œμ’… 응닡:", final_state["final_response"])
15
- print("\n🧠 μ΅œμ’… 응닡:", final_state.get("final_response", "[응닡 μ—†μŒ]"))
16
-
 
 
 
 
 
 
 
 
 
1
+ from fastapi import FastAPI
2
+
3
+ from graph.rag_pipeline_graph import get_rag_pipeline_graph
4
+ from type.state_schema import RAGState
5
+
6
+
7
+ api = FastAPI()
8
+
9
+ # if __name__ == "__main__":
10
+ @app.get("/")
11
+ def rag_pipeline_graph():
12
+ graph = get_rag_pipeline_graph()
13
+
14
+ input_data = RAGState(
15
+ query="μœ μš©ν•œ μž¬λ¬΄μ •λ³΄μ˜ μ§ˆμ νŠΉμ„±μ— κ΄€ν•œ μ„€λͺ…",
16
+ top_k=5
17
+ )
18
+
19
+ final_state = graph.invoke(input_data)
20
+ # print("\n🧠 μ΅œμ’… 응닡:", final_state.final_response)
21
+ # print("\n🧠 μ΅œμ’… 응닡:", final_state["final_response"])
22
+ # print("\n🧠 μ΅œμ’… 응닡:", final_state.get("final_response", "[응닡 μ—†μŒ]"))
23
+ return final_state.get("final_response", "[응닡 μ—†μŒ]")
24
+