Update blog_content_generator.py

blog_content_generator.py

@@ -16,6 +16,7 @@ def blog_content_agent(llm, state):
   print(topic)
   
   # tool
+  
   def tavily_search(query: str) -> str:
     """
     Performs a search on Tavily using the provided query.
@@ -26,6 +27,7 @@ def blog_content_agent(llm, state):
     Returns:
         str: The result of the search query.
     """
+    
     current_year = date.today().strftime("%Y")
     
     tavily = TavilySearchResults(
@@ -33,6 +35,7 @@ def blog_content_agent(llm, state):
     search_depth="advanced",
     include_answer=True,
     include_raw_content=True,
+    api_key=tavily_api_key, # type: ignore
     )
     
     text = tavily.invoke({"query": f"Search this in UK: {query} After: {current_year}"})
@@ -41,6 +44,7 @@ def blog_content_agent(llm, state):
     return text
 
   tool = [tavily_search]
+#   llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash-exp", api_key =  gemini_api_key)
   llm_with_tools = llm.bind_tools(tool)
 
 
@@ -186,4 +190,5 @@ Output Format:
 
   message = [HumanMessage(content=f"Search this query: {topic}")]
   messages = react_graph.invoke({"messages": message})
+#   print(messages['messages'][-1].content)
   return {"blog_content": messages['messages'][-1].content}