Skip to content

Instantly share code, notes, and snippets.

@jgoodie

jgoodie/TavilyAgentGraph.py

Created November 4, 2024 04:42
Show Gist options
  • Save jgoodie/9ad814a9597b2ac58d05798e1d3ad35f to your computer and use it in GitHub Desktop.
Save jgoodie/9ad814a9597b2ac58d05798e1d3ad35f to your computer and use it in GitHub Desktop.
Download ZIP
Raw
TavilyAgentGraph.py
tavily_builder = StateGraph(TavilyState)
tavily_builder.add_node("TavilySearch", TavilySearch)
tavily_builder.add_node("TavilySummary", TavilySummary)
tavily_builder.add_edge(START, "TavilySearch")
tavily_builder.add_edge("TavilySearch", "TavilySummary")
tavily_builder.add_edge("TavilySummary", END)
graph = tavily_builder.compile()
display(Image(graph.get_graph().draw_mermaid_png()))
result = graph.invoke({"question": "What are the primary product categories and product features for Databricks?"})
print(result['answer'].content)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment