Update app.py

app.py

@@ -46,6 +46,8 @@ llm = ChatGroq(model = "qwen-qwq-32b",temperature=0.1)
 
 memorysaver = MemorySaver()
 
+clearMemory = False
+
 class State(TypedDict):
      messages: Annotated[list[AnyMessage], add_messages]
     
@@ -68,8 +70,9 @@ def build_tools():
     tools = [TavilySearchResults(max_results=2)]
     return tools
 
-def get_llm(clearMemory):
+def get_llm():
     global llm
+    global clearMemory
     if clearMemory:
         llm = ChatGroq(model = "qwen-qwq-32b",temperature=0.1)
     tools = build_tools()
@@ -77,8 +80,7 @@ def get_llm(clearMemory):
     llm_with_tools = llm.bind_tools(tools)
     return llm_with_tools
 
-def llm_callingTools(state:State, clearMemory = False): 
-    
+def llm_callingTools(state:State):    
     format_instructions = parser.get_format_instructions()
        
     system_msg = SystemMessage(content=f"""You are a smart movie researcher.
@@ -95,15 +97,17 @@ def llm_callingTools(state:State, clearMemory = False):
                     Think carefully before responding: **Is the latest message is referring to a specific show or programme, even indirectly?** Only then use the formatted output.""")
     
     human_message = HumanMessage( content=f"{state['messages']}.")
-    llm_with_tools = get_llm(clearMemory)
+    llm_with_tools = get_llm()
     return {"messages": [llm_with_tools.invoke([system_msg]+ [human_message])]}
 
-def build_graph(clearMemory: bool = False):
+def build_graph(memory: bool = False):
     global memorysaver
+    global clearMemory
+    clearMemory = memory
     if clearMemory:
         memorysaver = MemorySaver()
     graph_builder = StateGraph(State)
-    graph_builder.add_node("llm_with_tool", llm_callingTools(clearMemory))
+    graph_builder.add_node("llm_with_tool", llm_callingTools)
     graph_builder.add_node("tools", ToolNode(build_tools()))
     graph_builder.add_edge(START, "llm_with_tool")
     graph_builder.add_conditional_edges("llm_with_tool", tools_condition)
@@ -118,6 +122,7 @@ def is_pattern_in_string(string: str) -> bool:
         
 @app.post("/api/v1/get_programme_info")
 def get_data_by_prompt(prompt: str, thread_id: str): 
+    global clearMemory
     clearMemory = False
     try:
         print(f"Prompt: {prompt}")