chatbot with tool - completed

README.md

@@ -148,10 +148,12 @@ To stop the Streamlit app, press `Ctrl+C` in the terminal where the app is runni
 ![Graph Basic Chatbot](screenshots/graph_basic_chatbot.png) 
  
 
-### 2. Chatbot with Tool  
+### 2. Chatbot with Tool
 **Prompt**: Latest news from the India vs Australia match?  
+**Prompt2**: search latest news from india  
 **Reference**:  
-![chatbot with search tool](screenshots/chatbot_with_search_tool.png)  
+![chatbot with search tool](screenshots/chatbot_with_search_tool.png) 
+![Graph chatbot with tool](screenshots/graph_chatbot_with_tool.png)  
 
 ### 3. Appointment Receptionist  
 **Prompt 1**: Book an appointment for priti.  

src/langgraphagenticai/graph/graph_builder.py

@@ -1,4 +1,8 @@
 from langgraph.graph import StateGraph
+from langgraph.prebuilt import tools_condition
+
+from src.langgraphagenticai.tools.search_tool import create_tool_node, get_tools
+from src.langgraphagenticai.node.chatbot_with_tool_node import ChatbotWithToolNode
 from src.langgraphagenticai.node.basic_chatbot_node import BasicChatbotNode
 from src.langgraphagenticai.state.state import State
 
@@ -9,16 +13,60 @@ class GraphBuilder:
     def __init__(self,model):
         self.llm = model
         self.graph_builder = StateGraph(State)
+        
+    def basic_chatbot_build_graph(self):
+        """
+        Builds a basic chatbot graph using LangGraph.
+
+        This method initializes a chatbot node using the `BasicChatbotNode` class 
+        and integrates it into the graph. The chatbot node is set as both the 
+        entry and exit point of the graph.
+        """
+        self.basic_chatbot_node = BasicChatbotNode(self.llm)
+        self.graph_builder.add_node("chatbot", self.basic_chatbot_node.process)
+        self.graph_builder.set_entry_point("chatbot")
+        self.graph_builder.set_finish_point("chatbot")
+
+    def chatbot_with_tool_build_graph(self):
+        """
+        Builds an advanced chatbot graph with tool integration.
+
+        This method creates a chatbot graph that includes both a chatbot node 
+        and a tool node. It defines tools, initializes the chatbot with tool 
+        capabilities, and sets up conditional and direct edges between nodes. 
+        The chatbot node is set as the entry point.
+        """
+        # Define tools and tool node
+        tools = get_tools()
+        tool_node = create_tool_node(tools)
+
+        # Define LLM
+        llm = self.llm
+
+        # Define chatbot node
+        obj_chatbot_with_node = ChatbotWithToolNode(llm)
+        chatbot_node = obj_chatbot_with_node.create_chatbot(tools)
+
+        # Add nodes
+        self.graph_builder.add_node("chatbot", chatbot_node)
+        self.graph_builder.add_node("tools", tool_node)
+
+        # Define conditional and direct edges
+        self.graph_builder.add_conditional_edges("chatbot", tools_condition)
+        self.graph_builder.add_edge("tools", "chatbot")
+
+        # Set entry point and compile graph
+        self.graph_builder.set_entry_point("chatbot")
+
 
     def setup_graph(self, usecase: str):
         """
         Sets up the graph for the selected use case.
         """
         if usecase == "Basic Chatbot":
-            self.basic_chatbot_node = BasicChatbotNode(self.llm)
-            self.graph_builder.add_node("chatbot", self.basic_chatbot_node.process)
-            self.graph_builder.set_entry_point("chatbot")
-            self.graph_builder.set_finish_point("chatbot")
+            self.basic_chatbot_build_graph()
+        elif usecase == "Chatbot with Tool":
+            self.chatbot_with_tool_build_graph()
         else:
             raise ValueError("Invalid use case selected.")
         return self.graph_builder.compile()

src/langgraphagenticai/main.py

@@ -1,8 +1,9 @@
-from src.langgraphagenticai.ui.configfile import Config
 from src.langgraphagenticai.LLMS.groqllm import GroqLLM
 from src.langgraphagenticai.graph.graph_builder import GraphBuilder
 from src.langgraphagenticai.ui.streamlitui.loadui import LoadStreamlitUI
+
 import streamlit as st
