naman1102 commited on
Commit
b9ed6f9
·
1 Parent(s): 4906fcc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ import time
7
  import pandas as pd
8
  from datetime import datetime
9
  from typing import List, Dict, Any, Annotated, Optional
10
- from langgraph.graph import Graph, StateGraph
11
  from typing_extensions import TypedDict
12
  from openai import OpenAI
13
  from tools import simple_search
@@ -487,7 +487,7 @@ QUESTION:
487
 
488
  # Add edges from each processing node to END
489
  for node in ["process_image", "process_video", "process_spreadsheet", "process_python", "process_text"]:
490
- sg.add_edge(node, "END")
491
 
492
  return sg.compile()
493
 
 
7
  import pandas as pd
8
  from datetime import datetime
9
  from typing import List, Dict, Any, Annotated, Optional
10
+ from langgraph.graph import Graph, StateGraph, END
11
  from typing_extensions import TypedDict
12
  from openai import OpenAI
13
  from tools import simple_search
 
487
 
488
  # Add edges from each processing node to END
489
  for node in ["process_image", "process_video", "process_spreadsheet", "process_python", "process_text"]:
490
+ sg.add_edge(node, END)
491
 
492
  return sg.compile()
493