Update rag_langgraph.py

rag_langgraph.py

@@ -1,23 +1,14 @@
-from datetime import date
+import datetime
 
-from typing import Annotated, List, Tuple, Union
-
-from langchain_community.tools.tavily_search import TavilySearchResults
-from langchain_core.tools import tool
+from typing import Annotated, Any, Dict, List, Optional, Sequence, Tuple, TypedDict, Union
 
 from langchain.agents import AgentExecutor, create_openai_tools_agent
-from langchain_core.messages import BaseMessage, HumanMessage
-from langchain_openai import ChatOpenAI
-
-from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
+from langgraph.graph import StateGraph, END
+from langchain_community.tools.tavily_search import TavilySearchResults
 from langchain_core.output_parsers.openai_functions import JsonOutputFunctionsParser
-
-import operator
-from typing import Annotated, Any, Dict, List, Optional, Sequence, TypedDict
-import functools
-
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
-from langgraph.graph import StateGraph, END
+from langchain_core.tools import BaseMessage, HumanMessage, tool
+from langchain_openai import ChatOpenAI
 
 LLM = "gpt-4o"
 
@@ -130,7 +121,4 @@ def run_multi_agent(topic, word_count):
             HumanMessage(content=topic)
         ]
     })
-    print("#######")
-    print(result['messages'][-1].content)
-    print("#######")
     return result['messages'][-1].content