Umer797 commited on
Commit
fdfde63
·
verified ·
1 Parent(s): 2ec3c20

Update graph_builder.py

Browse files
Files changed (1) hide show
  1. graph_builder.py +4 -2
graph_builder.py CHANGED
@@ -1,14 +1,16 @@
1
  from langgraph.graph import StateGraph
2
  from langchain_core.tools import Tool
3
  from langchain_community.chat_models import ChatOpenAI
4
- from langchain_community.tools.tavily_search import TavilySearchAPIWrapper
 
5
 
6
 
7
  def build_graph():
8
  graph = StateGraph(dict)
9
 
10
  # Setup tools
11
- search = TavilySearchAPIWrapper(api_key="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")
 
12
 
13
  def search_step(state):
14
  question = state.get("question")
 
1
  from langgraph.graph import StateGraph
2
  from langchain_core.tools import Tool
3
  from langchain_community.chat_models import ChatOpenAI
4
+ from langchain_community.tools import TavilySearchResults
5
+
6
 
7
 
8
  def build_graph():
9
  graph = StateGraph(dict)
10
 
11
  # Setup tools
12
+ search = TavilySearchResults()
13
+
14
 
15
  def search_step(state):
16
  question = state.get("question")