Spaces:
Build error
Build error
Update agent/planning_agent.py
Browse files- 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,
|
| 27 |
|
| 28 |
llm = llm_instance
|
| 29 |
chat_memory = chat_memory_instance
|
| 30 |
query_memory = query_memory_instance
|
| 31 |
-
|
| 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)
|