Spaces:
Sleeping
Sleeping
Commit
·
5d970ca
1
Parent(s):
0dcbd75
remove print statement
Browse files- src/blogGenerator/LLMs/groqllm.py +1 -1
- src/blogGenerator/graph/graph_builder.py +1 -2
- src/blogGenerator/nodes/content_generator_node.py +1 -1
- src/blogGenerator/nodes/route_deciding_node.py +3 -3
- src/blogGenerator/nodes/yt_transcript_node.py +1 -1
- src/blogGenerator/ui/streamlit/display_result.py +1 -1
src/blogGenerator/LLMs/groqllm.py
CHANGED
@@ -6,7 +6,7 @@ from langchain_groq import ChatGroq
|
|
6 |
class GroqLLM:
|
7 |
def __init__(self, user_controls_input):
|
8 |
self.user_controls_input = user_controls_input
|
9 |
-
print(f"#### self.user_controls_input : {self.user_controls_input}")
|
10 |
|
11 |
def get_llm_model(self):
|
12 |
try:
|
|
|
6 |
class GroqLLM:
|
7 |
def __init__(self, user_controls_input):
|
8 |
self.user_controls_input = user_controls_input
|
9 |
+
# print(f"#### self.user_controls_input : {self.user_controls_input}")
|
10 |
|
11 |
def get_llm_model(self):
|
12 |
try:
|
src/blogGenerator/graph/graph_builder.py
CHANGED
@@ -16,7 +16,7 @@ class GraphBuilder:
|
|
16 |
Build a blog generation graph
|
17 |
"""
|
18 |
|
19 |
-
print("Build a blog generation graph.......")
|
20 |
try:
|
21 |
self.route_deciding_node = RouteDecidingNode(self.llm)
|
22 |
self.yt_transcript_node = YTTranscriptNode()
|
@@ -44,6 +44,5 @@ class GraphBuilder:
|
|
44 |
|
45 |
def setup_graph(self):
|
46 |
self.blog_generator_build_graph()
|
47 |
-
print("setup_graph.......")
|
48 |
|
49 |
return self.graph_builder.compile()
|
|
|
16 |
Build a blog generation graph
|
17 |
"""
|
18 |
|
19 |
+
# print("Build a blog generation graph.......")
|
20 |
try:
|
21 |
self.route_deciding_node = RouteDecidingNode(self.llm)
|
22 |
self.yt_transcript_node = YTTranscriptNode()
|
|
|
44 |
|
45 |
def setup_graph(self):
|
46 |
self.blog_generator_build_graph()
|
|
|
47 |
|
48 |
return self.graph_builder.compile()
|
src/blogGenerator/nodes/content_generator_node.py
CHANGED
@@ -15,7 +15,7 @@ class ContentGeneratorNode:
|
|
15 |
|
16 |
def process(self, state: State):
|
17 |
|
18 |
-
print(f"Node Called : ContentGeneratorNode ---> {state[
|
19 |
|
20 |
blog_llm = self.llm.with_structured_output(BlogContent)
|
21 |
|
|
|
15 |
|
16 |
def process(self, state: State):
|
17 |
|
18 |
+
# print(f"Node Called : ContentGeneratorNode ---> {state['decision']}")
|
19 |
|
20 |
blog_llm = self.llm.with_structured_output(BlogContent)
|
21 |
|
src/blogGenerator/nodes/route_deciding_node.py
CHANGED
@@ -21,7 +21,7 @@ class RouteDecidingNode:
|
|
21 |
def process(self, state: State):
|
22 |
|
23 |
# print(f"state['user_message'] : {state['user_message']}")
|
24 |
-
print(f"Node Called : RouteDecidingNode")
|
25 |
|
26 |
route = self.llm.with_structured_output(Route)
|
27 |
|
@@ -32,7 +32,7 @@ class RouteDecidingNode:
|
|
32 |
]
|
33 |
)
|
34 |
|
35 |
-
print(f"decision : {decision}")
|
36 |
|
37 |
if decision.step == "youtube":
|
38 |
extract_url = self.llm.invoke(
|
@@ -50,7 +50,7 @@ class RouteDecidingNode:
|
|
50 |
|
51 |
def route_decision(self, state: State):
|
52 |
|
53 |
-
print(f"state : {state}")
|
54 |
# Return the node name you want to visit next
|
55 |
if state["decision"] == "youtube":
|
56 |
return "youtube"
|
|
|
21 |
def process(self, state: State):
|
22 |
|
23 |
# print(f"state['user_message'] : {state['user_message']}")
|
24 |
+
# print(f"Node Called : RouteDecidingNode")
|
25 |
|
26 |
route = self.llm.with_structured_output(Route)
|
27 |
|
|
|
32 |
]
|
33 |
)
|
34 |
|
35 |
+
# print(f"decision : {decision}")
|
36 |
|
37 |
if decision.step == "youtube":
|
38 |
extract_url = self.llm.invoke(
|
|
|
50 |
|
51 |
def route_decision(self, state: State):
|
52 |
|
53 |
+
# print(f"state : {state}")
|
54 |
# Return the node name you want to visit next
|
55 |
if state["decision"] == "youtube":
|
56 |
return "youtube"
|
src/blogGenerator/nodes/yt_transcript_node.py
CHANGED
@@ -10,7 +10,7 @@ class YTTranscriptNode:
|
|
10 |
def process(self, state: State) -> dict:
|
11 |
"""Fetches transcript from a given YouTube URL"""
|
12 |
|
13 |
-
print(f"Node Called : yt_transcipt")
|
14 |
|
15 |
video_id = state["yt_url"].replace("https://www.youtube.com/watch?v=", "")
|
16 |
|
|
|
10 |
def process(self, state: State) -> dict:
|
11 |
"""Fetches transcript from a given YouTube URL"""
|
12 |
|
13 |
+
# print(f"Node Called : yt_transcipt")
|
14 |
|
15 |
video_id = state["yt_url"].replace("https://www.youtube.com/watch?v=", "")
|
16 |
|
src/blogGenerator/ui/streamlit/display_result.py
CHANGED
@@ -10,7 +10,7 @@ class DisplayResultStreamlit:
|
|
10 |
graph = self.graph
|
11 |
user_message = self.user_message
|
12 |
|
13 |
-
print(f"### user_message : {user_message}")
|
14 |
|
15 |
with st.chat_message("user"):
|
16 |
st.write(user_message)
|
|
|
10 |
graph = self.graph
|
11 |
user_message = self.user_message
|
12 |
|
13 |
+
# print(f"### user_message : {user_message}")
|
14 |
|
15 |
with st.chat_message("user"):
|
16 |
st.write(user_message)
|