+from langchain_core.messages import HumanMessage,AIMessage,ToolMessage
 
 
 # MAIN Function START
@@ -25,14 +26,31 @@ def load_langgraph_agenticai_app():
         graph = graph_builder.setup_graph(usecase)
         
         # Display output in UI
-        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)
+        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)
+        elif usecase =="Chatbot with Tool":
+            # Prepare state and invoke the graph
+            initial_state = {"messages": [user_message]}
+            res = graph.invoke(initial_state)
+            for message in res['messages']:
+                if type(message) == HumanMessage:
+                    with st.chat_message("user"):
+                        st.write(message.content)
+                elif type(message)==ToolMessage:
+                    with st.chat_message("ai"):
+                        st.write("Tool Call Start")
+                        st.write(message.content)
+                        st.write("Tool Call End")
+                elif type(message)==AIMessage and message.content:
+                    with st.chat_message("assistant"):
+                        st.write(message.content)
         
         # display graph
         if graph:

src/langgraphagenticai/node/chatbot_with_tool_node.py

@@ -0,0 +1,37 @@
+from src.langgraphagenticai.state.state import State
+
+class ChatbotWithToolNode:
+    """
+    Chatbot logic enhanced with tool integration.
+    """
+    def __init__(self,model):
+        self.llm = model
+
+    def process(self, state: State) -> dict:
+        """
+        Processes the input state and generates a response with tool integration.
+        """
+        user_input = state["messages"][-1] if state["messages"] else ""
+        llm_response = self.llm.invoke([{"role": "user", "content": user_input}])
+
+        # Simulate tool-specific logic
+        tools_response = f"Tool integration for: '{user_input}'"
+
+        return {"messages": [llm_response, tools_response]}
+
+
+
+    def create_chatbot(self, tools):
+        """
+        Returns a chatbot node function.
+        """
+        llm_with_tools = self.llm.bind_tools(tools)
+
+        def chatbot_node(state: State):
+            """
+            Chatbot logic for processing the input state and returning a response.
+            """
+            return {"messages": [llm_with_tools.invoke(state["messages"])]}
+
+        return chatbot_node
+

src/langgraphagenticai/tools/search_tool.py

@@ -0,0 +1,15 @@
+from langchain_community.tools.tavily_search import TavilySearchResults
+from langgraph.prebuilt import ToolNode
+
+def get_tools():
+    """
+    Returns a list of tools to be used in the chatbot.
+    """
+    tool = TavilySearchResults( max_results=2)
+    return [tool]
+
+def create_tool_node(tools):
+    """
+    Creates and returns a tool node for the graph.
+    """
+    return ToolNode(tools=tools)

src/langgraphagenticai/ui/configfile.ini

@@ -1,5 +1,5 @@
 [DEFAULT]
-PAGE_TITLE = Langgraph IN ACTION
+PAGE_TITLE = LangGraph: Build Stateful Agentic AI graph
 LLM_OPTIONS = Groq
 USECASE_OPTIONS = Basic Chatbot, Chatbot with Tool, Appointment Receptionist, Customer Support
 GROQ_MODEL_OPTIONS = mixtral-8x7b-32768, llama3-8b-8192, llama3-70b-8192, gemma-7b-i

src/langgraphagenticai/ui/streamlitui/loadui.py

@@ -39,19 +39,19 @@ class LoadStreamlitUI:
                                                                                                       type="password")
                 
         if self.user_controls['selected_usecase'] == "Appointment Receptionist":
-                col1, col2 = st.columns(2)
+            col1, col2 = st.columns(2)
 
-                with col1:
-                    st.subheader("Appointment Manager")
+            with col1:
+                st.subheader("Appointment Manager")
 
-                with col2:
-                    st.subheader("Appointments")
+            with col2:
+                st.subheader("Appointments")
+                
+            # TODO :: to be removed
+            st.warning("⌛Revamp is in progress...")
                     
-        elif self.user_controls['selected_usecase']=="Custome Support":
-            st.set_page_config(layout='wide', page_title='Flower Shop Chatbot', page_icon='💐')
-            
-        # TODO :: to be removed
-        if self.user_controls['selected_usecase'] != "Basic Chatbot":
+        elif self.user_controls['selected_usecase']=="Customer Support":
+            # TODO :: to be removed
             st.warning("⌛Revamp is in progress...")
 
         return self.user_controls