more changes

src/langgraphagenticai/graph/graph_builder.py

@@ -3,7 +3,7 @@ from langgraph.prebuilt import tools_condition,ToolNode
 from langchain_core.prompts import ChatPromptTemplate
 from src.langgraphagenticai.state.state import State
 from src.langgraphagenticai.node.basic_chatbot_node import BasicChatbotNode
-from src.langgraphagenticai.node.chatbot_wiyh_tools import ChatbotWithToolNode
+from src.langgraphagenticai.node.chatbot_with_tools import ChatbotWithToolNode
 from src.langgraphagenticai.tools.searchtool import get_tools,create_tool_node
 
 

src/langgraphagenticai/ui/streamlitui/display_result.py

@@ -14,17 +14,21 @@ class DisplayResultStreamlit:
         graph = self.graph
         user_message = self.user_message
         if usecase =="Basic Chatbot":
-                for event in graph.stream({'messages':("user",user_message)}):
-                    print(event.values())
-                    for value in event.values():
-                        print(value['messages'])
-                        with st.chat_message("user"):
-                            st.write(user_message)
-                        with st.chat_message("assistant"):
-                            st.write(value["messages"].content)
+            print("I am here in basic chatbot")
+            for event in graph.stream({'messages':("user",user_message)}):
+                
+                for value in event.values():
+                    print(value['messages'])
+                    with st.chat_message("user"):
+                        st.write(user_message)
+                        
+                    with st.chat_message("assistant"):
+                        st.write(value["messages"].content)
+                        
 
         elif usecase=="Chatbot with Tool":
              # Prepare state and invoke the graph
+            print("I am here ")
             initial_state = {"messages": [user_message]}
             res = graph.invoke(initial_state)
             for message in res['messages']: