Spaces:
Sleeping
Sleeping
File size: 2,619 Bytes
637fd0b 12be6a2 637fd0b 12be6a2 637fd0b 12be6a2 637fd0b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 |
from langgraph.graph import StateGraph, START, END, MessagesState
from langgraph.prebuilt import tools_condition, tool_node
from langchain_core.prompts import ChatPromptTemplate
from src.langgraphagenticai.state.state import State
from src.langgraphagenticai.nodes.basic_chatbot_node import BasicChatbotNode
from src.langgraphagenticai.nodes.chatbot_with_tool_node import ChatbotWithToolNode
from src.langgraphagenticai.tools.search_tools import get_tools, create_tool_node
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 exi 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.add_edge(START, "chatbot")
self.graph_builder.add_edge("chatbot", END)
def chatbot_with_tools_build_graph(self):
"""
Builds an advanced chatbot with tool integration.
This method creates a chatbot graph that includes both
a chatbot node and 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 an
entry point.
"""
## Define the tool and tool node
tools = get_tools()
tool_node = create_tool_node(tools)
## Define LLM
llm = self.llm
# Define the 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_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):
"""
Sets up the graph for the selected use case.
"""
if usecase == "Basic Chatbot":
self.basic_chatbot_build_graph()
if usecase == "Chatbot with Tools":
self.chatbot_with_tools_build_graph()
return self.graph_builder.compile()
|