Skip to content

Commit 9ac74de

Browse files
committed
Update speech_graph.py
1 parent 0b4cfd6 commit 9ac74de

File tree

1 file changed

+1
-11
lines changed

1 file changed

+1
-11
lines changed

scrapegraphai/graphs/speech_graph.py

Lines changed: 1 addition & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111
from ..nodes import (
1212
FetchNode,
1313
ParseNode,
14-
RAGNode,
1514
GenerateAnswerNode,
1615
TextToSpeechNode,
1716
)
@@ -72,13 +71,6 @@ def _create_graph(self) -> BaseGraph:
7271
"chunk_size": self.model_token
7372
}
7473
)
75-
rag_node = RAGNode(
76-
input="user_prompt & (parsed_doc | doc)",
77-
output=["relevant_chunks"],
78-
node_config={
79-
"llm_model": self.llm_model,
80-
"embedder_model": self.embedder_model }
81-
)
8274
generate_answer_node = GenerateAnswerNode(
8375
input="user_prompt & (relevant_chunks | parsed_doc | doc)",
8476
output=["answer"],
@@ -100,14 +92,12 @@ def _create_graph(self) -> BaseGraph:
10092
nodes=[
10193
fetch_node,
10294
parse_node,
103-
rag_node,
10495
generate_answer_node,
10596
text_to_speech_node
10697
],
10798
edges=[
10899
(fetch_node, parse_node),
109-
(parse_node, rag_node),
110-
(rag_node, generate_answer_node),
100+
(parse_node, generate_answer_node),
111101
(generate_answer_node, text_to_speech_node)
112102
],
113103
entry_point=fetch_node,

0 commit comments

Comments
 (0)