We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b423bfc commit 885e3f5Copy full SHA for 885e3f5
llm_utils/graph.py
@@ -11,6 +11,7 @@
11
from llm_utils.chains import (
12
query_refiner_chain,
13
query_maker_chain,
14
+ query_refiner_with_profile_chain,
15
)
16
17
from llm_utils.tools import get_info_from_db
@@ -38,6 +39,17 @@ class QueryMakerState(TypedDict):
38
39
device: str
40
41
42
+# 노드 함수: PROFILE_EXTRACTION 노드
43
+def profile_extraction_node(state: QueryMakerState):
44
+
45
+ result = query_refiner_with_profile_chain.invoke(
46
+ {"question": state["messages"][0].content}
47
+ )
48
49
+ state["question_profile"] = result
50
+ return state
51
52
53
# 노드 함수: QUERY_REFINER 노드
54
def query_refiner_node(state: QueryMakerState):
55
res = query_refiner_chain.invoke(
0 commit comments