sahanacp commited on
Commit
9856a4f
·
verified ·
1 Parent(s): 7c6b34c

Update agent/planning_agent.py

Browse files
Files changed (1) hide show
  1. agent/planning_agent.py +2 -2
agent/planning_agent.py CHANGED
@@ -23,12 +23,12 @@ query_memory = None
23
  agent = None
24
 
25
  def initialize_planning_agent(llm_instance, chat_memory_instance, query_memory_instance, callback_fn):
26
- global llm, chat_memory, query_memory, agent, callback
27
 
28
  llm = llm_instance
29
  chat_memory = chat_memory_instance
30
  query_memory = query_memory_instance
31
- callback = callback_fn
32
 
33
  # Initialize agents
34
  router_agent.initialize_router_agent(llm, chat_memory)
 
23
  agent = None
24
 
25
  def initialize_planning_agent(llm_instance, chat_memory_instance, query_memory_instance, callback_fn):
26
+ global llm, chat_memory, query_memory, agent, callbacks
27
 
28
  llm = llm_instance
29
  chat_memory = chat_memory_instance
30
  query_memory = query_memory_instance
31
+ callbacks = callback_fn
32
 
33
  # Initialize agents
34
  router_agent.initialize_router_agent(llm, chat_memory)