Chat bot with tools

src/langgraph_agenticai/graph/graph_builder.py

@@ -3,6 +3,8 @@ from langgraph.prebuilt import tools_condition, ToolNode
 from langchain_core.prompts import ChatPromptTemplate
 from src.langgraph_agenticai.state.state import State
 from src.langgraph_agenticai.nodes.basic_chatbot_node import BasicChatbotNode
+from src.langgraph_agenticai.nodes.chatbot_with_tools_node import ChatbotWithToolsNode
+from src.langgraph_agenticai.tools.search_tool import get_tools, create_tool_node
 
 class GraphBuilder:
     
@@ -24,7 +26,36 @@ class GraphBuilder:
         self.graph_builder.add_node("chatbot", self.basic_chatbot_node.process)
         self.graph_builder.add_edge(START, "chatbot")
         self.graph_builder.add_edge("chatbot", END)
+    
+    def chatbot_with_tools_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 the tool and tool node
+         tools = get_tools()
+         tool_node = create_tool_node(tools)
+         
+         ## Define the LLM
+         llm = self.llm
+         
+         ## Define Chatbot node
+         obj_chatbot_with_tools_node = ChatbotWithToolsNode(llm)
+         chatbot_node = obj_chatbot_with_tools_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_edge(START,"chatbot")
+         self.graph_builder.add_conditional_edges("chatbot", tools_condition)
+         self.graph_builder.add_edge("tools", "chatbot")
+         
         
     def setup_graph(self, usecase: str):
         """
@@ -34,6 +65,6 @@ class GraphBuilder:
             self.basic_chatbot_build_graph()
 
         if usecase == "Chatbot with Tool":
-            # self.chatbot_with_tools_build_graph()
-            pass
+            self.chatbot_with_tools_build_graph()
+            
         return self.graph_builder.compile()

src/langgraph_agenticai/nodes/chatbot_with_tools_node.py

@@ -0,0 +1,37 @@
+from src.langgraph_agenticai.state.state import State
+
+class ChatbotWithToolsNode:
+    """
+    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/langgraph_agenticai/tools/search_tool.py

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

src/langgraph_agenticai/ui/streamlit_ui/loadui.py

@@ -23,7 +23,7 @@ class LoadStreamlitUI:
     }
 
     def load_streamlit_ui(self):
-        st.set_page_config(page_title=self.config.get_page_title() + "⚡🧠", layout="wide")
+        st.set_page_config(page_title=self.config.get_page_title(), layout="wide")
         st.header(self.config.get_page_title())
         st.session_state.timeframe = ''
         st.session_state.IsFetchButtonClicked = False

src/langgraph_agenticai/ui/uiconfigfile.ini

@@ -1,5 +1,5 @@
 [DEFAULT]
-PAGE_TITLE = LangGraph AgenticAI
+PAGE_TITLE = LangGraph: AgenticAI Application Usecases
 LLM_OPTIONS = Groq, OpenAI
-USECASE_OPTIONS = Basic Chatbot, Chatbot with Tools, Travel Planner, AI News, SDLC Workflow, Blog Generator, Appointment Receptionist
+USECASE_OPTIONS = Basic Chatbot, Chatbot with Tool, Travel Planner, AI News, SDLC Workflow, Blog Generator, Appointment Receptionist
 GROQ_MODEL_OPTIONS = gemma2-9b-it, llama3-8b-8192, llama3-70b-8192