update

src/langgraphagenticai/graph/graph_builder.py

@@ -3,6 +3,7 @@ from langgraph.prebuilt import tools_condition,ToolNode
 from langchain_core.prompts import ChatPromptTemplate
 import datetime
 from src.langgraphagenticai.state.state import State
+from src.langgraphagenticai.nodes.basic_chatbot_node import BasicChatbotNode
 
 class GraphBuilder:
 
@@ -11,5 +12,25 @@ class GraphBuilder:
         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 into the graph.The chatbot node is set as both the entry 
+        entry and exit point of graph
+
+        """
+        self.basic_chatbot_node = BasicChatbotNode(self.llm)
+        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 setup_graph(self,usecase:str):
+        """ 
+        Sets up the graph for the selected use cases
+        """
+        if usecase == "Basic Chatbot":
+            self.basic_chatbot_build_graph()
+
+        return self.graph_builder.compile()
+
 

src/langgraphagenticai/main.py

@@ -2,6 +2,7 @@ import streamlit as st
 import json
 from src.langgraphagenticai.ui.streamlitui.loadui import LoadStreamlitUI
 from src.langgraphagenticai.LLMS.groqllm import GroqLLM
+from src.langgraphagenticai.graph.graph_builder import GraphBuilder
 
 def load_langgraph_agenticai_app():
     """
@@ -38,9 +39,19 @@ def load_langgraph_agenticai_app():
                 st.error("Error: Usecase not selected.")
                 return
             
+
+            # Graph Builder
+            graph_builder = GraphBuilder(model)
+            try:
+                graph = graph_builder.setup_graph(usecase=usecase)
+            except Exception as e:
+                raise ValueError(f"Error: Graph set up Failed - {e}")
+                return 
+            
         except Exception as e:
             raise ValueError(f"Error occured with exception : {e}")
         
-    # Graph Builder
-    
+ 
+
+