Created
April 26, 2025 06:31
-
-
Save ashen007/8efaa70de78538d6ea59eb14f8bb76bc to your computer and use it in GitHub Desktop.
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| from langgraph.graph import StateGraph, START, END | |
| from langgraph.graph.graph import CompiledGraph | |
| from states import State | |
| from tools.database_tools import db_query_tool | |
| from tools.tool_nodes import query_gen_node, should_continue, final_answer_node | |
| from tools.error_handling import create_tool_node_with_fallback | |
| def run() -> CompiledGraph: | |
| workflow = StateGraph(State) | |
| workflow.add_node("query_gen", query_gen_node) | |
| workflow.add_node("execute_query", create_tool_node_with_fallback([db_query_tool])) | |
| workflow.add_node('final_answer', final_answer_node) | |
| workflow.add_edge(START, "query_gen") | |
| workflow.add_edge("query_gen", "execute_query") | |
| workflow.add_conditional_edges("execute_query", should_continue) | |
| workflow.add_edge("final_answer", END) | |
| # Compile the workflow into a runnable | |
| app = workflow.compile() | |
| return app |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment