AhsanRazi commited on
Commit
4b46ed3
·
verified ·
1 Parent(s): 460f2c3

Update blog_title_generator.py

Browse files
Files changed (1) hide show
  1. blog_title_generator.py +2 -2
blog_title_generator.py CHANGED
@@ -20,7 +20,6 @@ tavily_api_key = os.getenv("TAVILY_API_KEY")
20
  def blog_titles_agent(llm, state):
21
 
22
  queries = state["search_queries"]
23
-
24
 
25
  def tavily_search(query: str) -> str:
26
  """
@@ -37,6 +36,7 @@ def blog_titles_agent(llm, state):
37
  search_depth="advanced",
38
  include_answer=True,
39
  include_raw_content=True,
 
40
  )
41
 
42
  text = tavily.invoke({"query": f"{query}"})
@@ -97,7 +97,7 @@ def blog_titles_agent(llm, state):
97
 
98
  titles = []
99
  for q in queries:
100
- time.sleep(5)
101
  messages = [HumanMessage(content=f"Search this query: {q}")]
102
  messages = react_graph.invoke({"messages": messages})
103
  print("\n",messages)
 
20
  def blog_titles_agent(llm, state):
21
 
22
  queries = state["search_queries"]
 
23
 
24
  def tavily_search(query: str) -> str:
25
  """
 
36
  search_depth="advanced",
37
  include_answer=True,
38
  include_raw_content=True,
39
+ api_key=tavily_api_key, # type: ignore
40
  )
41
 
42
  text = tavily.invoke({"query": f"{query}"})
 
97
 
98
  titles = []
99
  for q in queries:
100
+ # time.sleep(5)
101
  messages = [HumanMessage(content=f"Search this query: {q}")]
102
  messages = react_graph.invoke({"messages": messages})
103
  print("\n",messages)