AI News Agent

src/langgraph_agenticai/graph/graph_builder.py

@@ -4,6 +4,7 @@ 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.nodes.chatbot_ai_news_node import ChatbotAINewsNode
 from src.langgraph_agenticai.tools.search_tool import get_tools, create_tool_node
 
 class GraphBuilder:
@@ -55,6 +56,31 @@ class GraphBuilder:
          self.graph_builder.add_edge(START,"chatbot")
          self.graph_builder.add_conditional_edges("chatbot", tools_condition)
          self.graph_builder.add_edge("tools", "chatbot")
+    
+    def chatbot_ai_news_biuld_graph(self):
+        """
+        Chatbot which provided AI News 
+        
+        """
+        ## 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_ai_news_node = ChatbotAINewsNode(llm)
+        chatbot_node = obj_chatbot_ai_news_node.create_ai_new_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):
@@ -67,4 +93,8 @@ class GraphBuilder:
         if usecase == "Chatbot with Tool":
             self.chatbot_with_tools_build_graph()
             
+        if usecase == "AI News":
+            self.chatbot_ai_news_biuld_graph()   
+            pass 
+            
         return self.graph_builder.compile()

src/langgraph_agenticai/nodes/chatbot_ai_news_node.py

@@ -0,0 +1,39 @@
+from src.langgraph_agenticai.state.state import State
+from langchain_core.prompts import ChatPromptTemplate
+from langchain.agents import create_openai_tools_agent, AgentExecutor
+
+class ChatbotAINewsNode:
+    
+    def __init__(self, model):
+        self.llm = model
+        
+        
+    def create_ai_new_chatbot(self, tools):
+        
+       # AI expert system prompt
+        system_prompt = """You are a professional AI news analyst with expertise in:
+        - Breaking technology news
+        - Emerging AI research trends
+        - Industry developments in artificial intelligence
+        - Analysis of AI ethics and policy
+
+        Guidelines:
+        1. Provide up-to-date, accurate information
+        2. Always verify facts using Tavily search
+        3. Include relevant sources from search results
+        4. Explain technical terms when needed
+        5. Maintain neutral, objective tone
+        
+        Provide only information about the AI News and Trends. For other queries, please respond with the reson that you are only the AI News Expert.
+        """
+
+
+        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([system_prompt] + state["messages"])]}
+        
+        return chatbot_node

src/langgraph_agenticai/ui/streamlit_ui/display_result.py

@@ -23,7 +23,7 @@ class DisplayResultStreamlit:
                         with st.chat_message("assistant"):
                             st.write(value["messages"].content)
 
-        elif usecase=="Chatbot with Tool":
+        elif usecase=="Chatbot with Tool" or usecase=="AI News":
              # Prepare state and invoke the graph
             initial_state = {"messages": [user_message]}
             res = graph.invoke(initial_state)

src/langgraph_agenticai/ui/streamlit_ui/loadui.py

@@ -52,7 +52,7 @@ class LoadStreamlitUI:
             # Use case selection
             self.user_controls["selected_usecase"] = st.selectbox("Select Usecases", usecase_options)
 
-            if self.user_controls["selected_usecase"] =="Chatbot with Tool":
+            if self.user_controls["selected_usecase"] =="Chatbot with Tool" or self.user_controls["selected_usecase"] =="AI News":
                 # API key input
                 os.environ["TAVILY_API_KEY"] = self.user_controls["TAVILY_API_KEY"] = st.session_state["TAVILY_API_KEY"] = st.text_input("TAVILY API KEY",
                                                                                                       type="password")